Update agent.py

agent.py

@@ -12,28 +12,34 @@ from langchain_core.tools import tool
 
 @tool
 def multiply(a: int, b: int) -> int:
+    """Multiply two integers and return the result."""
     return a * b
 
 @tool
 def add(a: int, b: int) -> int:
+    """Add two integers and return the result."""
     return a + b
 
 @tool
 def subtract(a: int, b: int) -> int:
+    """Subtract b from a and return the result."""
     return a - b
 
 @tool
 def divide(a: int, b: int) -> float:
+    """Divide a by b and return the result. Raises an error if b is zero."""
     if b == 0:
         raise ValueError("Cannot divide by zero.")
     return a / b
 
 @tool
 def modulus(a: int, b: int) -> int:
+    """Return the modulus (remainder) of a divided by b."""
     return a % b
 
 @tool
 def wiki_search(query: str) -> dict:
+    """Search Wikipedia for a query and return up to 2 results."""
     search_docs = WikipediaLoader(query=query, load_max_docs=2).load()
     results = "\n\n---\n\n".join(
         f"<Document>\n{doc.page_content}\n</Document>" for doc in search_docs
@@ -42,6 +48,7 @@ def wiki_search(query: str) -> dict:
 
 @tool
 def web_search(query: str) -> dict:
+    """Search the web via Tavily and return up to 3 results."""
     search_docs = TavilySearchResults(max_results=3).invoke(query=query)
     results = "\n\n---\n\n".join(
         f"<Document>\n{doc.page_content}\n</Document>" for doc in search_docs
@@ -50,6 +57,7 @@ def web_search(query: str) -> dict:
 
 @tool
 def arvix_search(query: str) -> dict:
+    """Search Arxiv and return up to 3 truncated results."""
     search_docs = ArxivLoader(query=query, load_max_docs=3).load()
     results = "\n\n---\n\n".join(
         f"<Document>\n{doc.page_content[:500]}\n</Document>" for doc in search_docs