Update agent.py

agent.py

@@ -17,28 +17,34 @@ GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 # --- Tools ---
 @tool
 def multiply(a: int, b: int) -> int:
+    """Multiplies two integers and returns the result."""
     return a * b
 
 @tool
 def add(a: int, b: int) -> int:
+    """Adds two integers and returns the result."""
     return a + b
 
 @tool
 def subtract(a: int, b: int) -> int:
+    """Subtracts the second integer from the first."""
     return a - b
 
 @tool
 def divide(a: int, b: int) -> float:
+    """Divides the first integer by the second, returns float."""
     if b == 0:
         raise ValueError("Cannot divide by zero.")
     return a / b
 
 @tool
 def modulo(a: int, b: int) -> int:
+    """Returns the remainder of the division of two integers."""
     return a % b
 
 @tool
 def wiki_search(query: str) -> str:
+    """Search Wikipedia for a given query and return up to 2 results formatted."""
     search_docs = WikipediaLoader(query=query, load_max_docs=2).load()
     formatted = "\n\n---\n\n".join(
         [f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}">\n{doc.page_content}\n</Document>' for doc in search_docs]
@@ -47,6 +53,7 @@ def wiki_search(query: str) -> str:
 
 @tool
 def arxiv_search(query: str) -> str:
+    """Search Arxiv for a given query and return up to 3 results formatted."""
     search_docs = ArxivLoader(query=query, load_max_docs=3).load()
     formatted = "\n\n---\n\n".join(
         [f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}">\n{doc.page_content[:1000]}\n</Document>' for doc in search_docs]
@@ -55,7 +62,7 @@ def arxiv_search(query: str) -> str:
 
 @tool
 def web_search(query: str) -> str:
-    """Searches DuckDuckGo for a query."""
+    """Search DuckDuckGo (for websearch) for a query and return up to 5 links."""
     with DDGS() as ddgs:
         results = ddgs.text(query, max_results=5)
         if not results: