run linters

Author: erinmikailstaples

python/rag/healthcare-support-portal/packages/auth/src/auth_service/routers/auth.py

@@ -66,9 +66,7 @@ async def register(
             detail="Only administrators can create new users. Use POST /api/v1/users/ endpoint instead.",
         )
     # Check if user already exists
-    existing_user = (
-        db.query(User).filter((User.username == user_data.username) | (User.email == user_data.email)).first()
-    )
+    existing_user = db.query(User).filter((User.username == user_data.username) | (User.email == user_data.email)).first()
 
     if existing_user:
         raise HTTPException(

python/rag/healthcare-support-portal/packages/auth/src/auth_service/routers/users.py

@@ -130,9 +130,7 @@ async def create_user(
         )
 
     # Check if user already exists
-    existing_user = (
-        db.query(User).filter((User.username == user_data.username) | (User.email == user_data.email)).first()
-    )
+    existing_user = db.query(User).filter((User.username == user_data.username) | (User.email == user_data.email)).first()
 
     if existing_user:
         raise HTTPException(

python/rag/healthcare-support-portal/packages/common/src/common/migration_check.py

@@ -20,9 +20,7 @@ def get_alembic_config() -> Config:
     alembic_ini = common_dir / "alembic.ini"
 
     if not alembic_ini.exists():
-        raise FileNotFoundError(
-            f"alembic.ini not found at {alembic_ini}. " "Please ensure Alembic is properly initialized."
-        )
+        raise FileNotFoundError(f"alembic.ini not found at {alembic_ini}. " "Please ensure Alembic is properly initialized.")
 
     config = Config(str(alembic_ini))
 

python/rag/healthcare-support-portal/packages/common/src/common/seed_data.py

@@ -197,13 +197,9 @@ def seed_patients(db: Session, users: dict[str, User], admin_token: str) -> list
 
     for patient_data in demo_patients:
         # Check if patient already exists
-        existing_patient = (
-            db.query(Patient).filter(Patient.medical_record_number == patient_data["medical_record_number"]).first()
-        )
+        existing_patient = db.query(Patient).filter(Patient.medical_record_number == patient_data["medical_record_number"]).first()
         if existing_patient:
-            print(
-                f"✅ Patient '{patient_data['name']}' (MRN: {patient_data['medical_record_number']}) already exists, skipping"
-            )
+            print(f"✅ Patient '{patient_data['name']}' (MRN: {patient_data['medical_record_number']}) already exists, skipping")
             created_patients.append(existing_patient)
 
             # Sync OSO facts for existing patients (in case they weren't synced before)
@@ -233,9 +229,7 @@ def seed_patients(db: Session, users: dict[str, User], admin_token: str) -> list
                 # Get patient from database
                 new_patient = db.query(Patient).filter(Patient.id == patient_response["id"]).first()
                 created_patients.append(new_patient)
-                print(
-                    f"✅ Created patient via API (with OSO facts): {patient_data['name']} (MRN: {patient_data['medical_record_number']})"
-                )
+                print(f"✅ Created patient via API (with OSO facts): {patient_data['name']} (MRN: {patient_data['medical_record_number']})")
             else:
                 print(f"⚠️  Failed to create patient {patient_data['name']}: {response.text}")
 

python/rag/healthcare-support-portal/packages/patient/src/patient_service/routers/patients.py

@@ -81,9 +81,7 @@ async def create_patient(
     Create a new patient
     """
     # Check if medical record number already exists
-    existing_patient = (
-        db.query(Patient).filter(Patient.medical_record_number == patient_data.medical_record_number).first()
-    )
+    existing_patient = db.query(Patient).filter(Patient.medical_record_number == patient_data.medical_record_number).first()
 
     if existing_patient:
         raise HTTPException(

python/rag/healthcare-support-portal/packages/rag/src/rag_service/main.py

@@ -64,9 +64,7 @@ async def startup_event():
         # Verify database migrations are current
         require_migrations_current()
 
-        logger.info(
-            f"🚀 {settings.app_name} started successfully", port=settings.port, galileo_enabled=settings.galileo_enabled
-        )
+        logger.info(f"🚀 {settings.app_name} started successfully", port=settings.port, galileo_enabled=settings.galileo_enabled)
 
     except Exception as e:
         logger.error(f"Failed to start {settings.app_name}", error=str(e))

python/rag/healthcare-support-portal/packages/rag/src/rag_service/observability.py

@@ -90,9 +90,7 @@ async def rag_query_context(query_type: str, user_role: str, department: str = N
 
     try:
         # Log query start
-        logger.info(
-            "RAG query started", query_type=query_type, user_role=user_role, department=department, query_id=query_id
-        )
+        logger.info("RAG query started", query_type=query_type, user_role=user_role, department=department, query_id=query_id)
 
         # Start Galileo trace if enabled
         if galileo_enabled and galileo_logger:
@@ -193,9 +191,7 @@ async def embedding_generation_context(model: str, chunk_count: int, operation_i
         yield operation_id
 
     except Exception as e:
-        logger.error(
-            "Embedding generation failed", model=model, chunk_count=chunk_count, operation_id=operation_id, error=str(e)
-        )
+        logger.error("Embedding generation failed", model=model, chunk_count=chunk_count, operation_id=operation_id, error=str(e))
 
         # Log error to Galileo
         log_galileo_event(
@@ -338,9 +334,7 @@ async def ai_response_context(model: str, token_count: int, response_id: str = N
         yield response_id
 
     except Exception as e:
-        logger.error(
-            "AI response generation failed", model=model, token_count=token_count, response_id=response_id, error=str(e)
-        )
+        logger.error("AI response generation failed", model=model, token_count=token_count, response_id=response_id, error=str(e))
 
         # Log error to Galileo
         log_galileo_event(

python/rag/healthcare-support-portal/packages/rag/src/rag_service/routers/chat.py

@@ -61,9 +61,7 @@ async def search_documents(
     """
     settings = request.app.state.settings
 
-    async with rag_query_context(
-        query_type="search", user_role=current_user.role, department=search_request.department
-    ) as query_id:
+    async with rag_query_context(query_type="search", user_role=current_user.role, department=search_request.department) as query_id:
 
         # Get authorized documents query with OSO fallback
         try:
@@ -152,9 +150,7 @@ async def ask_question(
     """
     settings = request.app.state.settings
 
-    async with rag_query_context(
-        query_type="ask", user_role=current_user.role, department=chat_request.context_department
-    ) as query_id:
+    async with rag_query_context(query_type="ask", user_role=current_user.role, department=chat_request.context_department) as query_id:
 
         # Get authorized documents for context with OSO fallback
         try:
@@ -358,7 +354,9 @@ async def generate_ai_response(question: str, context_results: list[dict], user_
 
         # Add disclaimer if no context was used
         if not context:
-            ai_response += "\n\n*Note: This response was generated without specific document context. Please verify information with current medical guidelines.*"
+            ai_response += (
+                "\n\n*Note: This response was generated without specific document context. Please verify information with current medical guidelines.*"
+            )
 
         return ai_response
 

python/rag/healthcare-support-portal/packages/rag/src/rag_service/routers/documents.py

@@ -59,9 +59,7 @@ async def list_documents(
     try:
         query = db.query(Document).options(authorized(current_user, "read", Document))
     except Exception as oso_error:
-        logger.warning(
-            "OSO authorization failed, falling back to basic query", error=str(oso_error), user_role=current_user.role
-        )
+        logger.warning("OSO authorization failed, falling back to basic query", error=str(oso_error), user_role=current_user.role)
         # In development, fallback to showing documents based on role/department
         query = db.query(Document)
         if current_user.role != "admin":
@@ -169,9 +167,7 @@ async def get_document(
             pass
         else:
             # Access denied
-            raise HTTPException(
-                status_code=status.HTTP_403_FORBIDDEN, detail="Access denied - document not in your department"
-            )
+            raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail="Access denied - document not in your department")
 
     logger.info(
         "Document accessed",
@@ -296,9 +292,7 @@ async def upload_document(
             chunk_overlap=settings.chunk_overlap,
         )
 
-        async with embedding_generation_context(
-            model=settings.embedding_model, chunk_count=len(chunks)
-        ) as operation_id:
+        async with embedding_generation_context(model=settings.embedding_model, chunk_count=len(chunks)) as operation_id:
 
             success = await store_document_embeddings(
                 document=document,
@@ -354,9 +348,7 @@ async def upload_document(
                 }
 
     except Exception as e:
-        logger.error(
-            "Error processing uploaded document", document_id=document.id, user_role=current_user.role, error=str(e)
-        )
+        logger.error("Error processing uploaded document", document_id=document.id, user_role=current_user.role, error=str(e))
 
         # Log error to Galileo
         log_galileo_event(
@@ -392,9 +384,7 @@ async def regenerate_embeddings(
     document = db.query(Document).filter(Document.id == document_id).first()
 
     if not document:
-        logger.warning(
-            "Document not found for embedding regeneration", document_id=document_id, user_role=current_user.role
-        )
+        logger.warning("Document not found for embedding regeneration", document_id=document_id, user_role=current_user.role)
         raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Document not found")
 
     # Check authorization
@@ -418,9 +408,7 @@ async def regenerate_embeddings(
             chunk_overlap=settings.chunk_overlap,
         )
 
-        async with embedding_generation_context(
-            model=settings.embedding_model, chunk_count=len(chunks)
-        ) as operation_id:
+        async with embedding_generation_context(model=settings.embedding_model, chunk_count=len(chunks)) as operation_id:
 
             result = await regenerate_document_embeddings(
                 document=document,
@@ -464,9 +452,7 @@ async def regenerate_embeddings(
                 return result
 
     except Exception as e:
-        logger.error(
-            "Error regenerating document embeddings", document_id=document_id, user_role=current_user.role, error=str(e)
-        )
+        logger.error("Error regenerating document embeddings", document_id=document_id, user_role=current_user.role, error=str(e))
 
         # Log error to Galileo
         log_galileo_event(
@@ -503,9 +489,7 @@ async def delete_document(
         oso = get_oso()
         oso.authorize(current_user, "delete", document)
     except Exception as e:
-        logger.warning(
-            "Unauthorized document deletion attempt", document_id=document_id, user_role=current_user.role, error=str(e)
-        )
+        logger.warning("Unauthorized document deletion attempt", document_id=document_id, user_role=current_user.role, error=str(e))
         raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail="Access denied")
 
     # Remove Oso facts

python/rag/healthcare-support-portal/packages/rag/src/rag_service/utils/embeddings.py

@@ -160,11 +160,7 @@ async def similarity_search(
                 formatted_results.append(
                     {
                         "document_title": result["title"],
-                        "content_chunk": (
-                            result["content_chunk"][:200] + "..."
-                            if len(result["content_chunk"]) > 200
-                            else result["content_chunk"]
-                        ),
+                        "content_chunk": (result["content_chunk"][:200] + "..." if len(result["content_chunk"]) > 200 else result["content_chunk"]),
                         "similarity_score": result["similarity_score"],
                         "document_type": result["document_type"],
                         "department": result["department"],
@@ -182,9 +178,7 @@ async def similarity_search(
         return []
 
 
-async def regenerate_document_embeddings(
-    document: Document, db: Session, model: str = "text-embedding-3-small"
-) -> dict:
+async def regenerate_document_embeddings(document: Document, db: Session, model: str = "text-embedding-3-small") -> dict:
     """
     Regenerate embeddings for an existing document.
     Returns status dict with success and message.