Update api/routes.py

api/routes.py

@@ -4,9 +4,8 @@ import json
 from fastapi import APIRouter, Depends, HTTPException, Request, Response
 from fastapi.responses import StreamingResponse, JSONResponse
 from api.auth import verify_app_secret
-from api.config import GIZAI_API_ENDPOINT, GIZAI_BASE_URL
 from api.models import ChatRequest, ImageResponseModel, ChatCompletionResponse
-from api.utils import process_gizai_response, GizAI  # Import GizAI here
+from api.utils import process_gizai_stream_response, process_gizai_non_stream_response, GizAI
 from api.logger import setup_logger
 
 logger = setup_logger(__name__)
@@ -14,7 +13,7 @@ logger = setup_logger(__name__)
 router = APIRouter()
 
 @router.options("/v1/chat/completions")
-@router.options("/api/v1/gizai/chat/completions")
+@router.options("/api/v1/chat/completions")
 async def gizai_chat_completions_options():
     return Response(
         status_code=200,
@@ -26,12 +25,12 @@ async def gizai_chat_completions_options():
     )
 
 @router.get("/v1/models")
-@router.get("/api/v1/gizai/models")
+@router.get("/api/v1/models")
 async def list_gizai_models():
-    return {"object": "list", "data": GizAI.models}  # Access GizAI.models directly
+    return {"object": "list", "data": GizAI.models}
 
 @router.post("/v1/chat/completions")
-@router.post("/api/v1/gizai/chat/completions")
+@router.post("/api/v1/chat/completions")
 async def gizai_chat_completions(
     request: ChatRequest, app_secret: str = Depends(verify_app_secret)
 ):
@@ -46,11 +45,14 @@ async def gizai_chat_completions(
         )
 
     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_response(request, model), media_type="text/event-stream")
+        return StreamingResponse(process_gizai_stream_response(request, model), media_type="text/event-stream")
     else:
         logger.info("Non-streaming response")
-        return await process_gizai_response(request, model)
+        response = await process_gizai_non_stream_response(request, model)
+        return response
 
 @router.route('/')
 @router.route('/healthz')