Update app.py

app.py

@@ -14,20 +14,20 @@ def my_custom_tool(arg1:int, arg2:int, arg3:str)-> str: #it's import to specify
     Args:
         arg1: the first integer
         arg2: the second integer
-        arg3: the operator. Supported operations are +,-,*,
+        arg3: the operator as string. Supported operations are "+","-","*","/"
     """
     match arg3:
         case "*":
-            return arg1 * arg2
+            return  f"The result is: '{arg1 * arg2}'"
         case "/":
             if(arg2!=0)
-                return arg1 / arg2
+                return f"The result is: '{arg1 / arg2}'"
             else
                 return f"Error: Division by zero not possible"
         case "+":
-            return arg1 + arg2
+            return f"The result is: '{arg1 + arg2}'"
         case "-":
-            return arg1 - arg2
+            return f"The result is: '{arg1 - arg2}'"
         case _:
             return f"Error: '{arg3}' is an unsupported operation, Supported operations are +,-,*,/"
 
@@ -64,7 +64,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, my_custom_tool, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,