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

fix: better text phrasing

Browse Source
This commit is contained in:
ckt1031 2023-07-13 20:49:57 +08:00
parent 61912f5e2c
commit 47cb77de53
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
controller/channel-test.go
View File

@ -83,12 +83,18 @@ func testChannel(channel *model.Channel, request ChatRequest) error {
common.SysError("invalid stream response: " + data) common.SysError("invalid stream response: " + data)
continue continue
} }
if !strings.HasPrefix(data, "data:") {
continue
}
data = data[6:] data = data[6:]
if !strings.HasPrefix(data, "[DONE]") { if !strings.HasPrefix(data, "[DONE]") {
var streamResponse ChatCompletionsStreamResponse var streamResponse ChatCompletionsStreamResponse
err = json.Unmarshal([]byte(data), &streamResponse) err = json.Unmarshal([]byte(data), &streamResponse)
if err != nil { if err != nil {
common.SysError("error unmarshalling stream response: " + err.Error()) // Prinnt the body in string
buf := new(bytes.Buffer)
buf.ReadFrom(resp.Body)
common.SysError("error unmarshalling stream response: " + err.Error() + " " + buf.String())
return err return err
} }
for _, choice := range streamResponse.Choices { for _, choice := range streamResponse.Choices {

3
controller/relay-text.go
View File

@ -278,6 +278,9 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
common.SysError("invalid stream response: " + data) common.SysError("invalid stream response: " + data)
continue continue
} }
if !strings.HasPrefix(data, "data:") {
continue
}
dataChan <- data dataChan <- data
data = data[6:] data = data[6:]
if !strings.HasPrefix(data, "[DONE]") { if !strings.HasPrefix(data, "[DONE]") {