Update app.py

app.py

@@ -33,6 +33,28 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def calculator(num_1: float, num_2: float, sign: str) -> float|str:
+    """ A tool tha receives two numbers and a sign and execute the sign operation over that two numbers.
+    Args:
+        num_1: first number to be computed
+        num_2: second number to be computed
+        sign: the operation sign
+    """
+    if sign == '+':
+        return num_1+num_2
+    elif sign == '-':
+        return num_1-num_2
+    elif sign == '*':
+        return num_1*num_2
+    elif sign == '/':
+        if num_2 != 0:
+            return num_1/num_2
+        else:
+            return 'You can't make a division by 0!'
+    else:
+        return 'Sign not recognized!'
+    
 
 final_answer = FinalAnswerTool()