Update api/utils.py

api/utils.py

@@ -47,13 +47,13 @@ def create_chat_completion_data(
         "usage": None,
     }
 
-# Function to convert message to dictionary format with optional model prefix
+# 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
+    content = message.content if isinstance(message.content, str) else message.content[0]["text"]
     if model_prefix:
         content = f"{model_prefix} {content}"
-    # Allow base64 image data for all models
-    if isinstance(message.content, list) and len(message.content) == 2:
+    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,
@@ -63,8 +63,7 @@ def message_to_dict(message, model_prefix: Optional[str] = None):
                 "title": "snapshot",
             },
         }
-    else:
-        return {"role": message.role, "content": content}
+    return {"role": message.role, "content": content}
 
 # Function to strip model prefix from content if present
 def strip_model_prefix(content: str, model_prefix: Optional[str] = None) -> str:
@@ -76,7 +75,7 @@ def strip_model_prefix(content: str, model_prefix: Optional[str] = None) -> str:
 
 # Function to get the correct referer URL for logging
 def get_referer_url(chat_id: str, model: str) -> str:
-    """Generate the referer URL based on specific models."""
+    """Generate the referer URL based on specific models listed in MODEL_REFERERS."""
     if model in MODEL_REFERERS:
         return f"{BASE_URL}/chat/{chat_id}?model={model}"
     return BASE_URL