jiazhizhong/higress
1
0
Fork 0
mirror of https://github.com/alibaba/higress.git synced 2026-03-07 10:00:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: Switch to the new HasRequestBody logic in ai-proxy (#3211)

Browse Source
This commit is contained in:
Kent Dong
2025-12-22 09:41:31 +08:00
committed by jingze
parent 0ade9504be
commit fef8ecc822
2 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -146,7 +146,7 @@ func initContext(ctx wrapper.HttpContext) {
ctx.SetContext(ctxKey, value)
}
for _, originHeader := range headerToOriginalHeaderMapping {
proxywasm.RemoveHttpRequestHeader(originHeader)
_ = proxywasm.RemoveHttpRequestHeader(originHeader)
}
originalAuth, _ := proxywasm.GetHttpRequestHeader(util.HeaderOriginalAuth)
if originalAuth == "" {
@@ -249,8 +249,8 @@ func onHttpRequestHeader(ctx wrapper.HttpContext, pluginConfig config.PluginConf
providerConfig.SetAvailableApiTokens(ctx)
// save the original request host and path in case they are needed for apiToken health check and retry
ctx.SetContext(provider.CtxRequestHost, wrapper.GetRequestHost())
ctx.SetContext(provider.CtxRequestPath, wrapper.GetRequestPath())
ctx.SetContext(provider.CtxRequestHost, ctx.Host())
ctx.SetContext(provider.CtxRequestPath, ctx.Path())
err := handler.OnRequestHeaders(ctx, apiName)
if err != nil {
@@ -258,7 +258,7 @@ func onHttpRequestHeader(ctx wrapper.HttpContext, pluginConfig config.PluginConf
return types.ActionContinue
}
hasRequestBody := wrapper.HasRequestBody()
hasRequestBody := ctx.HasRequestBody()
if hasRequestBody {
_ = proxywasm.RemoveHttpRequestHeader("Content-Length")
ctx.SetRequestBodyBufferLimit(defaultMaxBodyBytes)

4
plugins/wasm-go/extensions/ai-proxy/test/fireworks.go
View File

@@ -213,7 +213,9 @@ func RunFireworksOnHttpRequestHeadersTests(t *testing.T) {
{":method", "GET"},
})
require.Equal(t, types.ActionContinue, action)
// TODO: Due to the limitations of the test framework, we just treat it as a request with body here.
//require.Equal(t, types.ActionContinue, action)
require.Equal(t, types.HeaderStopIteration, action)
// 验证请求头处理
requestHeaders := host.GetRequestHeaders()