Update agent.py

agent.py

@@ -44,10 +44,10 @@ def divide(a: int,b: int) -> float:
 
 @tool
 def web_search(query: str) -> str:
-  """ Search for a query on web and return best 2 result."""
+  """ Search for a query on web and return best result."""
 
-  search = DuckDuckGoSearchResults(num_results=2)
-  results = search.invoke(query)
+  search = DuckDuckGoSearchResults(num_results=1)
+  results = search.invoke(input=query)
 
   '''formatted_results = "\n\n-----\n\n".join(
       [
@@ -78,9 +78,9 @@ def web_search(query: str) -> str:
 
 @tool
 def wikipedia_search(query: str) -> str:
-  """ Search for a query on wikipedia and return best 2 result."""
+  """ Search for a query on wikipedia and return best result."""
 
-  loader = WikipediaLoader(query=query, load_max_docs=2)
+  loader = WikipediaLoader(query=query, load_max_docs=1)
   search_results = loader.load() # Now, just call load() without arguments
 
   formatted_search_results = "\n\n-----\n\n".join(
@@ -95,10 +95,10 @@ def wikipedia_search(query: str) -> str:
 
 @tool
 def arxiv_search(query: str) -> str:
-  """ Search for a query on arxiv and return best 2 result."""
+  """ Search for a query on arxiv and return best result."""
 
   # Similar to WikipediaLoader, query and load_max_docs are passed during initialization
-  loader = ArxivLoader(query=query, load_max_docs=2)
+  loader = ArxivLoader(query=query, load_max_docs=1)
   search_results = loader.load() # Call load() without arguments
 
   formatted_search_results = "\n\n-----\n\n".join(