fix: Update chat message handling

src/interfaces/gradio_interface.py

@@ -1,5 +1,4 @@
 import gradio as gr
-from gradio import ChatMessage
 from typing import Generator, List
 import json
 import os
@@ -271,23 +270,41 @@ In the fast-moving world of AI and technology, staying informed while managing i
     return about_tab
 
 
-def chat_with_agent(message: str, history: List[dict]) -> Generator[List[dict], None, None]:
+def chat_with_agent(message: str, history: List) -> Generator[List, None, None]:
     """
     Chat with the agent using custom streaming functionality for real-time thinking display
     """
     try:
+        # Convert history to proper format if needed
+        if history is None:
+            history = []
+
+        # Ensure all history items are properly formatted
+        formatted_history = []
+        for item in history:
+            if isinstance(item, dict):
+                # Already a dict, check if it has required keys
+                if "role" in item and "content" in item:
+                    formatted_history.append(item)
+                else:
+                    # Try to convert from ChatMessage format
+                    if hasattr(item, "role") and hasattr(item, "content"):
+                        formatted_history.append({"role": item.role, "content": item.content})
+            elif hasattr(item, "role") and hasattr(item, "content"):
+                # ChatMessage object
+                formatted_history.append({"role": item.role, "content": item.content})
+
         # Reset memory for long conversations to prevent token overflow
-        reset_memory = len(history) > 10  # Reset after 5 user-assistant exchanges
+        reset_memory = len(formatted_history) > 10  # Reset after 5 user-assistant exchanges
 
         # Start with user message in history
-        new_history = history.copy()
+        new_history = formatted_history.copy()
 
         # Show initial thinking message
-        thinking_message = ChatMessage(
-            role="assistant",
-            content="🧠 **Agent Planning**\n\nAnalyzing your request and creating execution plan...",
-            metadata={"title": "🤔 Agent Thinking", "status": "pending"},
-        )
+        thinking_message = {
+            "role": "assistant",
+            "content": "🧠 **Agent Planning**\n\nAnalyzing your request and creating execution plan...",
+        }
         new_history.append(thinking_message)
         yield new_history
 
@@ -321,8 +338,7 @@ def chat_with_agent(message: str, history: List[dict]) -> Generator[List[dict],
                             obs_text += "..."
                         step_content += f"👁️ **Observation:** {obs_text}\n\n"
 
-                    thinking_message.content = step_content
-                    thinking_message.metadata = {"title": f"🤔 Agent Step {step.step_number}", "status": "processing"}
+                    thinking_message["content"] = step_content
                     new_history[-1] = thinking_message
                     yield new_history
 
@@ -330,8 +346,7 @@ def chat_with_agent(message: str, history: List[dict]) -> Generator[List[dict],
             # If streaming fails, fall back to regular execution
             print(f"Streaming failed: {stream_error}, falling back to regular execution")
 
-            thinking_message.content = "🧠 **Agent Working**\n\nProcessing your request using available tools..."
-            thinking_message.metadata = {"title": "🤔 Agent Thinking", "status": "processing"}
+            thinking_message["content"] = "🧠 **Agent Working**\n\nProcessing your request using available tools..."
             new_history[-1] = thinking_message
             yield new_history
 
@@ -385,34 +400,28 @@ def chat_with_agent(message: str, history: List[dict]) -> Generator[List[dict],
                     tool_usage_content = "Agent executed actions successfully"
 
             # Update thinking to show completion
-            thinking_message.content = (
+            thinking_message["content"] = (
                 "🧠 **Agent Complete**\n\n✅ Request processed successfully\n✅ Response prepared"
             )
-            thinking_message.metadata = {"title": "🤔 Agent Thinking", "status": "done"}
             new_history[-1] = thinking_message
             yield new_history
 
             # Add tool usage message if there were tools used
             if tool_usage_content:
-                tool_message = ChatMessage(
-                    role="assistant",
-                    content=f"🛠️ **Tools & Actions Used**\n\n{tool_usage_content}",
-                    metadata={"title": "🛠️ Agent Tools", "status": "done"},
-                )
+                tool_message = {"role": "assistant", "content": f"🛠️ **Tools & Actions Used**\n\n{tool_usage_content}"}
                 new_history.append(tool_message)
                 yield new_history
 
             # Add final response
             final_response = str(result) if result else "I couldn't process your request."
-            final_message = ChatMessage(role="assistant", content=final_response)
+            final_message = {"role": "assistant", "content": final_response}
             new_history.append(final_message)
             yield new_history
             return
 
         # If we get here, streaming worked, so get the final result
         # The streaming should have shown all the steps, now get final answer
-        thinking_message.content = "🧠 **Agent Complete**\n\n✅ All steps executed\n✅ Preparing final response"
-        thinking_message.metadata = {"title": "🤔 Agent Thinking", "status": "done"}
+        thinking_message["content"] = "🧠 **Agent Complete**\n\n✅ All steps executed\n✅ Preparing final response"
         new_history[-1] = thinking_message
         yield new_history
 
@@ -424,19 +433,18 @@ def chat_with_agent(message: str, history: List[dict]) -> Generator[List[dict],
             if hasattr(last_step, "observations") and last_step.observations:
                 final_response = str(last_step.observations)
 
-        final_message = ChatMessage(role="assistant", content=final_response)
+        final_message = {"role": "assistant", "content": final_response}
         new_history.append(final_message)
         yield new_history
 
     except Exception as e:
         # Fallback error handling
-        error_message = ChatMessage(
-            role="assistant",
-            content=f"❌ **System Error:** {str(e)}\n\nPlease try again with a different approach.",
-            metadata={"title": "🚨 Error", "status": "error"},
-        )
-        history.append(error_message)
-        yield history
+        error_message = {
+            "role": "assistant",
+            "content": f"❌ **System Error:** {str(e)}\n\nPlease try again with a different approach.",
+        }
+        new_history.append(error_message)
+        yield new_history
 
 
 # Create the main chat interface