Update api/utils.py

api/utils.py

@@ -61,6 +61,7 @@ def message_to_dict(message) -> Dict[str, Any]:
 
 # Function to retrieve agent modes for a specific model
 def get_agent_modes(model: str) -> Dict[str, Any]:
+    """Returns specific agent mode configurations for agent models."""
     agent_mode = AGENT_MODE.get(model, {})
     trending_agent_mode = TRENDING_AGENT_MODE.get(model, {})
 
@@ -79,7 +80,7 @@ async def process_streaming_response(request: ChatRequest):
     chat_id = generate_chat_id()
     logger.info(f"Generated Chat ID: {chat_id} - Model: {request.model}")
 
-    # Retrieve agent modes based on the model
+    # Check if the model is an agent model and apply corresponding agent modes
     agent_modes = get_agent_modes(request.model)
     headers = get_headers()
 
@@ -107,7 +108,7 @@ async def process_streaming_response(request: ChatRequest):
         "visitFromDelta": False,
     }
 
-    logger.debug(f"Data Payload: {json_data}")  # Inspect payload for accuracy
+    logger.debug(f"Data Payload for {request.model}: {json_data}")  # Inspect payload for accuracy
     async with httpx.AsyncClient() as client:
         try:
             async with client.stream(
@@ -164,7 +165,7 @@ async def process_non_streaming_response(request: ChatRequest):
         "visitFromDelta": False,
     }
 
-    logger.debug(f"Data Payload: {json_data}")  # Inspect payload for accuracy
+    logger.debug(f"Data Payload for {request.model}: {json_data}")  # Inspect payload for accuracy
     full_response = ""
     async with httpx.AsyncClient() as client:
         try: