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

Fix GetImageSizeFromUrl for image url without Content-Type

Browse Source
This commit is contained in:
WqyJh 2024-02-27 20:28:07 +08:00
parent b747cdbc6f
commit 052092391d
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
common/image/image.go
View File

@ -16,7 +16,7 @@ import (
) )
// Regex to match data URL pattern // Regex to match data URL pattern
var dataURLPattern = regexp.MustCompile(`data:image/([^;]+);base64,(.*)`) var dataURLPattern = regexp.MustCompile(`data:image/([^;]+);base64,(.*)`)
func IsImageUrl(url string) (bool, error) { func IsImageUrl(url string) (bool, error) {
resp, err := http.Head(url) resp, err := http.Head(url)
@ -30,10 +30,6 @@ func IsImageUrl(url string) (bool, error) {
} }
func GetImageSizeFromUrl(url string) (width int, height int, err error) { func GetImageSizeFromUrl(url string) (width int, height int, err error) {
isImage, err := IsImageUrl(url)
if !isImage {
return
}
resp, err := http.Get(url) resp, err := http.Get(url)
if err != nil { if err != nil {
return return