fix typos

app.py

@@ -53,13 +53,13 @@ def web_search(query: str) -> str:
     return {"web_results": formatted_search_docs}
 
 @tool
-def arxiv_search(query: str, load_max_docs: int) -> Dict[str, str]:
+def arxiv_search(query: str) -> str:
     """Search Arxiv for a query and return maximum 3 result.
     Args:
         query: The search query.
         load_max_docs: The maximum number of documents to load.
         """
-    search_docs = ArxivLoader(query=query, load_max_docs=load_max_docs).load()
+    search_docs = ArxivLoader(query=query, load_max_docs=3).load()
     formatted_search_docs = "\n\n---\n\n".join(
         [
             f'<Document Title="{doc.metadata["Title"]}" Published="{doc.metadata["Published"]}" Authors="{doc.metadata["Authors"]} Summary={doc.metadata["Summary"]}"/>\n{doc.page_content}\n</Document>'
@@ -76,7 +76,8 @@ def wikipedia_search(query: str, load_max_docs: int=3) -> str:
         load_max_docs: The maximum number of documents to load.
     """
     search_docs = WikipediaLoader(query=query, load_max_docs=load_max_docs).load()
-    return {"wikipedia_results": search_docs}
+    # return {"wikipedia_results": search_docs}
+    return search_docs
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------