doubao support configuration for domain (#3184)

This commit is contained in:
rinfx
2025-12-04 21:17:50 +08:00
committed by GitHub
parent 896bcacf4c
commit 8a7a375ebd
3 changed files with 13 additions and 2 deletions

View File

@@ -1,4 +1,7 @@
.DEFAULT:
build:
tinygo build -o ai-proxy.wasm -scheduler=none -target=wasi -gc=custom -tags='custommalloc nottinygc_finalizer proxy_wasm_version_0_2_100' ./main.go
mv ai-proxy.wasm ../../../../docker-compose-test/
mv ai-proxy.wasm ../../../../docker-compose-test/
build-go:
GOOS=wasip1 GOARCH=wasm go build -buildmode=c-shared -o main.wasm main.go

View File

@@ -70,7 +70,11 @@ func (m *doubaoProvider) OnRequestBody(ctx wrapper.HttpContext, apiName ApiName,
func (m *doubaoProvider) TransformRequestHeaders(ctx wrapper.HttpContext, apiName ApiName, headers http.Header) {
util.OverwriteRequestPathHeaderByCapability(headers, string(apiName), m.config.capabilities)
util.OverwriteRequestHostHeader(headers, doubaoDomain)
if m.config.doubaoDomain != "" {
util.OverwriteRequestHostHeader(headers, m.config.doubaoDomain)
} else {
util.OverwriteRequestHostHeader(headers, doubaoDomain)
}
util.OverwriteRequestAuthorizationHeader(headers, "Bearer "+m.config.GetApiTokenInUse(ctx))
headers.Del("Content-Length")
}

View File

@@ -429,6 +429,9 @@ type ProviderConfig struct {
// @Title zh-CN vLLM主机地址
// @Description zh-CN 仅适用于vLLM服务指定vLLM服务器的主机地址例如vllm-service.cluster.local
vllmServerHost string `required:"false" yaml:"vllmServerHost" json:"vllmServerHost"`
// @Title zh-CN 豆包服务域名
// @Description zh-CN 仅适用于豆包服务,默认转发域名为 ark.cn-beijing.volces.com
doubaoDomain string `required:"false" yaml:"doubaoDomain" json:"doubaoDomain"`
}
func (c *ProviderConfig) GetId() string {
@@ -627,6 +630,7 @@ func (c *ProviderConfig) FromJson(json gjson.Result) {
c.genericHost = json.Get("genericHost").String()
c.vllmServerHost = json.Get("vllmServerHost").String()
c.vllmCustomUrl = json.Get("vllmCustomUrl").String()
c.doubaoDomain = json.Get("doubaoDomain").String()
}
func (c *ProviderConfig) Validate() error {