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, security
+from fastapi import Depends
 from fastapi.security import HTTPAuthorizationCredentials
 
 from api.config import APP_SECRET, BASE_URL
@@ -41,33 +41,31 @@ 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):
-        return {"role": message.role, "content": message.content}
+        masked_content = "[HIDDEN]"
     elif isinstance(message.content, list) and len(message.content) == 2:
-        return {
-            "role": message.role,
-            "content": message.content[0]["text"],
-            "data": {
-                "imageBase64": message.content[1]["image_url"]["url"],
-                "fileText": "",
-                "title": "snapshot",
-            },
-        }
+        masked_content = "[HIDDEN]"
     else:
-        return {"role": message.role, "content": message.content}
+        masked_content = "[HIDDEN]"
+    
+    return {"role": message.role, "content": masked_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 reduced information, excluding user content
+    # Log only non-sensitive information
     logger.info(
         f"Streaming request for model: '{request.model}', "
         f"agent mode: {agent_mode}, trending agent mode: {trending_agent_mode}"
     )
 
     json_data = {
-        "messages": [{"role": "user", "content": "User message hidden"} for _ in request.messages],
+        "messages": [message_to_dict(msg) for msg in request.messages],
         "previewToken": None,
         "userId": None,
         "codeModelMode": True,
@@ -126,14 +124,14 @@ 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 reduced information, excluding user content
+    # Log only non-sensitive information
     logger.info(
         f"Non-streaming request for model: '{request.model}', "
         f"agent mode: {agent_mode}, trending agent mode: {trending_agent_mode}"
     )
 
     json_data = {
-        "messages": [{"role": "user", "content": "User message hidden"} for _ in request.messages],
+        "messages": [message_to_dict(msg) for msg in request.messages],
         "previewToken": None,
         "userId": None,
         "codeModelMode": True,