diff --git a/plugins/wasm-go/extensions/ai-proxy/Makefile b/plugins/wasm-go/extensions/ai-proxy/Makefile index e5c7fa8de..95de30517 100644 --- a/plugins/wasm-go/extensions/ai-proxy/Makefile +++ b/plugins/wasm-go/extensions/ai-proxy/Makefile @@ -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/ \ No newline at end of file + mv ai-proxy.wasm ../../../../docker-compose-test/ + +build-go: + GOOS=wasip1 GOARCH=wasm go build -buildmode=c-shared -o main.wasm main.go \ No newline at end of file diff --git a/plugins/wasm-go/extensions/ai-proxy/provider/doubao.go b/plugins/wasm-go/extensions/ai-proxy/provider/doubao.go index 4aa0509fe..8794ef91f 100644 --- a/plugins/wasm-go/extensions/ai-proxy/provider/doubao.go +++ b/plugins/wasm-go/extensions/ai-proxy/provider/doubao.go @@ -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") } diff --git a/plugins/wasm-go/extensions/ai-proxy/provider/provider.go b/plugins/wasm-go/extensions/ai-proxy/provider/provider.go index 62c4e1ec9..c77cb17cc 100644 --- a/plugins/wasm-go/extensions/ai-proxy/provider/provider.go +++ b/plugins/wasm-go/extensions/ai-proxy/provider/provider.go @@ -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 {