Skip to content

Commit d6f784e

Browse files
committed
feat: ai-proxy support dashscope-finance
1 parent d609497 commit d6f784e

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

plugins/wasm-go/extensions/ai-proxy/provider/provider.go

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -188,6 +188,9 @@ type ProviderConfig struct {
188188
// @Title zh-CN 启用通义千问搜索服务
189189
// @Description zh-CN 仅适用于通义千问服务,表示是否启用通义千问的互联网搜索功能。
190190
qwenEnableSearch bool `required:"false" yaml:"qwenEnableSearch" json:"qwenEnableSearch"`
191+
// @Title zh-CN 启用通义千问金融云服务服务
192+
// @Description zh-CN 仅适用于通义千问服务,默认转发域名为 dashscope.aliyuncs.com, 当启用金融云服务时,转发域名为 dashscope-finance.aliyuncs.com
193+
qwenFinanceDomain bool `required:"false" yaml:"qwenFinanceDomain" json:"qwenFinanceDomain"`
191194
// @Title zh-CN 开启通义千问兼容模式
192195
// @Description zh-CN 启用通义千问兼容模式后,将调用千问的兼容模式接口,同时对请求/响应不做修改。
193196
qwenEnableCompatible bool `required:"false" yaml:"qwenEnableCompatible" json:"qwenEnableCompatible"`

plugins/wasm-go/extensions/ai-proxy/provider/qwen.go

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ const (
2424
qwenResultFormatMessage = "message"
2525

2626
qwenDomain = "dashscope.aliyuncs.com"
27+
qwenFinanceDomain = "dashscope-finance.aliyuncs.com"
2728
qwenChatCompletionPath = "/api/v1/services/aigc/text-generation/generation"
2829
qwenTextEmbeddingPath = "/api/v1/services/embeddings/text-embedding/text-embedding"
2930
qwenCompatiblePath = "/compatible-mode/v1/chat/completions"
@@ -64,7 +65,11 @@ type qwenProvider struct {
6465
}
6566

6667
func (m *qwenProvider) TransformRequestHeaders(ctx wrapper.HttpContext, apiName ApiName, headers http.Header, log wrapper.Log) {
67-
util.OverwriteRequestHostHeader(headers, qwenDomain)
68+
if m.config.qwenFinanceDomain {
69+
util.OverwriteRequestHostHeader(headers, qwenFinanceDomain)
70+
} else {
71+
util.OverwriteRequestHostHeader(headers, qwenDomain)
72+
}
6873
util.OverwriteRequestAuthorizationHeader(headers, "Bearer "+m.config.GetApiTokenInUse(ctx))
6974

7075
if m.config.qwenEnableCompatible {

0 commit comments

Comments
 (0)