Hadnle multiple images request

app.py

@@ -19,24 +19,27 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     return "What magic will you build ?"
 
 @tool
-def get_cat_image(limit: int = 1) -> str:
-    """A tool that get a random image of a cat
+def get_cat_images(limit: int = 1) -> list:
+    """A tool that gets random cat images
     Args:
         limit: Number of images to return between 1 and 100
     """
-    # URL for the cat image API
-    url = "https://api.thecatapi.com/v1/images/search"
+    # Ensure the limit is within the allowed range
+    if limit < 1 or limit > 100:
+        return "Limit must be between 1 and 100."
+    
+    # URL for the cat image API with the limit parameter
+    url = f"https://api.thecatapi.com/v1/images/search?limit={limit}"
     
     # Send the GET request to the API
     response = requests.get(url)
     
     # Check if the request was successful
     if response.status_code == 200:
-        # Extract the URL of the image from the response JSON
+        # Extract the URLs of the images from the response JSON
         data = response.json()  # Parse the response as JSON
-        image_url = data[0]['url']  # Extract the image URL
-        # print("Cat Image URL:", image_url)
-        return image_url
+        image_urls = [item['url'] for item in data]  # Extract URLs for all images
+        return image_urls
     else:
         return f"Failed to fetch data. Status code: {response.status_code}"