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

fix: fix xunfei crash (#451)

Browse Source
This commit is contained in:
glzjin 2023-08-20 22:07:50 +08:00 committed by GitHub
parent 05e4f2b439
commit efeb9a16ce
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
controller/relay-text.go
View File

@ -315,12 +315,12 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
return errorWrapper(err, "close_request_body_failed", http.StatusInternalServerError) return errorWrapper(err, "close_request_body_failed", http.StatusInternalServerError)
} }
isStream = isStream || strings.HasPrefix(resp.Header.Get("Content-Type"), "text/event-stream") isStream = isStream || strings.HasPrefix(resp.Header.Get("Content-Type"), "text/event-stream")
}
if resp.StatusCode != http.StatusOK { if resp.StatusCode != http.StatusOK {
return errorWrapper( return errorWrapper(
fmt.Errorf("bad status code: %d", resp.StatusCode), "bad_status_code", resp.StatusCode) fmt.Errorf("bad status code: %d", resp.StatusCode), "bad_status_code", resp.StatusCode)
} }
}
var textResponse TextResponse var textResponse TextResponse
tokenName := c.GetString("token_name") tokenName := c.GetString("token_name")