Update app.py

app.py

@@ -7,6 +7,23 @@ from tools.final_answer import FinalAnswerTool
 import random
 from Gradio_UI import GradioUI
 
+from pydantic import BaseModel, Field
+import random
+
+class RandomNumberInput(BaseModel):
+    start_range: int = Field(..., description="The start number of the range")
+    end_range: int = Field(..., description="The end number of the range")
+
+@tool(args_schema=RandomNumberInput)
+def get_random_number_between_range(start_range: int, end_range: int) -> int:
+    """A tool that generates a random number given a range.
+    
+    Args:
+        start_range: The start number of the range.
+        end_range: The end number of the range.
+    """
+    return random.randint(start_range, end_range)
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def get_random_number_between_range(start_range:int, end_range:int)-> int: #it's import to specify the return type