Update app.py

app.py

@@ -287,29 +287,37 @@ class BasicAgent:
             document_qa,
             python_execution
         ]
-    
-
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question: {question[:50]}{'...' if len(question) > 50 else ''}")
-        
-        # Ensure proper HumanMessage in history
-        state = init_state(question)
-    
+
+        # Initialize state with proper structure
+        state = {
+            "question": question,
+            "context": {},  # Ensure it's a dict
+            "reasoning": "",
+            "iterations": 0,
+            "history": [HumanMessage(content=question)]
+        }
+
         final_state = self.workflow.invoke(state)
-    
+
         print(f"Final state keys: {list(final_state.keys())}")
         if 'history' in final_state:
             print(f"History length: {len(final_state['history'])}")
             for i, msg in enumerate(final_state['history']):
-                print(f"Message {i}: {type(msg).__name__} - {msg.content[:100]}...")
-    
-        for msg in reversed(final_state['history']):
+                if isinstance(msg, dict):
+                    print(f"Message {i}: dict - {msg}")
+                else:
+                    print(f"Message {i}: {type(msg).__name__} - {msg.content[:100]}...")
+
+        # Extract the FINAL ANSWER from history
+        for msg in reversed(final_state["history"]):
             if isinstance(msg, AIMessage) and "FINAL ANSWER:" in msg.content:
                 answer = msg.content.split("FINAL ANSWER:")[1].strip()
                 print(f"Agent returning answer: {answer}")
                 return answer
-    
+
         raise ValueError("No FINAL ANSWER found in agent history.")