Update app.py

Pascal's sum tool added.

app.py

@@ -46,7 +46,20 @@ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may
 custom_role_conversions=None,
 )
 
-
+# My tool
+@tool
+def sum_to_num(num:int) -> int:
+    """A tool that calculates the sum of all natural numbers up to the number passed as parameter.
+    Args:
+        num: a positive integer number.
+    """
+    if isinstance(num, int):
+        if num > 1:
+            return num*(num-1)/2
+        else:
+            raise Value("Please insert a value greater than 1.")
+    else:
+        raise TypeError("Only natural numbers greater than 1 are accepted.")
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
@@ -55,7 +68,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, image_generation_tool, sum_to_num], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,