tool

app.py

@@ -139,11 +139,14 @@ class BasicAgent:
     def _use_calculator(self, state: AgentState) -> AgentState:
         """Use the calculator tool."""
         try:
-            result = self.calculator.invoke({"input": eval(state.tool_output)})
+            # Create calculator state with input from tool_output
+            calc_input = eval(state.tool_output)
+            result_state = self.calculator.invoke(CalculatorState(input=calc_input))
+            
             state.history.append({
                 'step': 'calculator',
                 'input': state.tool_output,
-                'output': str(result['output'].result)
+                'output': str(result_state.output.result)
             })
             state.current_step = 'final_answer'
         except Exception as e:
@@ -157,16 +160,17 @@ class BasicAgent:
     def _use_search(self, state: AgentState) -> AgentState:
         """Use the search tool."""
         try:
-            result = self.search_tool.invoke({
-                "input": {
-                    "query": state.search_query,
-                    "max_results": 3
-                }
-            })
+            # Create search state with input from search_query
+            search_input = SearchInput(
+                query=state.search_query,
+                max_results=3
+            )
+            result_state = self.search_tool.invoke(SearchState(input=search_input))
+            
             state.history.append({
                 'step': 'search',
                 'query': state.search_query,
-                'results': [str(r) for r in result['output'].results]
+                'results': [str(r) for r in result_state.output.results]
             })
             state.needs_more_info = False
             state.current_step = 'final_answer'

tools.py

@@ -35,7 +35,7 @@ class SearchState(BaseModel):
 
 def create_calculator_tool() -> Graph:
     """Creates a calculator tool using LangGraph that can perform basic arithmetic operations."""
-    
+    print("Creating calculator tool")
     def calculator_function(state: CalculatorState) -> CalculatorState:
         if len(state.input.numbers) < 2:
             raise ValueError("At least two numbers are required for calculation")