From 137c56812957c1558ce665abed552311de9d7ceb Mon Sep 17 00:00:00 2001 From: Matt Low Date: Sat, 25 Nov 2023 01:26:37 +0000 Subject: [PATCH] Minor cleanup --- pkg/cli/cmd.go | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/pkg/cli/cmd.go b/pkg/cli/cmd.go index befb794..d30cef2 100644 --- a/pkg/cli/cmd.go +++ b/pkg/cli/cmd.go @@ -64,12 +64,11 @@ func LLMRequest(messages []Message) (string, error) { go HandleDelayedContent(receiver) response, err := CreateChatCompletionStream(model, messages, maxTokens, receiver) - if err != nil && response != "" { - Warn("Received partial response. Error: %v\n", err) - err = nil // ignore partial response error - } - if response != "" { + if err != nil { + Warn("Received partial response. Error: %v\n", err) + err = nil + } // there was some content, so break to a new line after it fmt.Println() }