Update app.py

app.py

@@ -34,6 +34,34 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
+
+@tool
+def calculator(arg1: int, arg2: int, operation: str) -> str:
+    """
+    A basic calculator tool.
+
+    Args:
+        arg1: first integer
+        arg2: second integer
+        operation: one of ["sum", "product", "difference", "division"]
+    """
+
+    if operation == "sum":
+        result = arg1 + arg2
+    elif operation == "product":
+        result = arg1 * arg2
+    elif operation == "difference":
+        result = arg1 - arg2
+    elif operation == "division":
+        if arg2 == 0:
+            return "Error: division by zero"
+        result = arg1 / arg2
+    else:
+        return "Error: unsupported operation"
+
+    return f"{operation} of {arg1} and {arg2} is {result}"
+
+
 final_answer = FinalAnswerTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder: