From c2a15f6aa130732ff5b43a9c1dde5f270f81f79c Mon Sep 17 00:00:00 2001 From: Eugen Eisler Date: Wed, 18 Sep 2024 14:08:09 +0200 Subject: [PATCH] chore: #975 check choices available --- vendors/openai/openai.go | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/vendors/openai/openai.go b/vendors/openai/openai.go index cc70302..88997f4 100644 --- a/vendors/openai/openai.go +++ b/vendors/openai/openai.go @@ -82,7 +82,12 @@ func (o *Client) SendStream( for { var response openai.ChatCompletionStreamResponse if response, err = stream.Recv(); err == nil { - channel <- response.Choices[0].Delta.Content + if len(response.Choices) > 0 { + channel <- response.Choices[0].Delta.Content + } else { + fmt.Printf("No response (choices) from stream\n") + break + } } else if errors.Is(err, io.EOF) { channel <- "\n" close(channel)