Update api/utils.py

api/utils.py

@@ -82,7 +82,8 @@ 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}")
+    chat_url = f"/chat/{chat_id}?model={request.model}"
+    logger.info(f"Generated Chat ID: {chat_id} - URL: {chat_url}")
     
     agent_mode = AGENT_MODE.get(request.model, {})
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
@@ -155,14 +156,14 @@ 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}")
+    chat_url = f"/chat/{chat_id}?model={request.model}"
+    logger.info(f"Generated Chat ID: {chat_id} - URL: {chat_url}")
     
     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}"
-    chat_url = f"{BASE_URL}/chat/{chat_id}?model={request.model}"
 
     # Generate headers for API chat request and chat request with dynamic Referer
     headers_api_chat = get_headers_api_chat(referer_url)
@@ -232,4 +233,4 @@ async def process_non_streaming_response(request: ChatRequest):
             }
         ],
         "usage": None,
-    }
+    }