Update api/utils.py

api/utils.py

@@ -7,7 +7,7 @@ import uuid
 import httpx
 from api import validate
 from api.config import MODEL_MAPPING, headers, AGENT_MODE, TRENDING_AGENT_MODE
-from fastapi import Depends
+from fastapi import Depends, security
 from fastapi.security import HTTPAuthorizationCredentials
 
 from api.config import APP_SECRET, BASE_URL
@@ -41,24 +41,26 @@ def verify_app_secret(credentials: HTTPAuthorizationCredentials = Depends(securi
     return credentials.credentials
 
 def message_to_dict(message):
-    """
-    Convert a Message object to a dictionary.
-    Mask the content to prevent logging sensitive information.
-    """
     if isinstance(message.content, str):
-        masked_content = "[HIDDEN]"
+        return {"role": message.role, "content": message.content}
     elif isinstance(message.content, list) and len(message.content) == 2:
-        masked_content = "[HIDDEN]"
+        return {
+            "role": message.role,
+            "content": message.content[0]["text"],
+            "data": {
+                "imageBase64": message.content[1]["image_url"]["url"],
+                "fileText": "",
+                "title": "snapshot",
+            },
+        }
     else:
-        masked_content = "[HIDDEN]"
-    
-    return {"role": message.role, "content": masked_content}
+        return {"role": message.role, "content": message.content}
 
 async def process_streaming_response(request: ChatRequest):
     agent_mode = AGENT_MODE.get(request.model, {})
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
 
-    # Log only non-sensitive information
+    # Log reduced information
     logger.info(
         f"Streaming request for model: '{request.model}', "
         f"agent mode: {agent_mode}, trending agent mode: {trending_agent_mode}"
@@ -124,7 +126,7 @@ async def process_non_streaming_response(request: ChatRequest):
     agent_mode = AGENT_MODE.get(request.model, {})
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
 
-    # Log only non-sensitive information
+    # Log reduced information
     logger.info(
         f"Non-streaming request for model: '{request.model}', "
         f"agent mode: {agent_mode}, trending agent mode: {trending_agent_mode}"