better handle agent responses and improve error logging

app.py

@@ -358,16 +358,28 @@ async def stream_agent_response(question: str, chat_history: List) -> List[Dict]
         assistant_message = {"role": "assistant", "content": ""}
         messages.append(assistant_message)
         
-        # Execute the agent
-        response = await agent.ainvoke({"input": question, "chat_history": chat_history})
-        
-        # Process the response
-        if hasattr(response, 'output') and response.output:
-            response_text = response.output
+        # Execute the agent with proper error handling
+        try:
+            response = await agent.ainvoke({"input": question, "chat_history": chat_history})
+            logger.info(f"Agent response type: {type(response)}")
+            logger.info(f"Agent response content: {str(response)[:500]}...")
+            
+            # Handle different response formats
+            if hasattr(response, 'output') and response.output:
+                response_text = response.output
+            elif isinstance(response, str):
+                response_text = response
+            elif hasattr(response, 'get') and callable(response.get) and 'output' in response:
+                response_text = response['output']
+            else:
+                response_text = str(response)
+            
+            logger.info(f"Extracted response text: {response_text[:200]}...")
             
             # Check if the response contains an SQL query
             sql_query = extract_sql_query(response_text)
             if sql_query:
+                logger.info(f"Detected SQL query: {sql_query}")
                 # Execute the query and update the response
                 db_connection, _ = setup_database_connection()
                 if db_connection:
@@ -379,6 +391,11 @@ async def stream_agent_response(question: str, chat_history: List) -> List[Dict]
             # Update the assistant's message with the response
             assistant_message["content"] = response_text
             
+        except Exception as e:
+            error_msg = f"Error al ejecutar el agente: {str(e)}"
+            logger.error(error_msg, exc_info=True)
+            assistant_message["content"] = f"## ❌ Error\n\n{error_msg}"
+            
         else:
             assistant_message["content"] = "Error: No se recibió respuesta del agente."