First basic tool : Calculator

app.py

@@ -18,6 +18,37 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+@tool
+def tool_calculator(num1: int, num2: int, operation: str) -> int:
+    
+    """
+    A tool that performs an airthmetic operation on the input two numbers. takes two numbers and an operation as inputs.
+    
+    Args:
+    num1: The first number
+    num2: The second number
+    operation: The mathematical operation (e.g., 'add', 'subtract', 'multiply', 'divide')
+    """
+    
+    # Validate that inputs are numbers
+    if not isinstance(num1, (int, float)) or not isinstance(num2, (int, float)):
+        return "Error: Inputs must be numbers."
+    
+    # Perform the requested operation
+    if operation == "add":
+        return num1 + num2
+    elif operation == "subtract":
+        return num1 - num2
+    elif operation == "multiply":
+        return num1 * num2
+    elif operation == "divide":
+        # Check for division by zero
+        if num2 == 0:
+            return "Error: Division by zero is not allowed."
+        return num1 / num2
+    else:
+        return "Error: Invalid operation. Choose from 'add', 'subtract', 'multiply', 'divide'."
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.