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

support gemini & claude domain setting (#3638)

Browse Source
This commit is contained in:
rinfx
2026-03-26 11:12:30 +08:00
committed by GitHub
parent 3fc01913cf
commit 231ba1cd23
4 changed files with 71 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -62,11 +62,12 @@ func (g *geminiProviderInitializer) DefaultCapabilities() map[string]string {
func (g *geminiProviderInitializer) CreateProvider(config ProviderConfig) (Provider, error) {
config.setDefaultCapabilities(g.DefaultCapabilities())
domain := config.resolveDomain("", geminiDomain)
return &geminiProvider{
config: config,
contextCache: createContextCache(&config),
client: wrapper.NewClusterClient(wrapper.RouteCluster{
Host: geminiDomain,
Host: domain,
}),
}, nil
}
@@ -89,7 +90,8 @@ func (g *geminiProvider) OnRequestHeaders(ctx wrapper.HttpContext, apiName ApiNa
}
func (g *geminiProvider) TransformRequestHeaders(ctx wrapper.HttpContext, apiName ApiName, headers http.Header) {
util.OverwriteRequestHostHeader(headers, geminiDomain)
domain := g.config.resolveDomain("", geminiDomain)
util.OverwriteRequestHostHeader(headers, domain)
headers.Set(geminiApiKeyHeader, g.config.GetApiTokenInUse(ctx))
util.OverwriteRequestAuthorizationHeader(headers, "")
}