fix: support base64 encoded image_url (songquanpeng#872)

- Add support for base64 encoded image in OpenAI's image_url Co-authored-by: JustSong <39998050+songquanpeng@users.noreply.github.com>
buhu2k · Jan 1, 2024 · 7772064 · 7772064
1 parent c50c609
commit 7772064
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/common/image/image.go b/common/image/image.go
@@ -15,6 +15,9 @@ import (
 	_ "golang.org/x/image/webp"
 )
 
+// Regex to match data URL pattern
+var	dataURLPattern = regexp.MustCompile(`data:image/([^;]+);base64,(.*)`)
+
 func IsImageUrl(url string) (bool, error) {
 	resp, err := http.Head(url)
 	if err != nil {
@@ -44,9 +47,6 @@ func GetImageSizeFromUrl(url string) (width int, height int, err error) {
 }
 
 func GetImageFromUrl(url string) (mimeType string, data string, err error) {
-	// Regex to match data URL pattern
-	dataURLPattern := regexp.MustCompile(`data:image/([^;]+);base64,(.*)`)
-
 	// Check if the URL is a data URL
 	matches := dataURLPattern.FindStringSubmatch(url)
 	if len(matches) == 3 {