Update app.py

app.py

@@ -223,10 +223,10 @@ def modulus(a: int, b: int) -> int:
     
 @tool
 def wiki_search(query: str) -> str:
-    """Search Wikipedia for a query and return maximum 2 results.
+    """Search Wikipedia for a query and return maximum 10 results.
     Args:
         query: The search query."""
-    search_docs = WikipediaLoader(query=query, load_max_docs=2).load()
+    search_docs = WikipediaLoader(query=query, load_max_docs=10).load()
     formatted_search_docs = "\n\n---\n\n".join(
         [
             f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
@@ -252,7 +252,7 @@ tools = [
     # analyze_python_code,
     arvix_search,
     wiki_search,
-    GoogleSearchAPIWrapper(k=5).run,
+    GoogleSearchAPIWrapper(k=10).run,
     # DuckDuckGoSearchResults(),
 ]
 ###### state and behavior
@@ -285,15 +285,15 @@ def assistant(state: AgentState, llm_with_tools):
         # Do not ask the user to upload files."""
     previous_message = state["messages"]
     sys_msg = SystemMessage(content=f"""You are an agent that must use tools for computations or unknown info. Think step-by-step: 1. Analyze question. 2. Call tools if needed. 3. Summarize.
-    Please make sure you call GoogleSearchAPIWrapper(k=10).run tool or wiki_search tool multiple times if needed for verification or depth.
-    And you can use the relevant tools provided by the system. \n {file_info}\n. Make sure you try different tools and summarize the answer.
+    Please make sure you call GoogleSearchAPIWrapper tool or wiki_search tool multiple times if needed for verification or depth.
+    And you can use the relevant tools provided by the system. \n {file_info}\n. Make sure you try each tools at least six times with different searching strategy and conclude the answer.
     I will ask you a question. Please return your answer with the following template: [YOUR FINAL ANSWER] without brackets.
     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. 
     If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
     Please return YOUR FINAL ANSWER only, without explanation, keep it brief within a single word or several words.
-    If you infer a result, you should also summary result YOUR FINAL ANSWER within a single word or several words""")
+    If you infer a result, you should also conclude result YOUR FINAL ANSWER within a single word or several words""")
 
     return {"messages": [llm_with_tools.invoke([sys_msg] + state["messages"])], "input_file": state["input_file"]}