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

fix: patch testing

Browse Source
This commit is contained in:
ckt1031 2023-07-13 19:40:36 +08:00
parent 379d03798c
commit 61912f5e2c
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
controller/channel-test.go
View File

@ -58,7 +58,8 @@ func testChannel(channel *model.Channel, request ChatRequest) error {
return errors.New("invalid status code: " + strconv.Itoa(resp.StatusCode))
}
var streamResponseText string
var done = false
var streamResponseText = ""
scanner := bufio.NewScanner(resp.Body)
scanner.Split(func(data []byte, atEOF bool) (advance int, token []byte, err error) {
@ -93,13 +94,16 @@ func testChannel(channel *model.Channel, request ChatRequest) error {
for _, choice := range streamResponse.Choices {
streamResponseText += choice.Delta.Content
}
} else {
done = true
break
}
}
defer resp.Body.Close()
// Check if streaming is complete and streamResponseText is populated
if streamResponseText == "" {
if streamResponseText == "" || !done {
return errors.New("Streaming not complete")
}