Merge pull request #17 from tkawachi/issue-16

Add error handling for empty response in AI chatbot
tkawachi · Mar 11, 2023 · 690f23c · 690f23c
2 parents b4b23db + a0e4bee
commit 690f23c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/aichat.go b/aichat.go
@@ -41,6 +41,9 @@ func streamCompletion(client *gogpt.Client, request gogpt.ChatCompletionRequest,
 		if err != nil {
 			return fmt.Errorf("stream recv: %w", err)
 		}
+		if len(response.Choices) == 0 {
+			return fmt.Errorf("no choices returned")
+		}
 		_, err = fmt.Fprint(out, response.Choices[0].Delta.Content)
 		if err != nil {
 			return err