OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-06-26 18:15:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #126 from justinh-rahb/anthropic-image-fix

Browse Source 
fix: anthropic pipeline image upload
This commit is contained in:
Timothy Jaeryang Baek 2024-06-26 23:11:45 -07:00 committed by GitHub
commit 8163c34855
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
examples/pipelines/providers/anthropic_manifold_pipeline.py
View File

@ -2,7 +2,7 @@
title: Anthropic Manifold Pipeline title: Anthropic Manifold Pipeline
author: justinh-rahb author: justinh-rahb
date: 2024-06-20 date: 2024-06-20
version: 1.2 version: 1.3
license: MIT license: MIT
description: A pipeline for generating text and processing images using the Anthropic API. description: A pipeline for generating text and processing images using the Anthropic API.
requirements: requests, anthropic requirements: requests, anthropic
@ -58,8 +58,8 @@ class Pipeline:
return self.get_anthropic_models() return self.get_anthropic_models()
def process_image(self, image_data): def process_image(self, image_data):
if image_data["image_url"]["url"].startswith("data:image"): if image_data["url"].startswith("data:image"):
mime_type, base64_data = image_data["image_url"]["url"].split(",", 1) mime_type, base64_data = image_data["url"].split(",", 1)
media_type = mime_type.split(":")[1].split(";")[0] media_type = mime_type.split(":")[1].split(";")[0]
return { return {
"type": "image", "type": "image",
@ -72,7 +72,7 @@ class Pipeline:
else: else:
return { return {
"type": "image", "type": "image",
"source": {"type": "url", "url": image_data["image_url"]["url"]}, "source": {"type": "url", "url": image_data["url"]},
} }
def pipe( def pipe(
@ -99,7 +99,7 @@ class Pipeline:
if image_count >= 5: if image_count >= 5:
raise ValueError("Maximum of 5 images per API call exceeded") raise ValueError("Maximum of 5 images per API call exceeded")
processed_image = self.process_image(item) processed_image = self.process_image(item["image_url"])
processed_content.append(processed_image) processed_content.append(processed_image)
if processed_image["source"]["type"] == "base64": if processed_image["source"]["type"] == "base64":