jiazhizhong/higress
1
0
Fork 0
mirror of https://github.com/alibaba/higress.git synced 2026-06-08 20:27:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix ai proxy checkStream (#1645)

Browse Source
This commit is contained in:
澄潭
2025-01-06 15:30:02 +08:00
committed by GitHub
parent 2501895b66
commit d29b8d7ca8
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
plugins/wasm-go/extensions/ai-proxy/main.go
View File

@@ -199,7 +199,7 @@ func onHttpResponseHeaders(ctx wrapper.HttpContext, pluginConfig config.PluginCo
} }
util.ReplaceResponseHeaders(headers) util.ReplaceResponseHeaders(headers)
checkStream(&ctx, log) checkStream(ctx, log)
_, needHandleStreamingBody := activeProvider.(provider.StreamingResponseBodyHandler) _, needHandleStreamingBody := activeProvider.(provider.StreamingResponseBodyHandler)
if !needHandleStreamingBody { if !needHandleStreamingBody {
ctx.BufferResponseBody() ctx.BufferResponseBody()
@@ -254,13 +254,13 @@ func onHttpResponseBody(ctx wrapper.HttpContext, pluginConfig config.PluginConfi
return types.ActionContinue 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") contentType, err := proxywasm.GetHttpResponseHeader("Content-Type")
if err != nil || !strings.HasPrefix(contentType, "text/event-stream") { if err != nil || !strings.HasPrefix(contentType, "text/event-stream") {
if err != nil { if err != nil {
log.Errorf("unable to load content-type header from response: %v", err) log.Errorf("unable to load content-type header from response: %v", err)
} }
(*ctx).BufferResponseBody() ctx.BufferResponseBody()
ctx.SetResponseBodyBufferLimit(defaultMaxBodyBytes) ctx.SetResponseBodyBufferLimit(defaultMaxBodyBytes)
} }
} }