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

feat(provider): 优化 Azure multipart 处理 || feat(provider): Optimize Azure multipart processing (#3651)

Browse Source
This commit is contained in:
woody
2026-03-30 13:45:41 +08:00
committed by GitHub
parent 83461887dc
commit 889ea67013
6 changed files with 756 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
plugins/wasm-go/extensions/ai-proxy/provider/azure.go
View File

@@ -9,6 +9,7 @@ import (
"strings"
"github.com/alibaba/higress/plugins/wasm-go/extensions/ai-proxy/util"
"github.com/higress-group/proxy-wasm-go-sdk/proxywasm"
"github.com/higress-group/proxy-wasm-go-sdk/proxywasm/types"
"github.com/higress-group/wasm-go/pkg/log"
"github.com/higress-group/wasm-go/pkg/wrapper"
@@ -151,17 +152,44 @@ func (m *azureProvider) OnRequestBody(ctx wrapper.HttpContext, apiName ApiName,
return m.config.handleRequestBody(m, m.contextCache, ctx, apiName, body)
}
func isAzureMultipartImageRequest(apiName ApiName, contentType string) bool {
if apiName != ApiNameImageEdit && apiName != ApiNameImageVariation {
return false
}
return isMultipartFormData(contentType)
}
func (m *azureProvider) TransformRequestBody(ctx wrapper.HttpContext, apiName ApiName, body []byte) (transformedBody []byte, err error) {
transformedBody = body
err = nil
contentType, _ := proxywasm.GetHttpRequestHeader(util.HeaderContentType)
isMultipartImageRequest := isAzureMultipartImageRequest(apiName, contentType)
transformedBody, err = m.config.defaultTransformRequestBody(ctx, apiName, body)
if isMultipartImageRequest {
if err != nil {
log.Debugf("[azure multipart] body transform failed: api=%s, err=%v", apiName, err)
} else {
log.Debugf("[azure multipart] body transformed: api=%s, originalModel=%s, mappedModel=%s, bodyBytes=%d->%d",
apiName,
ctx.GetStringContext(ctxKeyOriginalRequestModel, ""),
ctx.GetStringContext(ctxKeyFinalRequestModel, ""),
len(body),
len(transformedBody),
)
}
}
if err != nil {
return
}
// This must be called after the body is transformed, because it uses the model from the context filled by that call.
if path := m.transformRequestPath(ctx, apiName); path != "" {
if isMultipartImageRequest {
log.Debugf("[azure multipart] body path overwrite: api=%s, path=%s, modelInContext=%s",
apiName, path, ctx.GetStringContext(ctxKeyFinalRequestModel, ""))
}
err = util.OverwriteRequestPath(path)
if err == nil {
log.Debugf("azureProvider: overwrite request path to %s succeeded", path)
@@ -222,16 +250,30 @@ func (m *azureProvider) transformRequestPath(ctx wrapper.HttpContext, apiName Ap
}
func (m *azureProvider) TransformRequestHeaders(ctx wrapper.HttpContext, apiName ApiName, headers http.Header) {
contentType := headers.Get(util.HeaderContentType)
isMultipartImageRequest := isAzureMultipartImageRequest(apiName, contentType)
// We need to overwrite the request path in the request headers stage,
// because for some APIs, we don't read the request body and the path is model irrelevant.
if overwrittenPath := m.transformRequestPath(ctx, apiName); overwrittenPath != "" {
util.OverwriteRequestPathHeader(headers, overwrittenPath)
if isMultipartImageRequest {
log.Debugf("[azure multipart] header path overwrite: api=%s, path=%s, modelInContext=%s",
apiName, overwrittenPath, ctx.GetStringContext(ctxKeyFinalRequestModel, ""))
}
}
util.OverwriteRequestHostHeader(headers, m.serviceUrl.Host)
headers.Set("api-key", m.config.GetApiTokenInUse(ctx))
headers.Del("Content-Length")
if !m.config.isSupportedAPI(apiName) || !m.config.needToProcessRequestBody(apiName) {
supportedAPI := m.config.isSupportedAPI(apiName)
needProcessBody := m.config.needToProcessRequestBody(apiName)
if isMultipartImageRequest {
log.Debugf("[azure multipart] body processing decision: api=%s, supported=%t, needProcessBody=%t",
apiName, supportedAPI, needProcessBody)
}
if !supportedAPI || !needProcessBody {
// If the API is not supported or there is no need to process the body,
// we should not read the request body and keep it as it is.
ctx.DontReadRequestBody()