Update api/utils.py

api/utils.py

@@ -67,13 +67,6 @@ def message_to_dict(message, model_prefix: Optional[str] = None):
             content = f"{model_prefix} {content}"
         return {"role": message.role, "content": content}
 
-def strip_model_prefix(content: str, model_prefix: Optional[str] = None) -> str:
-    """Remove the model prefix from the response content if present."""
-    if model_prefix and content.startswith(model_prefix):
-        logger.debug(f"Stripping prefix '{model_prefix}' from content.")
-        return content[len(model_prefix):].lstrip()
-    return content
-
 async def process_streaming_response(request: ChatRequest):
     agent_mode = AGENT_MODE.get(request.model, {})
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
@@ -118,18 +111,17 @@ async def process_streaming_response(request: ChatRequest):
                 async for line in response.aiter_lines():
                     timestamp = int(datetime.now().timestamp())
                     if line:
-                        content = line.strip()  # Remove leading/trailing whitespace
+                        content = line  # Do not modify content
                         if "https://www.blackbox.ai" in content:
                             validate.getHid(True)
-                            content = "The HID has been refreshed; please try again.\n"
+                            content = "hid已刷新,重新对话即可"
                             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$@$"):
-                            content = content[21:].lstrip()
-                        # Strip model prefix from content
-                        cleaned_content = strip_model_prefix(content, model_prefix)
-                        yield f"data: {json.dumps(create_chat_completion_data(cleaned_content, request.model, timestamp))}\n\n"
+                            content = content[21:]
+                        # Do not strip model prefix or modify content
+                        yield f"data: {json.dumps(create_chat_completion_data(content, request.model, timestamp))}\n\n"
 
                 yield f"data: {json.dumps(create_chat_completion_data('', request.model, timestamp, 'stop'))}\n\n"
                 yield "data: [DONE]\n\n"
@@ -188,13 +180,12 @@ async def process_non_streaming_response(request: ChatRequest):
 
     if "https://www.blackbox.ai" in full_response:
         validate.getHid(True)
-        full_response = "The HID has been refreshed; please try again."
-        logger.info("HID refreshed due to the content of the response.")
+        full_response = "hid已刷新，重新对话即可"
+        logger.info("hid refreshed due to response content")
 
     if full_response.startswith("$@$v=undefined-rv1$@$"):
-        full_response = full_response[21:].lstrip()
-    # Strip model prefix from full_response
-    cleaned_full_response = strip_model_prefix(full_response.strip(), model_prefix)
+        full_response = full_response[21:]
+    # Do not strip model prefix or modify content
     return {
         "id": f"chatcmpl-{uuid.uuid4()}",
         "object": "chat.completion",
@@ -203,7 +194,7 @@ async def process_non_streaming_response(request: ChatRequest):
         "choices": [
             {
                 "index": 0,
-                "message": {"role": "assistant", "content": cleaned_full_response},
+                "message": {"role": "assistant", "content": full_response},
                 "finish_reason": "stop",
             }
         ],