Update api/utils.py

api/utils.py

@@ -20,8 +20,7 @@ logger = setup_logger(__name__)
 def create_chat_completion_data(
     content: str, model: str, timestamp: int, finish_reason: Optional[str] = None
 ) -> Dict[str, Any]:
-    # Clean the content to remove extra spaces
-    cleaned_content = '\n'.join([line.rstrip() for line in content.splitlines() if line.strip() != ''])
+    # Removed the content cleaning function
     return {
         "id": f"chatcmpl-{uuid.uuid4()}",
         "object": "chat.completion.chunk",
@@ -30,7 +29,7 @@ def create_chat_completion_data(
         "choices": [
             {
                 "index": 0,
-                "delta": {"content": cleaned_content, "role": "assistant"},
+                "delta": {"content": content, "role": "assistant"},
                 "finish_reason": finish_reason,
             }
         ],
@@ -85,7 +84,7 @@ async def process_streaming_response(request: ChatRequest):
         try:
             async with client.stream(
                 "POST",
-                f"{BASE_URL}/api/chat",  # BASE_URL is now defined
+                f"{BASE_URL}/api/chat",
                 headers=headers,
                 json=json_data,
                 timeout=100,
@@ -94,10 +93,7 @@ async def process_streaming_response(request: ChatRequest):
                 async for line in response.aiter_lines():
                     timestamp = int(datetime.now().timestamp())
                     if line:
-                        # Clean the content to remove extra spaces
-                        content = line.strip()
-                        content = '\n'.join([l.rstrip() for l in content.splitlines() if l.strip() != ''])
-                        content += "\n"
+                        content = line + "\n"
                         if content.startswith("$@$v=undefined-rv1$@$"):
                             yield f"data: {json.dumps(create_chat_completion_data(content[21:], request.model, timestamp))}\n\n"
                         else:
@@ -154,10 +150,6 @@ async def process_non_streaming_response(request: ChatRequest):
     if full_response.startswith("$@$v=undefined-rv1$@$"):
         full_response = full_response[21:]
 
-    # Clean the content to remove extra spaces
-    full_response = full_response.strip()
-    full_response = '\n'.join([line.rstrip() for line in full_response.splitlines() if line.strip() != ''])
-
     return {
         "id": f"chatcmpl-{uuid.uuid4()}",
         "object": "chat.completion",