Add codegate package path to poetry (#1012)

* Fix codegate not a package when poetry install

Signed-off-by: Radoslav Dimitrov <[email protected]>

* Add a log message when an integration test has started

Signed-off-by: Radoslav Dimitrov <[email protected]>

---------

Signed-off-by: Radoslav Dimitrov <[email protected]>

Author: rdimitrov

pyproject.toml

@@ -4,7 +4,9 @@ version = "0.1.7"
 description = "Generative AI CodeGen security gateway"
 readme = "README.md"
 authors = []
-
+packages = [
+    { include = "codegate", from = "src" },
+]
 [tool.poetry.dependencies]
 python = ">=3.12,<4.0"
 click = "==8.1.8"

tests/integration/integration_tests.py

@@ -136,6 +136,8 @@ async def run_test(self, test: dict, test_headers: dict) -> bool:
         streaming = data.get("stream", False)
         provider = test["provider"]
 
+        logger.info(f"Starting test: {test_name}")
+
         response = self.call_codegate(url, test_headers, data, provider)
         if not response:
             logger.error(f"Test {test_name} failed: No response received")