Update api/routes.py

api/routes.py

@@ -39,7 +39,9 @@ async def chat_completions(
     logger.info("Entering chat_completions route")
     logger.info(f"Processing chat completion request for model: {request.model}")
 
-    if request.model not in AmigoChat.models and request.model not in AmigoChat.model_aliases:
+    valid_models = AmigoChat.models + list(AmigoChat.model_aliases.keys())
+
+    if request.model not in valid_models:
         raise HTTPException(
             status_code=400,
             detail=f"Model {request.model} is not allowed. Allowed models are: {', '.join(AmigoChat.models)}",
@@ -47,17 +49,21 @@ async def chat_completions(
 
     if request.stream:
         logger.info("Streaming response")
-        return StreamingResponse(process_streaming_response(request), media_type="text/event-stream")
+        return StreamingResponse(
+            process_streaming_response(request),
+            media_type="text/event-stream",
+            headers={"Cache-Control": "no-cache"},
+        )
     else:
         logger.info("Non-streaming response")
         response = await process_non_streaming_response(request)
         return Response(content=json.dumps(response), media_type="application/json")
 
-@router.route('/')
-@router.route('/healthz')
-@router.route('/ready')
-@router.route('/alive')
-@router.route('/status')
+@router.get('/')
+@router.get('/healthz')
+@router.get('/ready')
+@router.get('/alive')
+@router.get('/status')
 @router.get("/health")
 def health_check(request: Request):
     return Response(content=json.dumps({"status": "ok"}), media_type="application/json")