From e7f223a7932c525c04012691477b9b99e06ec5ae Mon Sep 17 00:00:00 2001 From: "Laisky.Cai" Date: Wed, 27 Dec 2023 03:14:19 +0000 Subject: [PATCH] fix: Support base64 encoded image_url - Add support for base64 encoded image in OpenAI's image_url --- common/image/image.go | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/common/image/image.go b/common/image/image.go index a602936a..eae76286 100644 --- a/common/image/image.go +++ b/common/image/image.go @@ -44,6 +44,11 @@ func GetImageSizeFromUrl(url string) (width int, height int, err error) { } func GetImageFromUrl(url string) (mimeType string, data string, err error) { + // openai's image_url support base64 encoded image + if strings.HasPrefix(url, "data:image/jpeg;base64,") { + return "image/jpeg", strings.TrimPrefix(url, "data:image/jpeg;base64,"), nil + } + isImage, err := IsImageUrl(url) if !isImage { return