Update app.py

app.py

@@ -6,11 +6,12 @@ import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 import re
+from typing import Optional
 
 from Gradio_UI import GradioUI
 
 @tool
-def my_custom_tool(animal_type=None, breed=None):
+def my_custom_tool(animal_type: Optional[str] = None, breed: Optional[str] = None):
     """Finds animals and breeds from Brampton Animal Shelter website."""
     
     base_url = "https://www.brampton.ca/EN/residents/Animal-Services/Pages/Adoption.aspx"