Update api/routes.py

api/routes.py

@@ -1,11 +1,10 @@
-# api/routes.py
-
 import json
 from fastapi import APIRouter, Depends, HTTPException, Request, Response
-from fastapi.responses import StreamingResponse, JSONResponse
+from fastapi.responses import StreamingResponse
 from api.auth import verify_app_secret
-from api.models import ChatRequest, ImageResponseModel, ChatCompletionResponse
-from api.utils import process_gizai_stream_response, process_gizai_non_stream_response, GizAI
+from api.config import ALLOWED_MODELS
+from api.models import ChatRequest
+from api.utils import process_non_streaming_response, process_streaming_response
 from api.logger import setup_logger
 
 logger = setup_logger(__name__)
@@ -14,7 +13,7 @@ router = APIRouter()
 
 @router.options("/v1/chat/completions")
 @router.options("/api/v1/chat/completions")
-async def gizai_chat_completions_options():
+async def chat_completions_options():
     return Response(
         status_code=200,
         headers={
@@ -26,33 +25,29 @@ async def gizai_chat_completions_options():
 
 @router.get("/v1/models")
 @router.get("/api/v1/models")
-async def list_gizai_models():
-    return {"object": "list", "data": GizAI.models}
+async def list_models():
+    return {"object": "list", "data": ALLOWED_MODELS}
 
 @router.post("/v1/chat/completions")
 @router.post("/api/v1/chat/completions")
-async def gizai_chat_completions(
+async def chat_completions(
     request: ChatRequest, app_secret: str = Depends(verify_app_secret)
 ):
-    logger.info("Entering GizAI chat_completions route")
+    logger.info("Entering chat_completions route")
     logger.info(f"Processing chat completion request for model: {request.model}")
 
-    model = GizAI.get_model(request.model)
-    if model not in GizAI.models:
+    if request.model not in [model["id"] for model in ALLOWED_MODELS]:
         raise HTTPException(
             status_code=400,
-            detail=f"Model {request.model} is not supported. Supported models are: {', '.join(GizAI.models)}",
+            detail=f"Model {request.model} is not allowed. Allowed models are: {', '.join(model['id'] for model in ALLOWED_MODELS)}",
         )
 
     if request.stream:
-        if GizAI.is_image_model(model):
-            raise HTTPException(status_code=400, detail="Image generation does not support streaming.")
         logger.info("Streaming response")
-        return StreamingResponse(process_gizai_stream_response(request, model), media_type="text/event-stream")
+        return StreamingResponse(process_streaming_response(request), media_type="text/event-stream")
     else:
         logger.info("Non-streaming response")
-        response = await process_gizai_non_stream_response(request, model)
-        return response
+        return await process_non_streaming_response(request)
 
 @router.route('/')
 @router.route('/healthz')