Selaa lähdekoodia

Return empty choices on usage chunk

Anuraag Agrawal 7 kuukautta sitten
vanhempi
commit
7355ab3703
1 muutettua tiedostoa jossa 2 lisäystä ja 2 poistoa
  1. 2 2
      openai/openai.go

+ 2 - 2
openai/openai.go

@@ -647,7 +647,7 @@ func (w *ChatWriter) writeResponse(data []byte) (int, error) {
 		if chatResponse.Done {
 		if chatResponse.Done {
 			if w.streamUsage {
 			if w.streamUsage {
 				u := toUsage(chatResponse)
 				u := toUsage(chatResponse)
-				d, err := json.Marshal(ChatCompletionChunk{Usage: &u})
+				d, err := json.Marshal(ChatCompletionChunk{Choices: []ChunkChoice{}, Usage: &u})
 				if err != nil {
 				if err != nil {
 					return 0, err
 					return 0, err
 				}
 				}
@@ -711,7 +711,7 @@ func (w *CompleteWriter) writeResponse(data []byte) (int, error) {
 		if generateResponse.Done {
 		if generateResponse.Done {
 			if w.streamUsage {
 			if w.streamUsage {
 				u := toUsageGenerate(generateResponse)
 				u := toUsageGenerate(generateResponse)
-				d, err := json.Marshal(CompletionChunk{Usage: &u})
+				d, err := json.Marshal(CompletionChunk{Choices: []CompleteChunkChoice{}, Usage: &u})
 				if err != nil {
 				if err != nil {
 					return 0, err
 					return 0, err
 				}
 				}