Update api/routes.py

api/routes.py

@@ -1,8 +1,10 @@
+# api/routes.py
+
 import json
 from fastapi import APIRouter, Depends, HTTPException, Request, Response
 from fastapi.responses import StreamingResponse
 from api.auth import verify_app_secret
-from api.config import ALLOWED_MODELS
+from api.config import MODELS
 from api.models import ChatRequest
 from api.utils import process_non_streaming_response, process_streaming_response
 from api.logger import setup_logger
@@ -26,7 +28,7 @@ async def chat_completions_options():
 @router.get("/v1/models")
 @router.get("/api/v1/models")
 async def list_models():
-    return {"object": "list", "data": ALLOWED_MODELS}
+    return {"object": "list", "data": MODELS}
 
 @router.post("/v1/chat/completions")
 @router.post("/api/v1/chat/completions")
@@ -36,11 +38,10 @@ 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 [model["id"] for model in ALLOWED_MODELS]:
-        allowed = ', '.join(model['id'] for model in ALLOWED_MODELS)
+    if request.model not in MODELS:
         raise HTTPException(
             status_code=400,
-            detail=f"Model '{request.model}' is not allowed. Allowed models are: {allowed}",
+            detail=f"Model {request.model} is not allowed. Allowed models are: {', '.join(MODELS)}",
         )
 
     if request.stream:
@@ -50,11 +51,11 @@ async def chat_completions(
         logger.info("Non-streaming response")
         return await process_non_streaming_response(request)
 
-# Health check endpoints
+@router.route('/')
+@router.route('/healthz')
+@router.route('/ready')
+@router.route('/alive')
+@router.route('/status')
 @router.get("/health")
-@router.get("/healthz")
-@router.get("/ready")
-@router.get("/alive")
-@router.get("/status")
 def health_check(request: Request):
     return Response(content=json.dumps({"status": "ok"}), media_type="application/json")