Update api/utils.py

api/utils.py

@@ -9,11 +9,11 @@ import httpx
 from fastapi import HTTPException
 from api.config import (
     MODEL_MAPPING,
-    get_headers_api_chat,
     BASE_URL,
     AGENT_MODE,
     TRENDING_AGENT_MODE,
     MODEL_PREFIXES,
+    headers,
 )
 from api.models import ChatRequest
 from api.logger import setup_logger
@@ -23,7 +23,6 @@ from api import validate
 
 logger = setup_logger(__name__)
 
-
 # Helper function to create chat completion data
 def create_chat_completion_data(
     content: str,
@@ -46,7 +45,6 @@ def create_chat_completion_data(
         "usage": None,
     }
 
-
 # Function to convert message to dictionary format, ensuring base64 data and optional model prefix
 def message_to_dict(message, model_prefix: Optional[str] = None):
     content = (
@@ -73,7 +71,6 @@ def message_to_dict(message, model_prefix: Optional[str] = None):
         }
     return {"role": message.role, "content": content}
 
-
 # Function to strip model prefix from content if present
 def strip_model_prefix(content: str, model_prefix: Optional[str] = None) -> str:
     """Remove the model prefix from the response content if present."""
@@ -82,26 +79,14 @@ def strip_model_prefix(content: str, model_prefix: Optional[str] = None) -> str:
         return content[len(model_prefix):].strip()
     return content
 
-
-# Function to get the referer URL
-def get_referer_url(model: str) -> str:
-    """Generate the referer URL based on the model."""
-    return BASE_URL
-
-
-# Process streaming response with headers from config.py
+# Process streaming response
 async def process_streaming_response(request: ChatRequest):
-    referer_url = get_referer_url(request.model)
-    logger.info(
-        f"Model: {request.model} - URL: {referer_url}"
-    )
+    logger.info(f"Model: {request.model}")
 
     agent_mode = AGENT_MODE.get(request.model, {})
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
     model_prefix = MODEL_PREFIXES.get(request.model, "")
 
-    headers_api_chat = get_headers_api_chat(referer_url)
-
     if request.model == 'o1-preview':
         delay_seconds = random.randint(1, 60)
         logger.info(
@@ -139,7 +124,7 @@ async def process_streaming_response(request: ChatRequest):
             async with client.stream(
                 "POST",
                 f"{BASE_URL}/api/chat",
-                headers=headers_api_chat,
+                headers=headers,
                 json=json_data,
                 timeout=100,
             ) as response:
@@ -175,20 +160,14 @@ async def process_streaming_response(request: ChatRequest):
             logger.error(f"Error occurred during request: {e}")
             raise HTTPException(status_code=500, detail=str(e))
 
-
-# Process non-streaming response with headers from config.py
+# Process non-streaming response
 async def process_non_streaming_response(request: ChatRequest):
-    referer_url = get_referer_url(request.model)
-    logger.info(
-        f"Model: {request.model} - URL: {referer_url}"
-    )
+    logger.info(f"Model: {request.model}")
 
     agent_mode = AGENT_MODE.get(request.model, {})
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
     model_prefix = MODEL_PREFIXES.get(request.model, "")
 
-    headers_api_chat = get_headers_api_chat(referer_url)
-
     if request.model == 'o1-preview':
         delay_seconds = random.randint(20, 60)
         logger.info(
@@ -226,7 +205,7 @@ async def process_non_streaming_response(request: ChatRequest):
         try:
             response = await client.post(
                 url=f"{BASE_URL}/api/chat",
-                headers=headers_api_chat,
+                headers=headers,
                 json=json_data,
             )
             response.raise_for_status()