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

fix: testing issue

Browse Source
This commit is contained in:
ckt1031 2023-07-16 16:01:52 +08:00
parent 481c4ebf49
commit edd2c4f6e9
1 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
controller/channel-test.go
View File

@ -130,7 +130,6 @@ func testChannel(channel *model.Channel, request ChatRequest) error {
return errors.New("error response: " + strconv.Itoa(resp.StatusCode)) return errors.New("error response: " + strconv.Itoa(resp.StatusCode))
} }
var done = false
var streamResponseText = "" var streamResponseText = ""
scanner := bufio.NewScanner(resp.Body) scanner := bufio.NewScanner(resp.Body)
@ -158,8 +157,8 @@ func testChannel(channel *model.Channel, request ChatRequest) error {
if len(data) < 6 { // must be something wrong! if len(data) < 6 { // must be something wrong!
continue continue
} }
if channel.Type != common.ChannelTypeChatGPTWeb {
if channel.Type != common.ChannelTypeChatGPTWeb {
// If data has event: event content inside, remove it, it can be prefix or inside the data // If data has event: event content inside, remove it, it can be prefix or inside the data
if strings.HasPrefix(data, "event:") || strings.Contains(data, "event:") { if strings.HasPrefix(data, "event:") || strings.Contains(data, "event:") {
// Remove event: event in the front or back // Remove event: event in the front or back
@ -196,9 +195,6 @@ 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
} }
} else if channel.Type == common.ChannelTypeChatGPTWeb { } else if channel.Type == common.ChannelTypeChatGPTWeb {
@ -225,7 +221,7 @@ func testChannel(channel *model.Channel, request ChatRequest) error {
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 == "" || !done && channel.Type != common.ChannelTypeChatGPTWeb { if streamResponseText == "" {
return errors.New("Streaming not complete") return errors.New("Streaming not complete")
} }
@ -239,7 +235,7 @@ func buildTestRequest() *ChatRequest {
} }
testMessage := Message{ testMessage := Message{
Role: "user", Role: "user",
Content: "say hi word only", Content: "Hello ChatGPT!",
} }
testRequest.Messages = append(testRequest.Messages, testMessage) testRequest.Messages = append(testRequest.Messages, testMessage)
return testRequest return testRequest