From 14460646524ca189906f519cc4e051cf8cb173c4 Mon Sep 17 00:00:00 2001 From: zijiren233 Date: Mon, 24 Feb 2025 14:03:49 +0800 Subject: [PATCH] fix: reset buf when usage --- service/aiproxy/relay/adaptor/openai/main.go | 2 ++ 1 file changed, 2 insertions(+) diff --git a/service/aiproxy/relay/adaptor/openai/main.go b/service/aiproxy/relay/adaptor/openai/main.go index 62a1875a1e85..eb5f7341ba78 100644 --- 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)