Fix index out of range

sunshineplan · May 6, 2024 · 6c6b2d8 · 6c6b2d8
1 parent 76e0db1
commit 6c6b2d8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/chatgpt/chatgpt.go b/chatgpt/chatgpt.go
@@ -200,7 +200,7 @@ func (stream *ChatStream) Next() (ai.ChatResponse, error) {
 		stream.merged = ""
 		return nil, err
 	}
-	if stream.cs != nil {
+	if stream.cs != nil && len(resp.Choices) > 0 {
 		stream.merged += resp.Choices[0].Delta.Content
 	}
 	return &ChatResponse[openai.ChatCompletionStreamResponse]{resp}, nil
@@ -256,7 +256,9 @@ func (session *ChatSession) Chat(ctx context.Context, messages ...string) (ai.Ch
 		return nil, err
 	}
 	addToHistory(&session.history, openai.ChatMessageRoleUser, messages...)
-	session.history = append(session.history, resp.Choices[0].Message)
+	if len(resp.Choices) > 0 {
+		session.history = append(session.history, resp.Choices[0].Message)
+	}
 	return &ChatResponse[openai.ChatCompletionResponse]{resp}, nil
 }