fix(factory.py): fix bedrock http:// handling

This commit is contained in:
Krrish Dholakia 2025-02-03 18:15:14 -08:00
parent 7ddb034b31
commit 5b08289d88

View file

@ -2307,7 +2307,7 @@ class BedrockImageProcessor:
"""Synchronous image processing.""" """Synchronous image processing."""
if "base64" in image_url: if "base64" in image_url:
img_bytes, mime_type, image_format = cls._parse_base64_image(image_url) img_bytes, mime_type, image_format = cls._parse_base64_image(image_url)
elif "https:/" in image_url: elif "http://" in image_url or "https://" in image_url:
img_bytes, mime_type = BedrockImageProcessor.get_image_details(image_url) img_bytes, mime_type = BedrockImageProcessor.get_image_details(image_url)
image_format = mime_type.split("/")[1] image_format = mime_type.split("/")[1]
else: else: