Update api/utils.py

api/utils.py

@@ -20,7 +20,7 @@ from api.config import (
 )
 from api.models import ChatRequest
 from api.logger import (
-    log_generated_chat_id,
+    log_generated_chat_id_with_referer,
     log_model_delay,
     log_http_error,
     log_request_error,
@@ -85,15 +85,17 @@ 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()
-    log_generated_chat_id(chat_id, request.model)
+    
+    # Get referer URL only if model is in MODEL_REFERERS
+    referer_path = MODEL_REFERERS.get(request.model, "")
+    referer_url = f"{BASE_URL}{referer_path}" if referer_path else BASE_URL
+
+    # Log with chat ID, model, and referer URL (if present)
+    log_generated_chat_id_with_referer(chat_id, request.model, referer_url)
 
     agent_mode = AGENT_MODE.get(request.model, {})
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
     model_prefix = MODEL_PREFIXES.get(request.model, "")
-    
-    # Check if the model is in MODEL_REFERERS, else no referer path
-    referer_path = MODEL_REFERERS.get(request.model, "")
-    referer_url = f"{BASE_URL}{referer_path}" if referer_path else BASE_URL
 
     # Generate headers for API chat request with dynamic Referer
     headers_api_chat = get_headers_api_chat(referer_url)
@@ -160,17 +162,19 @@ 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()
-    log_generated_chat_id(chat_id, 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, "")
 
-    # Check if the model is in MODEL_REFERERS, else no referer path
+    # Get referer URL only if model is in MODEL_REFERERS
     referer_path = MODEL_REFERERS.get(request.model, "")
     referer_url = f"{BASE_URL}{referer_path}" if referer_path else BASE_URL
     chat_url = f"{referer_url}/chat/{chat_id}?model={request.model}"
 
+    # Log with chat ID, model, and referer URL (if present)
+    log_generated_chat_id_with_referer(chat_id, request.model, 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, "")
+
     # Generate headers for API chat request and chat request with dynamic Referer
     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([""]))