Update api/utils.py

api/utils.py

@@ -1,5 +1,4 @@
 from datetime import datetime
-import json
 import uuid
 import asyncio
 import random
@@ -47,7 +46,6 @@ def message_to_dict(message, model_prefix: Optional[str] = None):
     if model_prefix:
         content = f"{model_prefix} {content}"
     if isinstance(message.content, list) and len(message.content) == 2 and "image_url" in message.content[1]:
-        # Ensure base64 images are always included for all models
         return {
             "role": message.role,
             "content": content,
@@ -68,7 +66,7 @@ async def process_streaming_response(request: ChatRequest):
     model_prefix = MODEL_PREFIXES.get(request.model, "")
 
     headers_api_chat = get_headers_api_chat(BASE_URL)
-    validated_token = validate.getHid()  # Get the validated token from validate.py
+    validated_token = validate.getHid()
     logger.info(f"Retrieved validated token: {validated_token}")
 
     if request.model == 'o1-preview':
@@ -112,10 +110,9 @@ async def process_streaming_response(request: ChatRequest):
                 async for line in response.aiter_lines():
                     timestamp = int(datetime.now().timestamp())
                     if line:
-                        # Directly yield each line without additional processing
-                        yield f"data: {json.dumps(create_chat_completion_data(line, request.model, timestamp))}\n\n"
+                        # Stream line directly without additional processing or transformation
+                        yield f"data: {line}\n\n"
 
-                yield f"data: {json.dumps(create_chat_completion_data('', request.model, timestamp, 'stop'))}\n\n"
                 yield "data: [DONE]\n\n"
         except httpx.HTTPStatusError as e:
             logger.error(f"HTTP error occurred: {e}")
@@ -134,7 +131,7 @@ async def process_non_streaming_response(request: ChatRequest):
 
     headers_api_chat = get_headers_api_chat(BASE_URL)
     headers_chat = get_headers_chat(BASE_URL, next_action=str(uuid.uuid4()), next_router_state_tree=json.dumps([""]))
-    validated_token = validate.getHid()  # Get the validated token from validate.py
+    validated_token = validate.getHid()
 
     if request.model == 'o1-preview':
         delay_seconds = random.randint(20, 60)