fix: Support base64 encoded image_url

- Add support for base64 encoded image in OpenAI's image_url
2023-12-27 03:14:19 +00:00 · 2023-12-27 03:14:19 +00:00 · e7f223a793
commit e7f223a793
parent f44fbe3fe7
1 changed files with 5 additions and 0 deletions
--- 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