str obligatoriu

app.py

@@ -21,11 +21,10 @@ 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: str, arg2: float, arg3: float) -> Union[float, str]:
+def calculus(arg1: str, arg2: float, arg3: float) -> str:
     """
-    A tool that does basic math operations: addition, subtraction,
+    A tool that performs basic math operations: addition, subtraction,
     multiplication, and division between two real numbers.
 
     Args:
@@ -34,20 +33,26 @@ def calculus(arg1: str, arg2: float, arg3: float) -> Union[float, str]:
         arg3: The second number
 
     Returns:
-        The result of the math operation, or an error message if invalid input.
+        A string representing the result of the operation or an error message.
     """
-    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 '+', '-', '*', '/'"
+    try:
+        if arg1 == '+':
+            result = arg2 + arg3
+        elif arg1 == '-':
+            result = arg2 - arg3
+        elif arg1 == '*':
+            result = arg2 * arg3
+        elif arg1 == '/':
+            if arg3 == 0:
+                return "Error: Division by zero"
+            result = arg2 / arg3
+        else:
+            return "Error: Invalid operator. Use one of '+', '-', '*', '/'"
+
+        return str(result)
+    except Exception as e:
+        return f"Error: {str(e)}"
+
 
 
 @tool