Update api/provider/gizai.py

api/provider/gizai.py

@@ -1,38 +1,17 @@
 import uuid
 from datetime import datetime
 import json
-from typing import Any, Dict, Optional
+from typing import Any, Dict
 
 import httpx
 from fastapi import HTTPException
 from api.models import ChatRequest
 from api.logger import setup_logger
+from api.config import MODEL_MAPPING, GIZAI_API_ENDPOINT, GIZAI_HEADERS
 
 logger = setup_logger(__name__)
 
-# Base URL for giz.ai
-GIZAI_BASE_URL = "https://app.giz.ai"
-GIZAI_API_ENDPOINT = f"{GIZAI_BASE_URL}/api/data/users/inferenceServer.infer"
-
-# Headers for giz.ai
-GIZAI_HEADERS = {
-    'Accept': 'application/json, text/plain, */*',
-    'Accept-Language': 'en-US,en;q=0.9',
-    'Cache-Control': 'no-cache',
-    'Connection': 'keep-alive',
-    'Content-Type': 'application/json',
-    'Origin': 'https://app.giz.ai',
-    'Pragma': 'no-cache',
-    'Sec-Fetch-Dest': 'empty',
-    'Sec-Fetch-Mode': 'cors',
-    'Sec-Fetch-Site': 'same-origin',
-    'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.0.0 Safari/537.36',
-    'sec-ch-ua': '"Not?A_Brand";v="99", "Chromium";v="130"',
-    'sec-ch-ua-mobile': '?0',
-    'sec-ch-ua-platform': '"Linux"'
-}
-
-# List of models supported by giz.ai
+# List of models supported by GizAI
 GIZAI_CHAT_MODELS = [
     'chat-gemini-flash',
     'chat-gemini-pro',
@@ -73,6 +52,7 @@ GIZAI_MODEL_ALIASES = {
 }
 
 def get_gizai_model(model: str) -> str:
+    model = MODEL_MAPPING.get(model, model)
     if model in GIZAI_MODELS:
         return model
     elif model in GIZAI_MODEL_ALIASES:
@@ -85,8 +65,7 @@ def is_image_model(model: str) -> bool:
     return model in GIZAI_IMAGE_MODELS
 
 async def process_streaming_response(request: ChatRequest):
-    # giz.ai does not support streaming
-    # So we can raise an error or process as non-streaming
+    # GizAI does not support streaming; handle as non-streaming
     return await process_non_streaming_response(request)
 
 async def process_non_streaming_response(request: ChatRequest):