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

fix(ai-proxy): Avoid overwriting the existing original auth header (#2904)

Browse Source
This commit is contained in:
澄潭
2025-09-11 16:54:26 +08:00
committed by GitHub
parent 28228edfe5
commit 89d414e49a
1 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -39,14 +39,12 @@ type pair[K, V any] struct {
var ( var (
headersCtxKeyMapping = map[string]string{ headersCtxKeyMapping = map[string]string{
util.HeaderAuthority: ctxOriginalHost, util.HeaderAuthority: ctxOriginalHost,
util.HeaderPath: ctxOriginalPath, util.HeaderPath: ctxOriginalPath,
util.HeaderAuthorization: ctxOriginalAuth,
} }
headerToOriginalHeaderMapping = map[string]string{ headerToOriginalHeaderMapping = map[string]string{
util.HeaderAuthority: util.HeaderOriginalHost, util.HeaderAuthority: util.HeaderOriginalHost,
util.HeaderPath: util.HeaderOriginalPath, util.HeaderPath: util.HeaderOriginalPath,
util.HeaderAuthorization: util.HeaderOriginalAuth,
} }
pathSuffixToApiName = []pair[string, provider.ApiName]{ pathSuffixToApiName = []pair[string, provider.ApiName]{
// OpenAI style // OpenAI style
@@ -144,6 +142,11 @@ func initContext(ctx wrapper.HttpContext) {
for _, originHeader := range headerToOriginalHeaderMapping { for _, originHeader := range headerToOriginalHeaderMapping {
proxywasm.RemoveHttpRequestHeader(originHeader) proxywasm.RemoveHttpRequestHeader(originHeader)
} }
originalAuth, _ := proxywasm.GetHttpRequestHeader(util.HeaderOriginalAuth)
if originalAuth == "" {
value, _ := proxywasm.GetHttpRequestHeader(util.HeaderAuthorization)
ctx.SetContext(ctxOriginalAuth, value)
}
} }
func saveContextsToHeaders(ctx wrapper.HttpContext) { func saveContextsToHeaders(ctx wrapper.HttpContext) {
@@ -161,6 +164,10 @@ func saveContextsToHeaders(ctx wrapper.HttpContext) {
_ = proxywasm.ReplaceHttpRequestHeader(originalHeader, originalValue) _ = proxywasm.ReplaceHttpRequestHeader(originalHeader, originalValue)
} }
} }
originalValue := ctx.GetStringContext(ctxOriginalAuth, "")
if originalValue != "" {
_ = proxywasm.ReplaceHttpRequestHeader(util.HeaderOriginalAuth, originalValue)
}
} }
func onHttpRequestHeader(ctx wrapper.HttpContext, pluginConfig config.PluginConfig) types.Action { func onHttpRequestHeader(ctx wrapper.HttpContext, pluginConfig config.PluginConfig) types.Action {