From a27a5bcc06aa18bf70034a8cddf1874aa92277b9 Mon Sep 17 00:00:00 2001 From: JustSong Date: Sat, 14 Oct 2023 16:11:15 +0800 Subject: [PATCH] fix: fix array index not checked (close #588) --- controller/relay-xunfei.go | 3 +++ 1 file changed, 3 insertions(+) diff --git a/controller/relay-xunfei.go b/controller/relay-xunfei.go index ff6bf065..cbaf38fe 100644 --- a/controller/relay-xunfei.go +++ b/controller/relay-xunfei.go @@ -220,6 +220,9 @@ func xunfeiHandler(c *gin.Context, textRequest GeneralOpenAIRequest, appId strin for !stop { select { case xunfeiResponse = <-dataChan: + if len(xunfeiResponse.Payload.Choices.Text) == 0 { + continue + } content += xunfeiResponse.Payload.Choices.Text[0].Content usage.PromptTokens += xunfeiResponse.Payload.Usage.Text.PromptTokens usage.CompletionTokens += xunfeiResponse.Payload.Usage.Text.CompletionTokens