Update agent.py

agent.py

@@ -70,19 +70,49 @@ def modulus(a: int, b: int) -> str:
 # ------------------ Retrieval Tools ------------------
 @tool
 def wiki_search(query: str) -> str:
+    """
+    Search Wikipedia for a given query and return the content of up to two matching articles.
+
+    Args:
+        query: A string query to search on Wikipedia.
+
+    Returns:
+        A string containing the content from up to two relevant Wikipedia articles, separated by dividers.
+    """
     docs = WikipediaLoader(query=query, load_max_docs=2).load()
     return "\n\n---\n\n".join(doc.page_content for doc in docs)
 
+
 @tool
 def web_search(query: str) -> str:
+    """
+    Perform a web search using Tavily and return content from the top three results.
+
+    Args:
+        query: A string query representing the web search topic.
+
+    Returns:
+        A string of up to three relevant result contents, separated by dividers.
+    """
     docs = TavilySearchResults(max_results=3).invoke(query)
     return "\n\n---\n\n".join(doc.page_content for doc in docs)
 
+
 @tool
 def arvix_search(query: str) -> str:
+    """
+    Search arXiv for academic papers matching the query and return excerpts from up to three results.
+
+    Args:
+        query: A string query to search on arXiv.
+
+    Returns:
+        A string containing excerpts (first 1000 characters) from up to three relevant papers.
+    """
     docs = ArxivLoader(query=query, load_max_docs=3).load()
     return "\n\n---\n\n".join(doc.page_content[:1000] for doc in docs)
 
+
 # ------------------ System Setup ------------------
 with open("system_prompt.txt", "r", encoding="utf-8") as f:
     system_prompt = f.read()