refactor: ChatCompletionStreamChoice.FinishReason from string to FinishReason

liushuangls · liushuangls · commit ff19a8b9020b · 2023-06-15T17:47:36.000+08:00
diff --git a/chat_stream.go b/chat_stream.go
@@ -15,7 +15,7 @@ type ChatCompletionStreamChoiceDelta struct {
 type ChatCompletionStreamChoice struct {
 	Index        int                             `json:"index"`
 	Delta        ChatCompletionStreamChoiceDelta `json:"delta"`
-	FinishReason string                          `json:"finish_reason"`
+	FinishReason FinishReason                    `json:"finish_reason"`
 }
 
 type ChatCompletionStreamResponse struct {

Original file line number	Diff line number	Diff line change
`@@ -15,7 +15,7 @@ type ChatCompletionStreamChoiceDelta struct {`
`15`	`15`	`type ChatCompletionStreamChoice struct {`
`16`	`16`	Index int `json:"index"`
`17`	`17`	Delta ChatCompletionStreamChoiceDelta `json:"delta"`
`18`		- FinishReason string `json:"finish_reason"`
	`18`	+ FinishReason FinishReason `json:"finish_reason"`
`19`	`19`	`}`
`20`	`20`
`21`	`21`	`type ChatCompletionStreamResponse struct {`