Update api/utils.py

api/utils.py

@@ -15,7 +15,8 @@ from api.config import (
     AGENT_MODE,
     TRENDING_AGENT_MODE,
     MODEL_PREFIXES,
-    MODEL_REFERERS
+    MODEL_REFERERS,
+    common_headers
 )
 from api.models import ChatRequest
 from api.logger import setup_logger
@@ -80,19 +81,23 @@ def strip_model_prefix(content: str, model_prefix: Optional[str] = None) -> str:
 # Process streaming response with headers from config.py
 async def process_streaming_response(request: ChatRequest):
     chat_id = generate_chat_id()
+    logger.info(f"Generated Chat ID: {chat_id}")
+    
     agent_mode = AGENT_MODE.get(request.model, {})
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
     model_prefix = MODEL_PREFIXES.get(request.model, "")
     referer_path = MODEL_REFERERS.get(request.model, f"/?model={request.model}")
     referer_url = f"{BASE_URL}{referer_path}"
 
-    # Generate headers for API chat request
+    # 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)
 
     # Introduce delay for 'o1-preview' model
     if request.model == 'o1-preview':
-        delay_seconds = random.randint(1, 60)
-        logger.info(f"Introducing a delay of {delay_seconds} seconds for model 'o1-preview'")
+        delay_seconds = random.randint(20, 60)
+        logger.info(f"Introducing a delay of {delay_seconds} seconds for model 'o1-preview' (Chat ID: {chat_id})")
         await asyncio.sleep(delay_seconds)
 
     json_data = {
@@ -151,6 +156,8 @@ async def process_streaming_response(request: ChatRequest):
 # Process non-streaming response with headers from config.py
 async def process_non_streaming_response(request: ChatRequest):
     chat_id = generate_chat_id()
+    logger.info(f"Generated Chat ID: {chat_id}")
+    
     agent_mode = AGENT_MODE.get(request.model, {})
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
     model_prefix = MODEL_PREFIXES.get(request.model, "")
@@ -158,10 +165,18 @@ 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
+    # 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([""]))
 
+    # Introduce delay for 'o1-preview' model
+    if request.model == 'o1-preview':
+        delay_seconds = random.randint(20, 60)
+        logger.info(f"Introducing a delay of {delay_seconds} seconds for model 'o1-preview' (Chat ID: {chat_id})")
+        await asyncio.sleep(delay_seconds)
+
     json_data = {
         "id": chat_id,
         "messages": [message_to_dict(msg, model_prefix=model_prefix) for msg in request.messages],