Update app.py

app.py

@@ -10,15 +10,21 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 @tool
-def dog_average_longevity(dog_breed: str) -> str: 
-    """A tool that fetches the average lifespan for a given dog breed.
-        Parameters:
-        dog_breed (str): The breed of the dog (e.g., "Labrador").
-        Returns:
-        str: The average lifespan of the specified dog breed in years.
-        """
+def dog_average_longevity(breed: str) -> str: 
+    """
+    Retrieves the average lifespan of a given dog breed using DuckDuckGo search.
+
+    Parameters
+    ----------
+    breed : str
+        The name of the dog breed (e.g., "Labrador").
 
-    query = f"average lifespan for {dog_breed} dog in years"
+    Returns
+    -------
+    str
+        A message stating the average lifespan of the breed, or an error message if not found.
+    """
+    query = f"average lifespan for {breed} dog in years"
     
     with DDGS() as ddgs:
         results = list(ddgs.text(query, max_results=5))
@@ -27,7 +33,7 @@ def dog_average_longevity(dog_breed: str) -> str:
         for result in results:
             lifespan = extract_lifespan(result)
             if lifespan:
-                return f"The average lifespan of a {dog_breed} dog is {lifespan} years."
+                return f"The average lifespan of a {breed} dog is {lifespan} years."
     
     return "No reliable lifespan information found. Please check with a veterinarian."