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

🐛 fix: Handle empty openaiResponse.Choices in HandlerChatStream

Browse Source
This commit is contained in:
Martial BE 2024-04-27 17:17:58 +08:00
parent 58a0b6f9bc
commit c6a0c87ad1
No known key found for this signature in database
GPG Key ID: D06C32DF0EDB9084
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
providers/openai/chat.go
View File

@ -109,6 +109,11 @@ func (h *OpenAIStreamHandler) HandlerChatStream(rawLine *[]byte, dataChan chan s
return return
} }
if len(openaiResponse.Choices) == 0 {
*rawLine = nil
return
}
dataChan <- string(*rawLine) dataChan <- string(*rawLine)
if h.isAzure { if h.isAzure {