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

@ -32,10 +32,17 @@ func relayAudioHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode
group := c.GetString("group") group := c.GetString("group")
var audioRequest AudioRequest var audioRequest AudioRequest
if !strings.HasPrefix(c.Request.URL.Path, "/v1/audio/transcriptions") {
err := common.UnmarshalBodyReusable(c, &audioRequest) err := common.UnmarshalBodyReusable(c, &audioRequest)
if err != nil { if err != nil {
return errorWrapper(err, "bind_request_body_failed", http.StatusBadRequest) return errorWrapper(err, "bind_request_body_failed", http.StatusBadRequest)
} }
} else {
audioRequest = AudioRequest{
Model: "whisper-1",
}
}
//err := common.UnmarshalBodyReusable(c, &audioRequest)
// request validation // request validation
if audioRequest.Model == "" { if audioRequest.Model == "" {

View File

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

View File

@ -46,8 +46,9 @@ func Distribute() func(c *gin.Context) {
if modelRequest.Model == "" { if modelRequest.Model == "" {
modelRequest.Model = "midjourney" modelRequest.Model = "midjourney"
} }
} } else if !strings.HasPrefix(c.Request.URL.Path, "/v1/audio/transcriptions") {
err = common.UnmarshalBodyReusable(c, &modelRequest) err = common.UnmarshalBodyReusable(c, &modelRequest)
}
if err != nil { if err != nil {
abortWithMessage(c, http.StatusBadRequest, "无效的请求") abortWithMessage(c, http.StatusBadRequest, "无效的请求")
return return