fix whisper-1

This commit is contained in:
CaIon
2023-11-19 16:35:37 +08:00
parent 81428751c7
commit cf663e7ac4
3 changed files with 14 additions and 6 deletions

View File

@@ -82,11 +82,11 @@ func getImageToken(imageUrl MessageImageUrl) (int, error) {
return 0, err
}
defer response.Body.Close()
// 限制读取的字节数,防止下载整个图片
limitReader := io.LimitReader(response.Body, 8192)
response.Body.Close()
// 读取图片的头部信息来获取图片尺寸
config, _, err := image.DecodeConfig(limitReader)
if err != nil {