Update agent.py

agent.py

@@ -21,26 +21,52 @@ load_dotenv()
 # ------------------ Arithmetic Tools ------------------
 @tool
 def multiply(a: int, b: int) -> str:
+    """Multiply two integers and return the product as a string."""
     return str(a * b)
 
+
 @tool
 def add(a: int, b: int) -> str:
+    """Add two integers and return the sum as a string."""
     return str(a + b)
 
 @tool
 def subtract(a: int, b: int) -> str:
+    """Subtract two integers and return the difference as a string."""
     return str(a - b)
 
 @tool
 def divide(a: int, b: int) -> str:
+    """
+    Divide two integers and return the result as a string.
+
+    Args:
+        a: The numerator (dividend).
+        b: The denominator (divisor). Must not be zero.
+
+    Returns:
+        A string representation of the division result, or an error message if b is zero.
+    """
     if b == 0:
         return "Error: Cannot divide by zero."
     return str(a / b)
 
+
 @tool
 def modulus(a: int, b: int) -> str:
+    """
+    Compute the modulus (remainder) of two integers and return the result as a string.
+
+    Args:
+        a: The numerator.
+        b: The denominator.
+
+    Returns:
+        A string representation of the remainder when a is divided by b.
+    """
     return str(a % b)
 
+
 # ------------------ Retrieval Tools ------------------
 @tool
 def wiki_search(query: str) -> str: