Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: Support base64 encoded image_url

Browse Source 
- Add support for base64 encoded image in OpenAI's image_url
This commit is contained in:
Laisky.Cai 2023-12-27 03:14:19 +00:00
parent f44fbe3fe7
commit e7f223a793
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
common/image/image.go
View File

@ -44,6 +44,11 @@ func GetImageSizeFromUrl(url string) (width int, height int, err error) {
} }
func GetImageFromUrl(url string) (mimeType string, data string, 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) isImage, err := IsImageUrl(url)
if !isImage { if !isImage {
return return