Merge pull request #70 from AET-DevOps25/genai_tests

Add tests to genai

Author: ArmanpreetGhotra

.github/workflows/genai-tests.yml

@@ -26,6 +26,9 @@ jobs:
           uv pip install -r ./genai/requirements.txt --system
 
       - name: GenAI tests
+        env: 
+         OPEN_WEB_UI_API_KEY: ${{ secrets.OPEN_WEB_UI_API_KEY }}
+         API_URL: ${{ vars.API_URL }}
         run: |
           cd genai
           pytest

compose.yml

@@ -120,6 +120,11 @@ services:
       DB_HOST: ${DB_HOST:-weaviate}
       DB_PORT: ${DB_PORT:-9090}
       DB_GRPC_PORT: ${DB_GRPC_PORT:-50051}
+      OPEN_WEB_UI_API_KEY: ${OPEN_WEB_UI_API_KEY}
+      SERVER_URL: ${SERVER_URL:-http://localhost:9091}
+      CLIENT_URL: ${CLIENT_URL:-http://localhost:3000}
+      GENAI_URL: ${GENAI_URL:-http://localhost:8000}
+      API_URL: ${API_URL:-https://gpu.aet.cit.tum.de/api/chat/completions}
     depends_on:
       - weaviate
     volumes:

genai/.env.example

@@ -1 +1,2 @@
-CHAIR_API_KEY=your-api-key-here
+OPEN_WEB_UI_API_KEY=your-api-key-here
+API_URL="https://your-api-url-here"

genai/app/main.py

@@ -20,13 +20,15 @@
 router = APIRouter()
 
 # Environment configuration
-CHAIR_API_KEY = os.getenv("CHAIR_API_KEY")
-API_URL = "https://gpu.aet.cit.tum.de/api/chat/completions"
-
+OPEN_WEB_UI_API_KEY = os.getenv("OPEN_WEB_UI_API_KEY")
+API_URL = os.getenv("API_URL")
+SERVER_URL = os.getenv("SERVER_URL")
+CLIENT_URL = os.getenv("CLIENT_URL")
+GENAI_URL = os.getenv("GENAI_URL")
 
 class OpenWebUILLM(LLM):
     api_url: str = API_URL
-    api_key: str = CHAIR_API_KEY
+    api_key: str = OPEN_WEB_UI_API_KEY
     model_name: str = "llama3.3:latest"
 
     @property
@@ -41,7 +43,7 @@ def _call(
         **kwargs: Any,
     ) -> str:
         if not self.api_key:
-            raise ValueError("CHAIR_API_KEY environment variable is required")
+            raise ValueError("API_KEY environment variable is required")
 
         headers = {
             "Authorization": f"Bearer {self.api_key}",
@@ -97,14 +99,14 @@ def custom_openapi():
             title=app.title,
             version=app.version,
             routes=app.routes,
-            servers=[{"url": "http://localhost:8000"}],
+            servers=[{"url": GENAI_URL}],
         )
     )
 
 
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["http://localhost:3000", "http://localhost:9091"],
+    allow_origins=[CLIENT_URL, SERVER_URL],
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],

genai/app/test.py

@@ -0,0 +1,29 @@
+from fastapi.testclient import TestClient
+from app.main import app
+
+client = TestClient(app)
+
+def test_health_check():
+    response =  client.get("/health")
+    assert response.status_code == 200
+    assert "status" in response.json()
+
+
+def test_completion():
+    payload = {"user_text": ["This is a test input."]}
+    response = client.post("/completion", json=payload)
+    assert response.status_code == 200
+    assert "llm_response" in response.json()
+
+def test_summarization():
+    payload = {"user_text": ["This is a long sentence that needs summarizing."]}
+    response = client.post("/summarization", json=payload)
+    assert response.status_code == 200
+    assert "llm_response" in response.json()
+
+
+def test_rephrase():
+    payload = {"user_text": ["This is a sample sentence."]}
+    response = client.post("/rephrase", json=payload)
+    assert response.status_code == 200
+    assert "llm_response" in response.json()

genai/pyproject.toml

@@ -74,4 +74,13 @@ line-ending = "auto"
 #
 # This only has an effect when the `docstring-code-format` setting is
 # enabled.
-#docstring-code-line-length = "dynamic"
+#docstring-code-line-length = "dynamic"
+
+[tool.pytest.ini_options]
+asyncio_mode = "auto"
+testpaths = ["app"]
+python_files = ["test_*.py", "*_test.py", "test.py"]
+addopts = "-v"
+markers = [
+    "asyncio: mark test as an async test",
+]

genai/requirements.txt

@@ -1,10 +1,9 @@
 annotated-types==0.7.0
-anyio==4.9.0
 authlib==1.3.1
 certifi==2025.4.26
 cffi==1.17.1
 click==8.1.8
-cryptography==44.0.3
+cryptography==43.0.3
 deprecation==2.1.0
 dnspython==2.7.0
 email-validator==2.2.0