Update agent.py

agent.py

@@ -38,9 +38,9 @@ def divide(a: int,b: int) -> float:
 
 @tool
 def web_search(query: str) -> str:
-  """ Search for a query on web and return best 5 result."""
+  """ Search for a query on web and return best 2 result."""
 
-  search_results = TavilySearchResults(max_results = 5).invoke(input=query)
+  search_results = TavilySearchResults(max_results = 2).invoke(input=query)
 
   formatted_search_results = "\n\n-----\n\n".join(
       [
@@ -53,9 +53,9 @@ def web_search(query: str) -> str:
 
 @tool
 def wikipedia_search(query: str) -> str:
-  """ Search for a query on wikipedia and return best 3 result."""
+  """ Search for a query on wikipedia and return best 2 result."""
 
-  loader = WikipediaLoader(query=query, load_max_docs=3)
+  loader = WikipediaLoader(query=query, load_max_docs=2)
   search_results = loader.load() # Now, just call load() without arguments
 
   formatted_search_results = "\n\n-----\n\n".join(
@@ -70,10 +70,10 @@ def wikipedia_search(query: str) -> str:
 
 @tool
 def arxiv_search(query: str) -> str:
-  """ Search for a query on arxiv and return best 3 result."""
+  """ Search for a query on arxiv and return best 2 result."""
 
   # Similar to WikipediaLoader, query and load_max_docs are passed during initialization
-  loader = ArxivLoader(query=query, load_max_docs=3)
+  loader = ArxivLoader(query=query, load_max_docs=2)
   search_results = loader.load() # Call load() without arguments
 
   formatted_search_results = "\n\n-----\n\n".join(
@@ -86,7 +86,7 @@ def arxiv_search(query: str) -> str:
 
 
 
-system_prompt = """You are a general AI assistant. I will ask you a question. Use your tools to report your thoughts, and finish your answer with the following template:
+system_prompt = """You are a general AI assistant. I will ask you a question. Use your tools and think carefully to report your thoughts, and finish your answer with the following template:
 FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
 If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
 If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.