Update app.py

app.py

@@ -229,20 +229,32 @@ def create_chat_function(agent):
             # Keep last 5 turns (10 messages) to balance context and token usage
             messages = []
             
-            # Add recent history (last 5 exchanges)
+            # Add recent history (last 5 exchanges) - handle both list and dict formats
             recent_history = history[-5:] if len(history) > 5 else history
-            for user_msg, assistant_msg in recent_history:
-                messages.append({"role": "user", "content": user_msg})
-                if assistant_msg:  # Sometimes assistant message might be None
-                    messages.append({"role": "assistant", "content": assistant_msg})
+            
+            for item in recent_history:
+                # Handle different Gradio history formats
+                if isinstance(item, (list, tuple)) and len(item) >= 2:
+                    user_msg, assistant_msg = item[0], item[1]
+                    messages.append({"role": "user", "content": user_msg})
+                    if assistant_msg:  # Sometimes assistant message might be None
+                        messages.append({"role": "assistant", "content": assistant_msg})
+                elif isinstance(item, dict):
+                    # Some Gradio versions use dict format
+                    if "role" in item and "content" in item:
+                        messages.append(item)
             
             # Add current message
             messages.append({"role": "user", "content": message})
             
+            # Configuration with thread_id for checkpointer
+            config = {"configurable": {"thread_id": "default_thread"}}
+            
             # Stream responses from agent
             results = []
             for step in agent.stream(
                 {"messages": messages},
+                config=config,
                 stream_mode="values",
             ):
                 last_message = step["messages"][-1]
@@ -257,7 +269,7 @@ def create_chat_function(agent):
             return "I apologize, but I couldn't generate a response. Please try rephrasing your question."
             
         except Exception as e:
-            logger.error(f"Error in chat function: {e}")
+            logger.error(f"Error in chat function: {e}", exc_info=True)
             return f"An error occurred: {str(e)}. Please try again or contact support."
     
     return chat