Update api/utils.py

api/utils.py

@@ -1,3 +1,5 @@
+# api/utils.py
+
 from datetime import datetime
 import json
 import uuid
@@ -13,7 +15,7 @@ from api.config import (
     AGENT_MODE,
     TRENDING_AGENT_MODE,
     MODEL_PREFIXES,
-    headers,
+    api_headers,
 )
 from api.models import ChatRequest
 from api.logger import setup_logger
@@ -124,7 +126,7 @@ async def process_streaming_response(request: ChatRequest):
             async with client.stream(
                 "POST",
                 f"{BASE_URL}/api/chat",
-                headers=headers,
+                headers=api_headers,
                 json=json_data,
                 timeout=100,
             ) as response:
@@ -135,7 +137,7 @@ async def process_streaming_response(request: ChatRequest):
                         if "https://www.blackbox.ai" in line:
                             # Refresh hid and inform the user
                             validate.getHid(True)
-                            content = "The HID has been refreshed; please try again.\n"
+                            content = "hid已刷新,重新对话即可"
                             yield (
                                 f"data: {json.dumps(create_chat_completion_data(content, request.model, int(datetime.now().timestamp())))}\n\n"
                             )
@@ -205,7 +207,7 @@ async def process_non_streaming_response(request: ChatRequest):
         try:
             response = await client.post(
                 url=f"{BASE_URL}/api/chat",
-                headers=headers,
+                headers=api_headers,
                 json=json_data,
             )
             response.raise_for_status()
@@ -220,7 +222,7 @@ async def process_non_streaming_response(request: ChatRequest):
     if "https://www.blackbox.ai" in full_response:
         # Refresh hid and inform the user
         validate.getHid(True)
-        full_response = "The HID has been refreshed; please try again."
+        full_response = "hid已刷新，重新对话即可"
     if full_response.startswith("$@$v=undefined-rv1$@$"):
         full_response = full_response[21:]