Update api/utils.py

api/utils.py

@@ -67,6 +67,9 @@ def message_to_dict(message, model_prefix: Optional[str] = None):
             content = f"{model_prefix} {content}"
         return {"role": message.role, "content": content}
 
+def generate_chat_id() -> str:
+    return str(uuid.uuid4())
+
 async def process_streaming_response(request: ChatRequest):
     agent_mode = AGENT_MODE.get(request.model, {})
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
@@ -78,6 +81,9 @@ async def process_streaming_response(request: ChatRequest):
     user_selected_model = MODEL_MAPPING.get(request.model, request.model)
     logger.info(f"User selected model: {user_selected_model}")
 
+    chat_id = generate_chat_id()
+    logger.info(f"Generated chat_id: {chat_id} for request.")
+
     json_data = {
         "messages": [message_to_dict(msg, model_prefix=model_prefix) for msg in request.messages],
         "previewToken": None,
@@ -97,6 +103,7 @@ async def process_streaming_response(request: ChatRequest):
         "clickedForceWebSearch": False,
         "visitFromDelta": False,
         "mobileClient": False,
+        "id": chat_id,
         "userSelectedModel": user_selected_model,
         "validated": hid
     }
@@ -117,7 +124,7 @@ async def process_streaming_response(request: ChatRequest):
                         content = line
                         if "https://www.blackbox.ai" in content:
                             logger.warning("Invalid hid detected, refreshing hid")
-                            hid = validate.getHid()
+                            validate.getHid(force_refresh=True)
                             content = "hid has been refreshed, please retry"
                             logger.info(f"hid refreshed due to content: {content}")
                             yield f"data: {json.dumps(create_chat_completion_data(content, request.model, timestamp))}\n\n"
@@ -146,6 +153,9 @@ async def process_non_streaming_response(request: ChatRequest):
     user_selected_model = MODEL_MAPPING.get(request.model, request.model)
     logger.info(f"User selected model: {user_selected_model}")
 
+    chat_id = generate_chat_id()
+    logger.info(f"Generated chat_id: {chat_id} for request.")
+
     json_data = {
         "messages": [message_to_dict(msg, model_prefix=model_prefix) for msg in request.messages],
         "previewToken": None,
@@ -165,6 +175,7 @@ async def process_non_streaming_response(request: ChatRequest):
         "clickedForceWebSearch": False,
         "visitFromDelta": False,
         "mobileClient": False,
+        "id": chat_id,
         "userSelectedModel": user_selected_model,
         "validated": hid
     }
@@ -186,7 +197,7 @@ async def process_non_streaming_response(request: ChatRequest):
 
     if "https://www.blackbox.ai" in full_response:
         logger.warning("Invalid hid detected in response, refreshing hid")
-        hid = validate.getHid()
+        validate.getHid(force_refresh=True)
         full_response = "hid has been refreshed, please retry"
         logger.info("hid refreshed due to response content")