Update app.py

app.py

@@ -4,6 +4,7 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+import math
 
 from Gradio_UI import GradioUI
 
@@ -34,6 +35,37 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
+@tool 
+def calculator(a: float, b: float, operation: str) -> str:
+    """
+    This tool calculates 2 numbers with given operation.
+    Args:
+        a: first number
+        b: second number
+        operation: we currently support this list of operation:
+            {'Multiplication':'*', 'Division':'/', 'Addition':'+', 'Subtraction':'-', 'Power':'**', 'Modulus': '%'}
+    Note:
+        we dont support numbers above `100000.0`
+        it is possible we return division by zero error.
+    """
+
+    if operation == '*':
+        return str(a * b)
+    elif operation == '/': 
+        if b == 0: 
+            return f"Cannot division {a} with zero! Error."
+        return str(a / b)
+    elif operation == '+':
+        return str(a + b)
+    elif operation == '-':
+        return str(a - b)
+    elif operation == '**':
+        return str(math.pow(a, b))
+    elif operation == '%':
+        return str(a % b)
+    else:
+        return f"Currently operation {operation} dose not supported, chose you'r needed operation from here \{'Multiplication':'*', 'Division':'/', 'Addition':'+', 'Subtraction':'-', 'Power':'**', 'Modulus': '%'\}"
+
 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: