memory upgrade

app.py

@@ -411,6 +411,8 @@ def convert_to_messages_format(chat_history):
 
 async def stream_agent_response(question: str, chat_history: List) -> List[Dict]:
     """Procesa la pregunta del usuario y devuelve la respuesta del agente con memoria de conversación."""
+    global agent  # Make sure we can modify the agent's memory
+    
     # Initialize response
     response_text = ""
     messages = []
@@ -439,6 +441,22 @@ async def stream_agent_response(question: str, chat_history: List) -> List[Dict]
         )
         assistant_message = {"role": "assistant", "content": error_msg}
         return [assistant_message]
+        
+    # Update the agent's memory with the full conversation history
+    try:
+        # Clear existing memory
+        if hasattr(agent, 'memory') and agent.memory is not None:
+            agent.memory.clear()
+            
+        # Add all messages to memory
+        for i in range(0, len(messages)-1, 2):  # Process in pairs (user, assistant)
+            if i+1 < len(messages):
+                agent.memory.save_context(
+                    {"input": messages[i].content},
+                    {"output": messages[i+1].content}
+                )
+    except Exception as e:
+        logger.error(f"Error updating agent memory: {str(e)}", exc_info=True)
     
     try:
         # Add empty assistant message that will be updated