Update api/utils.py

api/utils.py

@@ -10,8 +10,6 @@ import httpx
 from fastapi import HTTPException
 from api.config import (
     MODEL_MAPPING,
-    get_headers_api_chat,
-    get_headers_chat,
     BASE_URL,
     AGENT_MODE,
     TRENDING_AGENT_MODE,
@@ -80,7 +78,7 @@ def strip_model_prefix(content: str, model_prefix: Optional[str] = None) -> str:
     logger.debug("No prefix to strip from content.")
     return content
 
-# Process streaming response with headers from config.py
+# Process streaming response with combined headers
 async def process_streaming_response(request: ChatRequest):
     chat_id = generate_chat_id()
     logger.info(f"Generated Chat ID: {chat_id}")
@@ -91,10 +89,12 @@ async def process_streaming_response(request: ChatRequest):
     referer_path = MODEL_REFERERS.get(request.model, f"/?model={request.model}")
     referer_url = f"{BASE_URL}{referer_path}"
 
-    # Generate headers for API chat request with dynamic Referer
-    dynamic_headers = common_headers.copy()
-    dynamic_headers['Referer'] = referer_url
-    headers_api_chat = get_headers_api_chat(referer_url)
+    # Combined headers for API chat
+    headers_api_chat = {
+        **common_headers,
+        'Content-Type': 'application/json',
+        'Referer': referer_url
+    }
 
     # Introduce delay for 'o1-preview' model
     if request.model == 'o1-preview':
@@ -155,7 +155,7 @@ async def process_streaming_response(request: ChatRequest):
             logger.error(f"Error occurred during request for Chat ID {chat_id}: {e}")
             raise HTTPException(status_code=500, detail=str(e))
 
-# Process non-streaming response with headers from config.py
+# Process non-streaming response with combined headers
 async def process_non_streaming_response(request: ChatRequest):
     chat_id = generate_chat_id()
     logger.info(f"Generated Chat ID: {chat_id}")
@@ -167,11 +167,21 @@ async def process_non_streaming_response(request: ChatRequest):
     referer_url = f"{BASE_URL}{referer_path}"
     chat_url = f"{BASE_URL}/chat/{chat_id}?model={request.model}"
 
-    # Generate headers for API chat request and chat request with dynamic Referer
-    dynamic_headers = common_headers.copy()
-    dynamic_headers['Referer'] = referer_url
-    headers_api_chat = get_headers_api_chat(referer_url)
-    headers_chat = get_headers_chat(chat_url, next_action=str(uuid.uuid4()), next_router_state_tree=json.dumps([""]))
+    # Combined headers for API chat and chat request
+    headers_api_chat = {
+        **common_headers,
+        'Content-Type': 'application/json',
+        'Referer': referer_url
+    }
+    headers_chat = {
+        **common_headers,
+        'Accept': 'text/x-component',
+        'Content-Type': 'text/plain;charset=UTF-8',
+        'Referer': chat_url,
+        'next-action': str(uuid.uuid4()),
+        'next-router-state-tree': json.dumps([""]),
+        'next-url': '/'
+    }
 
     # Introduce delay for 'o1-preview' model
     if request.model == 'o1-preview':