Update api/routes.py

api/routes.py

@@ -1,3 +1,5 @@
+# api/routes.py
+
 import json
 from fastapi import APIRouter, Depends, HTTPException, Request, Response
 from fastapi.responses import StreamingResponse
@@ -39,17 +41,25 @@ async def chat_completions(
     # Validate if the requested model is allowed
     allowed_model_ids = [model["id"] for model in ALLOWED_MODELS]
     if request.model not in allowed_model_ids:
+        logger.warning(f"Model {request.model} is not allowed.")
         raise HTTPException(
             status_code=400,
             detail=f"Model {request.model} is not allowed. Allowed models are: {', '.join(allowed_model_ids)}",
         )
 
-    if request.stream:
-        logger.info("Streaming response")
-        return StreamingResponse(process_streaming_response(request), media_type="text/event-stream")
-    else:
-        logger.info("Non-streaming response")
-        return await process_non_streaming_response(request)
+    try:
+        if request.stream:
+            logger.info("Streaming response")
+            return StreamingResponse(process_streaming_response(request), media_type="text/event-stream")
+        else:
+            logger.info("Non-streaming response")
+            return await process_non_streaming_response(request)
+    except HTTPException as e:
+        logger.error(f"HTTPException: {e.detail}")
+        raise e
+    except Exception as e:
+        logger.error(f"Unexpected error: {str(e)}")
+        raise HTTPException(status_code=500, detail="Internal Server Error")
 
 @router.get("/health")
 @router.get("/")