Update api/utils.py

api/utils.py

@@ -71,7 +71,7 @@ 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):].strip()
+        return content[len(model_prefix):].lstrip()
     return content
 
 async def process_streaming_response(request: ChatRequest):
@@ -118,15 +118,15 @@ async def process_streaming_response(request: ChatRequest):
                 async for line in response.aiter_lines():
                     timestamp = int(datetime.now().timestamp())
                     if line:
-                        content = line + "\n"
+                        content = line.strip()  # Remove leading/trailing whitespace
                         if "https://www.blackbox.ai" in content:
                             validate.getHid(True)
-                            content = "The HID has been refreshed; please try again.\n"
+                            content = "The HID has been refreshed; please try again."
                             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:]
+                            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"
@@ -192,9 +192,9 @@ async def process_non_streaming_response(request: ChatRequest):
         logger.info("HID refreshed due to the content of the response.")
 
     if full_response.startswith("$@$v=undefined-rv1$@$"):
-        full_response = full_response[21:]
+        full_response = full_response[21:].lstrip()
     # Strip model prefix from full_response
-    cleaned_full_response = strip_model_prefix(full_response, model_prefix)
+    cleaned_full_response = strip_model_prefix(full_response.strip(), model_prefix)
     return {
         "id": f"chatcmpl-{uuid.uuid4()}",
         "object": "chat.completion",