瀏覽代碼

fix: fix array index not checked (close #588)

JustSong 2 年之前
父節點
當前提交
a27a5bcc06
共有 1 個文件被更改,包括 3 次插入0 次删除
  1. 3 0
      controller/relay-xunfei.go

+ 3 - 0
controller/relay-xunfei.go

@@ -220,6 +220,9 @@ func xunfeiHandler(c *gin.Context, textRequest GeneralOpenAIRequest, appId strin
 	for !stop {
 	for !stop {
 		select {
 		select {
 		case xunfeiResponse = <-dataChan:
 		case xunfeiResponse = <-dataChan:
+			if len(xunfeiResponse.Payload.Choices.Text) == 0 {
+				continue
+			}
 			content += xunfeiResponse.Payload.Choices.Text[0].Content
 			content += xunfeiResponse.Payload.Choices.Text[0].Content
 			usage.PromptTokens += xunfeiResponse.Payload.Usage.Text.PromptTokens
 			usage.PromptTokens += xunfeiResponse.Payload.Usage.Text.PromptTokens
 			usage.CompletionTokens += xunfeiResponse.Payload.Usage.Text.CompletionTokens
 			usage.CompletionTokens += xunfeiResponse.Payload.Usage.Text.CompletionTokens