chatgpt...

app.py

@@ -4,6 +4,8 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from typing import Union
+
 
 from Gradio_UI import GradioUI
 
@@ -19,24 +21,34 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     return "What magic will you build ?"
 
 
+
 @tool
-def calculus(arg1:chr, arg2:double, arg3: double)-> double:
-    """A tool that does basic math operations, like adition, substraction, multimplication and division between two real numbers. 
+def calculus(arg1: str, arg2: float, arg3: float) -> Union[float, str]:
+    """
+    A tool that does basic math operations: addition, subtraction,
+    multiplication, and division between two real numbers.
+
     Args:
-        arg1: the operator (should be one of the following: '+','-','*','/')
-        arg2: the first number
-        arg3: the second number
+        arg1: The operator (should be one of the following: '+', '-', '*', '/')
+        arg2: The first number
+        arg3: The second number
+
+    Returns:
+        The result of the math operation, or an error message if invalid input.
     """
-    if arg1=='+':
-        return arg2+arg3
-    elif arg1=='-':
-        return arg2-arg3
-    elif arg1=='*':
-        return arg2*arg3
-    elif arg1=='/':
-        return arg2/arg3
-    
-    return "The character in arg1 wasn't a valid mathematical operation!"
+    if arg1 == '+':
+        return arg2 + arg3
+    elif arg1 == '-':
+        return arg2 - arg3
+    elif arg1 == '*':
+        return arg2 * arg3
+    elif arg1 == '/':
+        if arg3 == 0:
+            return "Error: Division by zero!"
+        return arg2 / arg3
+
+    return "Error: Invalid operator. Use one of '+', '-', '*', '/'"
+
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: