From e4709dcceb1bfc123aaa6379051a2a4a2f58fa2c Mon Sep 17 00:00:00 2001 From: Jeffrey <572199+darkthread@users.noreply.github.com> Date: Wed, 6 Mar 2024 00:07:00 +0800 Subject: [PATCH 1/2] Check if response has a choice before debug logging to prevent null reference. --- pkg/openai/client.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pkg/openai/client.go b/pkg/openai/client.go index 1590f29c..477e5f69 100644 --- a/pkg/openai/client.go +++ b/pkg/openai/client.go @@ -422,7 +422,9 @@ func (c *Client) call(ctx context.Context, request openai.ChatCompletionRequest, } else if err != nil { return nil, err } - slog.Debug("stream", "content", response.Choices[0].Delta.Content) + if response.Choices != nil && len(response.Choices) > 0 { + slog.Debug("stream", "content", response.Choices[0].Delta.Content) + } if partial != nil { partialMessage = appendMessage(partialMessage, response) partial <- types.CompletionStatus{ From b45a4c94777bcdb932c044c0b10ff1af4919a686 Mon Sep 17 00:00:00 2001 From: Jeffrey Lee <572199+darkthread@users.noreply.github.com> Date: Wed, 6 Mar 2024 22:08:04 +0800 Subject: [PATCH 2/2] Update pkg/openai/client.go Co-authored-by: Nick Hale <4175918+njhale@users.noreply.github.com> Signed-off-by: Jeffrey Lee <572199+darkthread@users.noreply.github.com> --- pkg/openai/client.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pkg/openai/client.go b/pkg/openai/client.go index 477e5f69..ee0d827c 100644 --- a/pkg/openai/client.go +++ b/pkg/openai/client.go @@ -422,7 +422,7 @@ func (c *Client) call(ctx context.Context, request openai.ChatCompletionRequest, } else if err != nil { return nil, err } - if response.Choices != nil && len(response.Choices) > 0 { + if len(response.Choices) > 0 { slog.Debug("stream", "content", response.Choices[0].Delta.Content) } if partial != nil {