Update api/utils.py

api/utils.py

@@ -134,12 +134,15 @@ async def process_streaming_response(request: ChatRequest):
                     timestamp = int(datetime.now().timestamp())
                     if line:
                         content = line
-                        if "https://www.blackbox.ai" in content:
+                        logger.debug(f"Received content: {content}")
+                        # Modify the condition to detect specific error message
+                        if "Invalid or expired 'hid'" in content:
+                            logger.warning("Invalid or expired 'hid' detected. Refreshing 'hid'.")
                             validate.getHid(True)
                             content = "hid已刷新,重新对话即可\n"
                             yield f"data: {json.dumps(create_chat_completion_data(content, request.model, timestamp))}\n\n"
                             break
-                        if content.startswith("$@$v=undefined-rv1$@$"):
+                        elif content.startswith("$@$v=undefined-rv1$@$"):
                             content = content[21:]
                         cleaned_content = strip_model_prefix(content, model_prefix)
                         yield f"data: {json.dumps(create_chat_completion_data(cleaned_content, request.model, timestamp))}\n\n"
@@ -198,21 +201,28 @@ async def process_non_streaming_response(request: ChatRequest):
     full_response = ""
     async with httpx.AsyncClient() as client:
         try:
-            async with client.stream(
-                method="POST", url=f"{BASE_URL}/api/chat", headers=headers_api_chat, json=json_data
-            ) as response:
-                response.raise_for_status()
-                async for chunk in response.aiter_text():
-                    full_response += chunk
+            response = await client.post(
+                f"{BASE_URL}/api/chat",
+                headers=headers_api_chat,
+                json=json_data,
+                timeout=100,
+            )
+            response.raise_for_status()
+            full_response = response.text
+            logger.debug(f"Full response received: {full_response}")
         except httpx.HTTPStatusError as e:
             logger.error(f"HTTP error occurred for Chat ID {chat_id}: {e}")
             raise HTTPException(status_code=e.response.status_code, detail=str(e))
         except httpx.RequestError as e:
             logger.error(f"Error occurred during request for Chat ID {chat_id}: {e}")
             raise HTTPException(status_code=500, detail=str(e))
-    if "https://www.blackbox.ai" in full_response:
+
+    # Modify the condition to detect specific error message
+    if "Invalid or expired 'hid'" in full_response:
+        logger.warning("Invalid or expired 'hid' detected. Refreshing 'hid'.")
         validate.getHid(True)
         full_response = "hid已刷新，重新对话即可"
+
     if full_response.startswith("$@$v=undefined-rv1$@$"):
         full_response = full_response[21:]