fix: handle 404 images

utils_tools.py

@@ -4,6 +4,30 @@ import re
 from typing import Any
 from merchs.merch import Merchant, NaturaMerchant, MercadoLivreMerchant
 
+def get_valid_image_url(url):
+    """
+    Checks if an image URL is accessible.
+    Returns the URL if it's valid, otherwise returns None.
+    """
+    if not url:
+        return None
+    try:
+        # Use a HEAD request to check headers without downloading the full image
+        # A timeout is added as a good practice
+        response = requests.head(url, allow_redirects=True, timeout=5)
+        
+        # Check if the status code is 200 (OK)
+        if response.status_code == 200:
+            print(f"URL is valid: {url}")
+            return url
+        else:
+            print(f"URL is invalid, status code: {response.status_code}")
+            return None
+    except requests.RequestException as e:
+        # Handle connection errors, timeouts, etc.
+        print(f"An error occurred while checking the URL: {e}")
+        return None
+
 class CalculateDiscountedPriceTool(BaseTool):
     """
     A tool to calculate the final price of an item after a discount is applied.
@@ -113,6 +137,8 @@ class GetImageUrlTool(BaseTool):
         else:
             raise ValueError(f"Unsupported product type: {product_type}")
         
+        image_url = get_valid_image_url(image_url)
+        
         return image_url
 
 class MerchantSelectorTool(BaseTool):