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)) return errors.New("invalid status code: " + strconv.Itoa(resp.StatusCode))
} }
var streamResponseText string var done = false
var streamResponseText = ""
scanner := bufio.NewScanner(resp.Body) scanner := bufio.NewScanner(resp.Body)
scanner.Split(func(data []byte, atEOF bool) (advance int, token []byte, err error) { 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 { for _, choice := range streamResponse.Choices {
streamResponseText += choice.Delta.Content streamResponseText += choice.Delta.Content
} }
} else {
done = true
break
} }
} }
defer resp.Body.Close() defer resp.Body.Close()
// Check if streaming is complete and streamResponseText is populated // Check if streaming is complete and streamResponseText is populated
if streamResponseText == "" { if streamResponseText == "" || !done {
return errors.New("Streaming not complete") return errors.New("Streaming not complete")
} }