diff --git a/backend/ai_manager.py b/backend/ai_manager.py index 26895cc..7f34801 100644 --- a/backend/ai_manager.py +++ b/backend/ai_manager.py @@ -548,9 +548,7 @@ class AIManager: } content = [{"type": "text", "text": prompt}] - content.extend( - [{"type": "image_url", "image_url": {"url": url}} for url in image_urls] - ) + content.extend([{"type": "image_url", "image_url": {"url": url}} for url in image_urls]) payload = { "model": "gpt-4-vision-preview", @@ -582,9 +580,7 @@ class AIManager: "anthropic-version": "2023-06-01", } - content = [ - {"type": "image", "source": {"type": "url", "url": url}} for url in image_urls - ] + content = [{"type": "image", "source": {"type": "url", "url": url}} for url in image_urls] content.append({"type": "text", "text": prompt}) payload = {