Update tools.py

tools.py

@@ -35,8 +35,8 @@ 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:
+    
+    def calculator_function(state: CalculatorState) -> dict:
         if len(state.input.numbers) < 2:
             raise ValueError("At least two numbers are required for calculation")
         
@@ -56,12 +56,12 @@ def create_calculator_tool() -> Graph:
             else:
                 raise ValueError(f"Unsupported operation: {state.input.operation}")
         
-        state.output = CalculatorOutput(
-            result=result,
-            operation=state.input.operation
-        )
-        
-        return state
+        return {
+            "output": CalculatorOutput(
+                result=result,
+                operation=state.input.operation
+            )
+        }
 
     # Create the graph with state schema
     workflow = StateGraph(state_schema=CalculatorState)
@@ -78,7 +78,7 @@ def create_calculator_tool() -> Graph:
 def create_search_tool() -> Graph:
     """Creates a search tool using DuckDuckGo that can search for information online."""
     
-    def search_function(state: SearchState) -> SearchState:
+    def search_function(state: SearchState) -> dict:
         # Initialize DuckDuckGo search
         with DDGS() as ddgs:
             # Perform the search
@@ -97,12 +97,12 @@ def create_search_tool() -> Graph:
                 for result in search_results
             ]
             
-            state.output = SearchOutput(
-                results=results,
-                query=state.input.query
-            )
-            
-            return state
+            return {
+                "output": SearchOutput(
+                    results=results,
+                    query=state.input.query
+                )
+            }
 
     # Create the graph with state schema
     workflow = StateGraph(state_schema=SearchState)