error model

Author: pamelafox

src/backend/fastapi_app/__init__.py

@@ -52,10 +52,10 @@ def create_app(testing: bool = False):
     else:
         if not testing:
             load_dotenv(override=True)
-        logging.basicConfig(level=logging.DEBUG)
+        logging.basicConfig(level=logging.INFO)
     # Turn off particularly noisy INFO level logs from Azure Core SDK:
-    # logging.getLogger("azure.core.pipeline.policies.http_logging_policy").setLevel(logging.WARNING)
-    # logging.getLogger("azure.identity").setLevel(logging.WARNING)
+    logging.getLogger("azure.core.pipeline.policies.http_logging_policy").setLevel(logging.WARNING)
+    logging.getLogger("azure.identity").setLevel(logging.WARNING)
 
     if os.getenv("APPLICATIONINSIGHTS_CONNECTION_STRING"):
         logger.info("Configuring Azure Monitor")

src/backend/fastapi_app/api_models.py

@@ -52,6 +52,10 @@ class RAGContext(BaseModel):
     followup_questions: list[str] | None = None
 
 
+class ErrorResponse(BaseModel):
+    error: str
+
+
 class RetrievalResponse(BaseModel):
     message: Message
     context: RAGContext

src/backend/fastapi_app/routes/api_routes.py

@@ -9,6 +9,7 @@
 
 from fastapi_app.api_models import (
     ChatRequest,
+    ErrorResponse,
     ItemPublic,
     ItemWithDistance,
     RetrievalResponse,
@@ -89,7 +90,7 @@ async def search_handler(
     return [ItemPublic.model_validate(item.to_dict()) for item in results]
 
 
-@router.post("/chat", response_model=RetrievalResponse | dict)
+@router.post("/chat", response_model=RetrievalResponse | ErrorResponse)
 async def chat_handler(
     context: CommonDeps,
     database_session: DBSession,
@@ -130,7 +131,6 @@ async def chat_handler(
         )
         return response
     except Exception as e:
-        # return exception inside JSON
         return {"error": str(e)}