Update app.py

app.py

@@ -141,12 +141,12 @@ class BasicAgent:
         try:
             # Create calculator state with input from tool_output
             calc_input = eval(state.tool_output)
-            result_state = self.calculator.invoke(CalculatorState(input=calc_input))
+            result = self.calculator.invoke(CalculatorState(input=calc_input))
             
             state.history.append({
                 'step': 'calculator',
                 'input': state.tool_output,
-                'output': str(result_state.output.result)
+                'output': str(result['output'].result)
             })
             state.current_step = 'final_answer'
         except Exception as e:
@@ -165,12 +165,12 @@ class BasicAgent:
                 query=state.search_query,
                 max_results=3
             )
-            result_state = self.search_tool.invoke(SearchState(input=search_input))
+            result = self.search_tool.invoke(SearchState(input=search_input))
             
             state.history.append({
                 'step': 'search',
                 'query': state.search_query,
-                'results': [str(r) for r in result_state.output.results]
+                'results': [str(r) for r in result['output'].results]
             })
             state.needs_more_info = False
             state.current_step = 'final_answer'