fix: reset buf when usage

labring · Feb 24, 2025 · 1446064 · 1446064
1 parent c345fb3
commit 1446064
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/service/aiproxy/relay/adaptor/openai/main.go b/service/aiproxy/relay/adaptor/openai/main.go
@@ -105,6 +105,7 @@ func StreamHandler(meta *meta.Meta, c *gin.Context, resp *http.Response) (*model
 			}
 			if streamResponse.Usage != nil {
 				usage = streamResponse.Usage
+				responseText.Reset()
 			}
 			for _, choice := range streamResponse.Choices {
 				if usage == nil {
@@ -139,6 +140,7 @@ func StreamHandler(meta *meta.Meta, c *gin.Context, resp *http.Response) (*model
 			}
 			if streamResponse.Usage != nil {
 				usage = streamResponse.Usage
+				responseText.Reset()
 			} else {
 				for _, choice := range streamResponse.Choices {
 					responseText.WriteString(choice.Text)