Update app.py

app.py

@@ -27,6 +27,34 @@ def num_of_letters_in_a_word(word : str) -> int:
     """
     return len(word)
 
+
+@tool
+def calculate_between_two_numbers(num1 : float, num2 : float, operation : str) -> float:
+    """
+    A tool that takes in two different number and a specified operation as its inputs and then returns a result as a number
+    operation should be one of the following: (+, -, /, *)
+    Args:
+        num1: first number
+        num2: second number
+        operation: the operation that will be performed between these two numbers
+    """
+    if(operation == "+"):
+        return num1 + num2
+
+    elif(operation == "-"):
+        return num1 - num2
+
+    elif(operation == "/"):
+        return num1 / num2
+
+    elif(operation == "*"):
+        return num1 * num2
+
+    else:
+        print("You either entered a wrong operation or something else went wrong")
+        return -1.0
+        
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -65,7 +93,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, num_of_letters_in_a_word, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[final_answer, num_of_letters_in_a_word, get_current_time_in_timezone, calculate_between_two_numbers], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,