Update api/utils.py

api/utils.py

@@ -79,6 +79,9 @@ async def process_streaming_response(request: ChatRequest):
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
     model_prefix = MODEL_PREFIXES.get(request.model, "")
 
+    hid = validate.getHid()
+    logger.info(f"Using hid: {hid} for model: {request.model}")
+
     json_data = {
         "messages": [message_to_dict(msg, model_prefix=model_prefix) for msg in request.messages],
         "previewToken": None,
@@ -99,7 +102,7 @@ async def process_streaming_response(request: ChatRequest):
         "visitFromDelta": False,
         "mobileClient": False,
         "userSelectedModel": MODEL_MAPPING.get(request.model, request.model),
-        "validated": validate.getHid()
+        "validated": hid
     }
 
     async with httpx.AsyncClient() as client:
@@ -119,6 +122,7 @@ async def process_streaming_response(request: ChatRequest):
                         if "https://www.blackbox.ai" in content:
                             validate.getHid(True)
                             content = "hid已刷新,重新对话即可\n"
+                            logger.info(f"hid refreshed due to content: {content}")
                             yield f"data: {json.dumps(create_chat_completion_data(content, request.model, timestamp))}\n\n"
                             break
                         if content.startswith("$@$v=undefined-rv1$@$"):
@@ -141,6 +145,9 @@ async def process_non_streaming_response(request: ChatRequest):
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
     model_prefix = MODEL_PREFIXES.get(request.model, "")
 
+    hid = validate.getHid()
+    logger.info(f"Using hid: {hid} for model: {request.model}")
+
     json_data = {
         "messages": [message_to_dict(msg, model_prefix=model_prefix) for msg in request.messages],
         "previewToken": None,
@@ -161,7 +168,7 @@ async def process_non_streaming_response(request: ChatRequest):
         "visitFromDelta": False,
         "mobileClient": False,
         "userSelectedModel": MODEL_MAPPING.get(request.model, request.model),
-        "validated": validate.getHid()
+        "validated": hid
     }
     full_response = ""
     async with httpx.AsyncClient() as client:
@@ -182,6 +189,8 @@ async def process_non_streaming_response(request: ChatRequest):
     if "https://www.blackbox.ai" in full_response:
         validate.getHid(True)
         full_response = "hid已刷新，重新对话即可"
+        logger.info("hid refreshed due to response content")
+
     if full_response.startswith("$@$v=undefined-rv1$@$"):
         full_response = full_response[21:]
     # Strip model prefix from full_response