Update tools.py

tools.py

@@ -57,8 +57,8 @@ def create_calculator_tool() -> Graph:
         
         return {"output": output}
 
-    # Create the graph
-    workflow = StateGraph()
+    # Create the graph with state schema
+    workflow = StateGraph(dict)
     
     # Add the calculator tool node
     workflow.add_node("calculator", ToolNode(calculator_function))
@@ -100,8 +100,8 @@ def create_search_tool() -> Graph:
             
             return {"output": output}
 
-    # Create the graph
-    workflow = StateGraph()
+    # Create the graph with state schema
+    workflow = StateGraph(dict)
     
     # Add the search tool node
     workflow.add_node("search", ToolNode(search_function))