mirror of
https://github.com/linux-do/new-api.git
synced 2025-09-18 00:16:37 +08:00
fix whisper-1
This commit is contained in:
parent
81428751c7
commit
cf663e7ac4
@ -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
|
||||||
err := common.UnmarshalBodyReusable(c, &audioRequest)
|
if !strings.HasPrefix(c.Request.URL.Path, "/v1/audio/transcriptions") {
|
||||||
if err != nil {
|
err := common.UnmarshalBodyReusable(c, &audioRequest)
|
||||||
return errorWrapper(err, "bind_request_body_failed", http.StatusBadRequest)
|
if err != nil {
|
||||||
|
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 == "" {
|
||||||
|
@ -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 {
|
||||||
|
@ -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
|
||||||
|
Loading…
Reference in New Issue
Block a user