fix: the object should be pointer

sashabaranov · Oct 29, 2024 · e036684 · e036684
1 parent 52cdd83
commit e036684
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/chat_stream.go b/chat_stream.go
@@ -32,11 +32,11 @@ type ChatCompletionTokenLogprobTopLogprob struct {
 }
 
 type ChatCompletionStreamChoice struct {
-	Index                int                                `json:"index"`
-	Delta                ChatCompletionStreamChoiceDelta    `json:"delta"`
-	Logprobs             ChatCompletionStreamChoiceLogprobs `json:"logprobs,omitempty"`
-	FinishReason         FinishReason                       `json:"finish_reason"`
-	ContentFilterResults ContentFilterResults               `json:"content_filter_results,omitempty"`
+	Index                int                                 `json:"index"`
+	Delta                ChatCompletionStreamChoiceDelta     `json:"delta"`
+	Logprobs             *ChatCompletionStreamChoiceLogprobs `json:"logprobs,omitempty"`
+	FinishReason         FinishReason                        `json:"finish_reason"`
+	ContentFilterResults ContentFilterResults                `json:"content_filter_results,omitempty"`
 }
 
 type PromptFilterResult struct {

diff --git a/chat_stream_test.go b/chat_stream_test.go
@@ -534,7 +534,7 @@ func TestCreateChatCompletionStreamWithLogprobs(t *testing.T) {
 				{
 					Index: 0,
 					Delta: openai.ChatCompletionStreamChoiceDelta{},
-					Logprobs: openai.ChatCompletionStreamChoiceLogprobs{
+					Logprobs: &openai.ChatCompletionStreamChoiceLogprobs{
 						Content: []openai.ChatCompletionTokenLogprob{},
 					},
 				},
@@ -552,7 +552,7 @@ func TestCreateChatCompletionStreamWithLogprobs(t *testing.T) {
 					Delta: openai.ChatCompletionStreamChoiceDelta{
 						Content: "Hello",
 					},
-					Logprobs: openai.ChatCompletionStreamChoiceLogprobs{
+					Logprobs: &openai.ChatCompletionStreamChoiceLogprobs{
 						Content: []openai.ChatCompletionTokenLogprob{
 							{
 								Token:       "Hello",
@@ -577,7 +577,7 @@ func TestCreateChatCompletionStreamWithLogprobs(t *testing.T) {
 					Delta: openai.ChatCompletionStreamChoiceDelta{
 						Content: " World",
 					},
-					Logprobs: openai.ChatCompletionStreamChoiceLogprobs{
+					Logprobs: &openai.ChatCompletionStreamChoiceLogprobs{
 						Content: []openai.ChatCompletionTokenLogprob{
 							{
 								Token:       " World",