From d29b8d7ca8e095832c62d4946c6b776b6caf7ceb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=BE=84=E6=BD=AD?= Date: Mon, 6 Jan 2025 15:30:02 +0800 Subject: [PATCH] fix ai proxy checkStream (#1645) --- plugins/wasm-go/extensions/ai-proxy/main.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/plugins/wasm-go/extensions/ai-proxy/main.go b/plugins/wasm-go/extensions/ai-proxy/main.go index c08dcdb16..88b85949b 100644 --- a/plugins/wasm-go/extensions/ai-proxy/main.go +++ b/plugins/wasm-go/extensions/ai-proxy/main.go @@ -199,7 +199,7 @@ func onHttpResponseHeaders(ctx wrapper.HttpContext, pluginConfig config.PluginCo } util.ReplaceResponseHeaders(headers) - checkStream(&ctx, log) + checkStream(ctx, log) _, needHandleStreamingBody := activeProvider.(provider.StreamingResponseBodyHandler) if !needHandleStreamingBody { ctx.BufferResponseBody() @@ -254,13 +254,13 @@ func onHttpResponseBody(ctx wrapper.HttpContext, pluginConfig config.PluginConfi return types.ActionContinue } -func checkStream(ctx *wrapper.HttpContext, log wrapper.Log) { +func checkStream(ctx wrapper.HttpContext, log wrapper.Log) { contentType, err := proxywasm.GetHttpResponseHeader("Content-Type") if err != nil || !strings.HasPrefix(contentType, "text/event-stream") { if err != nil { log.Errorf("unable to load content-type header from response: %v", err) } - (*ctx).BufferResponseBody() + ctx.BufferResponseBody() ctx.SetResponseBodyBufferLimit(defaultMaxBodyBytes) } }