Update api/utils.py

api/utils.py

@@ -46,20 +46,14 @@ def create_chat_completion_data(
         "usage": None,
     }
 
-# Function to convert message to dictionary format, including data and id if present
+# Function to convert message to dictionary format, ensuring base64 data and optional model prefix
 def message_to_dict(message, model_prefix: Optional[str] = None):
     content = message.content if isinstance(message.content, str) else message.content[0]["text"]
     if model_prefix:
         content = f"{model_prefix} {content}"
-
     message_dict = {"role": message.role, "content": content}
-
-    if hasattr(message, 'id') and message.id:
-        message_dict['id'] = message.id
-
     if hasattr(message, 'data') and message.data:
         message_dict['data'] = message.data
-
     return message_dict
 
 # Function to strip model prefix from content if present
@@ -94,19 +88,14 @@ async def process_streaming_response(request: ChatRequest):
         logger.error("Failed to retrieve h-value for validation.")
         raise HTTPException(status_code=500, detail="Validation failed due to missing h-value.")
 
-    # Determine if images are included in messages
-    code_model_mode = any(
-        hasattr(msg, 'data') and msg.data and 'imagesData' in msg.data for msg in request.messages
-    )
-
     json_data = {
         "agentMode": agent_mode,
         "clickedAnswer2": False,
         "clickedAnswer3": False,
         "clickedForceWebSearch": False,
-        "codeModelMode": code_model_mode,
+        "codeModelMode": True,
         "githubToken": None,
-        "id": request_id,
+        "id": None,  # Using request_id instead of chat_id
         "isChromeExt": False,
         "isMicMode": False,
         "maxTokens": request.max_tokens,
@@ -122,7 +111,7 @@ async def process_streaming_response(request: ChatRequest):
         "validated": h_value,  # Dynamically set the validated field
         "visitFromDelta": False,
         "webSearchModePrompt": False,
-        "imageGenerationMode": False,
+        "imageGenerationMode": False,  # Added this line
     }
 
     async with httpx.AsyncClient() as client:
@@ -184,19 +173,14 @@ async def process_non_streaming_response(request: ChatRequest):
         logger.error("Failed to retrieve h-value for validation.")
         raise HTTPException(status_code=500, detail="Validation failed due to missing h-value.")
 
-    # Determine if images are included in messages
-    code_model_mode = any(
-        hasattr(msg, 'data') and msg.data and 'imagesData' in msg.data for msg in request.messages
-    )
-
     json_data = {
         "agentMode": agent_mode,
         "clickedAnswer2": False,
         "clickedAnswer3": False,
         "clickedForceWebSearch": False,
-        "codeModelMode": code_model_mode,
+        "codeModelMode": True,
         "githubToken": None,
-        "id": request_id,
+        "id": None,  # Using request_id instead of chat_id
         "isChromeExt": False,
         "isMicMode": False,
         "maxTokens": request.max_tokens,
@@ -212,7 +196,7 @@ async def process_non_streaming_response(request: ChatRequest):
         "validated": h_value,  # Dynamically set the validated field
         "visitFromDelta": False,
         "webSearchModePrompt": False,
-        "imageGenerationMode": False,
+        "imageGenerationMode": False,  # Added this line
     }
 
     full_response = ""