v3

app.py

@@ -423,7 +423,7 @@ def convert_to_messages_format(chat_history):
     
     return messages
 
-async def stream_agent_response(question: str, chat_history: List) -> List[Dict]:
+async def stream_agent_response(question: str, chat_history: List[Tuple[str, str]]) -> str:
     """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
     
@@ -432,13 +432,11 @@ async def stream_agent_response(question: str, chat_history: List) -> List[Dict]
     messages = []
     
     # Add previous chat history in the correct format for the agent
-    if chat_history:
-        # Convert chat history to the format expected by the agent's memory
-        for msg in chat_history:
-            if msg["role"] == "user":
-                messages.append(HumanMessage(content=msg["content"]))
-            elif msg["role"] == "assistant":
-                messages.append(AIMessage(content=msg["content"]))
+    for user_msg, assistant_msg in chat_history:
+        if user_msg:
+            messages.append(HumanMessage(content=user_msg))
+        if assistant_msg:
+            messages.append(AIMessage(content=assistant_msg))
     
     # Add current user's question
     user_message = HumanMessage(content=question)
@@ -797,7 +795,7 @@ def create_application():
     # Create the UI components
     demo, chatbot, question_input, submit_button, streaming_output_display = create_ui()
     
-    def user_message(user_input: str, chat_history: List[Dict]) -> Tuple[str, List[Dict]]:
+    def user_message(user_input: str, chat_history: List[Tuple[str, str]]) -> Tuple[str, List[Tuple[str, str]]]:
         """Add user message to chat history and clear input."""
         if not user_input.strip():
             return "", chat_history
@@ -808,33 +806,30 @@ def create_application():
         if chat_history is None:
             chat_history = []
         
-        # Add user message
-        chat_history.append({"role": "user", "content": user_input})
-        
-        # Add empty assistant response
-        chat_history.append({"role": "assistant", "content": ""})
+        # Add user message and empty assistant response
+        chat_history.append((user_input, None))
         
         # Clear the input
         return "", chat_history
     
-    async def bot_response(chat_history: List[Dict]) -> List[Dict]:
+    async def bot_response(chat_history: List[Tuple[str, str]]) -> List[Tuple[str, str]]:
         """Get bot response and update chat history."""
-        if not chat_history or chat_history[-1]["role"] != "assistant":
+        if not chat_history:
             return chat_history
-        
+            
+        # Get the last user message (first element of the last tuple if it exists)
+        if not chat_history[-1][0] or chat_history[-1][1] is not None:
+            return chat_history
+            
         try:
-            # Get the user's question (second to last message)
-            if len(chat_history) < 2:
-                return chat_history
-                
-            question = chat_history[-2]["content"]
+            question = chat_history[-1][0]
             logger.info(f"Processing question: {question}")
             
             # Call the agent and get the response
-            assistant_message = await stream_agent_response(question, chat_history[:-2])
+            assistant_message = await stream_agent_response(question, chat_history[:-1])
             
             # Update the assistant's message in the chat history
-            chat_history[-1]["content"] = assistant_message
+            chat_history[-1] = (question, assistant_message)
             
             logger.info("Response generation complete")
             return chat_history
@@ -842,7 +837,8 @@ def create_application():
         except Exception as e:
             error_msg = f"## ❌ Error\n\nError al procesar la solicitud:\n\n```\n{str(e)}\n```"
             logger.error(error_msg, exc_info=True)
-            chat_history[-1]["content"] = error_msg
+            if chat_history and len(chat_history[-1]) == 2 and chat_history[-1][1] is None:
+                chat_history[-1] = (chat_history[-1][0], error_msg)
             return chat_history
     
     # Event handlers