From f11a95cc1920c3f72a100819d95f3875ad0db57e Mon Sep 17 00:00:00 2001 From: Aiden Dai Date: Wed, 10 Apr 2024 09:49:48 +0800 Subject: [PATCH] Add support of encoded image url --- src/api/models/bedrock.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/api/models/bedrock.py b/src/api/models/bedrock.py index 5ada2cf..9e83460 100644 --- a/src/api/models/bedrock.py +++ b/src/api/models/bedrock.py @@ -191,12 +191,12 @@ class ClaudeModel(BedrockModel): # Send a request to the image URL response = requests.get(image_url) - content_type = response.headers.get('Content-Type') - if not content_type.startswith("image"): - content_type = "image/jpeg" - # Check if the request was successful if response.status_code == 200: + + content_type = response.headers.get('Content-Type') + if not content_type.startswith("image"): + content_type = "image/jpeg" # Get the image content image_content = response.content # Encode the image content as base64