Update agent.py

agent.py

@@ -14,17 +14,32 @@ load_dotenv()
 
 # ========== TOOLS ==========
 @tool
-def multiply(a: int, b: int) -> int: return a * b
+def multiply(a: int, b: int) -> int:
+    """Multiply two integers."""
+    return a * b
+
 @tool
-def add(a: int, b: int) -> int: return a + b
+def add(a: int, b: int) -> int:
+    """Add two integers."""
+    return a + b
+
 @tool
-def subtract(a: int, b: int) -> int: return a - b
+def subtract(a: int, b: int) -> int:
+    """Subtract two integers."""
+    return a - b
+
 @tool
 def divide(a: int, b: int) -> float:
-    if b == 0: raise ValueError("Cannot divide by zero.")
+    """Divide two numbers. Raises ValueError if dividing by zero."""
+    if b == 0:
+        raise ValueError("Cannot divide by zero.")
     return a / b
+
 @tool
-def modulus(a: int, b: int) -> int: return a % b
+def modulus(a: int, b: int) -> int:
+    """Compute modulus (remainder) of two integers."""
+    return a % b
+
 @tool
 def wiki_search(query: str) -> str:
     search_docs = WikipediaLoader(query=query, load_max_docs=2).load()