Update api/utils.py

api/utils.py

@@ -9,8 +9,7 @@ import httpx
 from fastapi import HTTPException
 from api.config import (
     MODEL_MAPPING,
-    get_headers_api_chat,
-    get_headers_chat,
+    common_headers,  # Import common_headers directly
     BASE_URL,
     AGENT_MODE,
     TRENDING_AGENT_MODE,
@@ -68,7 +67,7 @@ def strip_model_prefix(content: str, model_prefix: Optional[str] = None) -> str:
         return content[len(model_prefix):].strip()
     return content
 
-# Process streaming response with headers from config.py
+# Process streaming response without header functions
 async def process_streaming_response(request: ChatRequest):
     # Generate a unique ID for this request
     request_id = f"chatcmpl-{uuid.uuid4()}"
@@ -78,8 +77,8 @@ async def process_streaming_response(request: ChatRequest):
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
     model_prefix = MODEL_PREFIXES.get(request.model, "")
 
-    # Adjust headers_api_chat since referer_url is removed
-    headers_api_chat = get_headers_api_chat()
+    # Define headers directly
+    headers_api_chat = {**common_headers, 'Content-Type': 'application/json'}
 
     if request.model == 'o1-preview':
         delay_seconds = random.randint(1, 60)
@@ -144,7 +143,7 @@ async def process_streaming_response(request: ChatRequest):
             logger.error(f"Error occurred during request for Request ID {request_id}: {e}")
             raise HTTPException(status_code=500, detail=str(e))
 
-# Process non-streaming response with headers from config.py
+# Process non-streaming response without header functions
 async def process_non_streaming_response(request: ChatRequest):
     # Generate a unique ID for this request
     request_id = f"chatcmpl-{uuid.uuid4()}"
@@ -154,9 +153,16 @@ async def process_non_streaming_response(request: ChatRequest):
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
     model_prefix = MODEL_PREFIXES.get(request.model, "")
 
-    # Adjust headers_api_chat and headers_chat since referer_url is removed
-    headers_api_chat = get_headers_api_chat()
-    headers_chat = get_headers_chat(next_action=str(uuid.uuid4()), next_router_state_tree=json.dumps([""]))
+    # Define headers directly
+    headers_api_chat = {**common_headers, 'Content-Type': 'application/json'}
+    headers_chat = {
+        **common_headers,
+        'Accept': 'text/x-component',
+        'Content-Type': 'text/plain;charset=UTF-8',
+        'next-action': str(uuid.uuid4()),
+        'next-router-state-tree': json.dumps([""]),
+        'next-url': '/',
+    }
 
     if request.model == 'o1-preview':
         delay_seconds = random.randint(20, 60)
@@ -208,7 +214,7 @@ async def process_non_streaming_response(request: ChatRequest):
         except httpx.RequestError as e:
             logger.error(f"Error occurred during request for Request ID {request_id}: {e}")
             raise HTTPException(status_code=500, detail=str(e))
-    
+
     if full_response.startswith("$@$v=undefined-rv1$@$"):
         full_response = full_response[21:]