Update api/routes.py

api/routes.py

@@ -1,11 +1,13 @@
 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.models import ChatRequest
-from api.utils import process_non_streaming_response, process_streaming_response
-from api.logger import setup_logger
+from fastapi import APIRouter, Depends, HTTPException, Request
+from fastapi.responses import StreamingResponse, JSONResponse
+from typing import Optional
+
+from .auth import verify_app_secret
+from .models import ChatRequest, ImageResponse
+from .utils import strip_model_prefix
+from .gizai import GizAI
+from .logger import setup_logger
 
 logger = setup_logger(__name__)
 
@@ -49,6 +51,47 @@ async def chat_completions(
         logger.info("Non-streaming response")
         return await process_non_streaming_response(request)
 
+
+# GizAI Routes
+gizai_router = APIRouter(prefix="/gizai", tags=["GizAI"])
+
+@gizai_router.options("/v1/chat/completions")
+async def gizai_chat_completions_options():
+    return JSONResponse(
+        status_code=200,
+        headers={
+            "Access-Control-Allow-Origin": "*",
+            "Access-Control-Allow-Methods": "POST, OPTIONS",
+            "Access-Control-Allow-Headers": "Content-Type, Authorization",
+        },
+    )
+
+@gizai_router.post("/v1/chat/completions")
+async def gizai_chat_completions(
+    request: ChatRequest, app_secret: str = Depends(verify_app_secret)
+):
+    logger.info("Entering GizAI chat_completions route")
+    logger.info(f"Processing chat completion request for model: {request.model}")
+
+    # Validate model
+    model = GizAI.get_model(request.model)
+    if model not in GizAI.models:
+        raise HTTPException(
+            status_code=400,
+            detail=f"Model {request.model} is not supported by GizAI. Supported models are: {', '.join(GizAI.models)}",
+        )
+
+    try:
+        async_generator = GizAI.create_async_generator(model=model, messages=request.messages)
+        return StreamingResponse(async_generator, media_type="application/json")
+    except Exception as e:
+        logger.error(f"Error in GizAI chat_completions: {e}", exc_info=True)
+        raise HTTPException(status_code=500, detail=str(e))
+
+# Include GizAI router in the main router
+router.include_router(gizai_router)
+
+
 @router.route('/')
 @router.route('/healthz')
 @router.route('/ready')
@@ -57,3 +100,5 @@ async def chat_completions(
 @router.get("/health")
 def health_check(request: Request):
     return Response(content=json.dumps({"status": "ok"}), media_type="application/json")
+
+