Fix linting

Author: edkaya

genai/routes/routes.py

@@ -1,4 +1,11 @@
-from fastapi import APIRouter, UploadFile, File, HTTPException, Request, Depends
+from fastapi import (
+    APIRouter,
+    UploadFile,
+    File,
+    HTTPException,
+    Request,
+    Depends
+)
 from fastapi.responses import JSONResponse
 import os
 
@@ -103,7 +110,10 @@ async def upload_file(
                 filename
             )
         ):
-            logger.info("File already exists in qdrant for user %s", current_user.username)
+            logger.info(
+                "File already exists in qdrant for user %s",
+                current_user.username
+                )
             return {"message": f"File '{filename}' already uploaded."}
 
         vector_store = qdrant.create_and_get_vector_storage(collection_name)
@@ -136,7 +146,9 @@ async def generate(request: Request):
 
     body = await request.json()
     if "query" not in body or "messages" not in body or "user_id" not in body:
-        logger.error("Missing 'query', 'messages', or 'user_id' in the request body")
+        logger.error(
+            "Missing 'query', 'messages', or 'user_id' in the request body"
+            )
         raise HTTPException(
             status_code=400,
             detail="Missing 'query', 'messages', or 'user_id'"
@@ -146,7 +158,7 @@ async def generate(request: Request):
     messages_raw = body["messages"]
     user_id = body["user_id"]
     collection_name = f"recipes_{user_id}"
-    
+
     logger.info("Generate endpoint called for user_id: %s", user_id)
 
     try:

genai/service/auth_service.py

@@ -10,26 +10,35 @@ def __init__(self, user_id: int, username: str):
         self.username = username
 
 
-async def get_current_user(authorization: Optional[str] = Header(None)) -> UserInfo:
+async def get_current_user(
+    authorization: Optional[str] = Header(None)
+    ) -> UserInfo:
     """
     Extract user information from the Authorization header.
-    
+
     This function validates the OAuth token by calling the user service
     and returns the user information including user_id.
     """
     if not authorization:
         logger.error("Authorization header is missing")
-        raise HTTPException(status_code=401, detail="Authorization header required")
-    
+        raise HTTPException(
+            status_code=401,
+            detail="Authorization header required"
+            )
+
     if not authorization.startswith("Bearer "):
         logger.error("Invalid authorization header format")
-        raise HTTPException(status_code=401, detail="Invalid authorization header format")
-    
+        raise HTTPException(
+            status_code=401,
+            detail="Invalid authorization header format"
+            )
+
     token = authorization.split(" ")[1]
-    
+
+    # use dev and prod URLs for user service, where prod is the fallback
     user_service_urls = [
-    "http://localhost:8081/user/info",   # dev
-    "http://user-service:8081/user/info" # prod      
+    "http://user-service:8081/user/info",
+    "http://localhost:8081/user/info",
     ]
 
     for url in user_service_urls:
@@ -41,7 +50,9 @@ async def get_current_user(authorization: Optional[str] = Header(None)) -> UserI
             )
             if response.status_code == 200:
                 user_data = response.json()
-                logger.info(f"User authenticated: {user_data.get('username')}")
+                logger.info(
+                    f"User authenticated: {user_data.get('username')}"
+                    )
                 return UserInfo(
                     user_id=user_data.get("id"),
                     username=user_data.get("username")
@@ -51,4 +62,7 @@ async def get_current_user(authorization: Optional[str] = Header(None)) -> UserI
         except requests.exceptions.RequestException as e:
             logger.warning(f"Failed to reach {url}: {e}")
 
-    raise HTTPException(status_code=500, detail="Authentication service unavailable")
+    raise HTTPException(
+        status_code=500,
+        detail="Authentication service unavailable"
+        )

genai/tests/integration/generation_test.py

@@ -61,4 +61,6 @@ def test_generate_endpoint_missing_fields():
     response = client.post("/genai/generate", json=payload)
 
     assert response.status_code == 400
-    assert response.json() == {"detail": "Missing 'query', 'messages', or 'user_id'"}
+    assert response.json() == {
+        "detail": "Missing 'query', 'messages', or 'user_id'"
+        }

genai/tests/integration/upload_test.py

@@ -18,7 +18,7 @@ def test_upload_file_success(
     # Mock the authentication by overriding the dependency
     mock_user = UserInfo(user_id=123, username="test_user")
     app.dependency_overrides[get_current_user] = lambda: mock_user
-    
+
     mock_pipeline = MagicMock()
     mock_pipeline_class.return_value = mock_pipeline
 
@@ -36,7 +36,7 @@ def test_upload_file_success(
 
     mock_pipeline_class.assert_called_once()
     mock_pipeline.ingest.assert_called_once()
-    
+
     # Clean up the dependency override
     app.dependency_overrides.clear()
 
@@ -45,7 +45,7 @@ def test_upload_file_invalid_type():
     # Mock the authentication by overriding the dependency
     mock_user = UserInfo(user_id=123, username="test_user")
     app.dependency_overrides[get_current_user] = lambda: mock_user
-    
+
     file = io.BytesIO(b"just some text")
     file.name = "notes.txt"
 
@@ -57,7 +57,7 @@ def test_upload_file_invalid_type():
     assert response.status_code == 400
     assert (response.json()["detail"] ==
             "Invalid file type. Only PDF files are allowed.")
-    
+
     # Clean up the dependency override
     app.dependency_overrides.clear()
 
@@ -68,7 +68,7 @@ def test_upload_file_already_exists(_mock_contains, _mock_exists):
     # Mock the authentication
     mock_user = UserInfo(user_id=123, username="test_user")
     app.dependency_overrides[get_current_user] = lambda: mock_user
-    
+
     file = io.BytesIO(b"%PDF-1.4")
     file.name = "existing.pdf"
 
@@ -79,6 +79,6 @@ def test_upload_file_already_exists(_mock_contains, _mock_exists):
 
     assert response.status_code == 200
     assert "already uploaded" in response.json()["message"]
-    
+
     # Clean up the dependency override
     app.dependency_overrides.clear()