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

fix: ai-proxy dify provider extract hostname from difyApiUrl (#3257)

Browse Source
This commit is contained in:
Bingkun Zhao
2025-12-24 09:58:54 +08:00
committed by jingze
parent f71c1900a8
commit 8024a96881
1 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
plugins/wasm-go/extensions/ai-proxy/provider/dify.go
View File

@@ -5,6 +5,7 @@ import (
"errors"
"fmt"
"net/http"
"net/url"
"strings"
"github.com/higress-group/proxy-wasm-go-sdk/proxywasm"
@@ -59,7 +60,18 @@ func (d *difyProvider) OnRequestHeaders(ctx wrapper.HttpContext, apiName ApiName
func (d *difyProvider) TransformRequestHeaders(ctx wrapper.HttpContext, apiName ApiName, headers http.Header) {
if d.config.difyApiUrl != "" {
log.Debugf("use local host: %s", d.config.difyApiUrl)
util.OverwriteRequestHostHeader(headers, d.config.difyApiUrl)
// Extract hostname, including Full URL or Domain
host := d.config.difyApiUrl
if parsedUrl, err := url.Parse(d.config.difyApiUrl); err == nil && parsedUrl.Host != "" {
host = parsedUrl.Host
} else {
host = strings.TrimPrefix(strings.TrimPrefix(d.config.difyApiUrl, "http://"), "https://")
if idx := strings.Index(host, "/"); idx != -1 {
host = host[:idx]
}
}
log.Debugf("extracted hostname: %s", host)
util.OverwriteRequestHostHeader(headers, host)
} else {
util.OverwriteRequestHostHeader(headers, difyDomain)
}