Update tools.py

tools.py

@@ -15,6 +15,10 @@ class CalculatorState(TypedDict):
     input: CalculatorInput
     output: Optional[CalculatorOutput]
 
+class SearchInput(TypedDict):
+    query: str
+    max_results: int
+
 class SearchResult(TypedDict):
     title: str
     link: str
@@ -25,7 +29,7 @@ class SearchOutput(TypedDict):
     query: str
 
 class SearchState(TypedDict):
-    input: Dict[str, Any]  # Contains query and max_results
+    input: SearchInput  # Now typed correctly
     output: Optional[SearchOutput]
 
 def create_calculator_tool() -> Graph:
@@ -80,7 +84,7 @@ def create_search_tool() -> Graph:
         with DDGS() as ddgs:
             raw_results = list(ddgs.text(
                 state["input"]["query"],
-                max_results=state["input"].get("max_results", 3)
+                max_results=state["input"]["max_results"]
             ))
 
         results = []