Skip to content

Commit 3c9d519

Browse files
authored
fix: fix baidu url check (songquanpeng#1143)
添加百度的另外3个向量模型【"bge-large-zh", "bge-large-en", "tao-8k", 】
1 parent 87d62f2 commit 3c9d519

File tree

2 files changed

+20
-4
lines changed

2 files changed

+20
-4
lines changed

relay/channel/baidu/adaptor.go

Lines changed: 17 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,14 +3,15 @@ package baidu
33
import (
44
"errors"
55
"fmt"
6+
"io"
7+
"net/http"
8+
"strings"
9+
610
"github.com/gin-gonic/gin"
711
"github.com/songquanpeng/one-api/relay/channel"
812
"github.com/songquanpeng/one-api/relay/constant"
913
"github.com/songquanpeng/one-api/relay/model"
1014
"github.com/songquanpeng/one-api/relay/util"
11-
"io"
12-
"net/http"
13-
"strings"
1415
)
1516

1617
type Adaptor struct {
@@ -23,7 +24,13 @@ func (a *Adaptor) Init(meta *util.RelayMeta) {
2324
func (a *Adaptor) GetRequestURL(meta *util.RelayMeta) (string, error) {
2425
// https://cloud.baidu.com/doc/WENXINWORKSHOP/s/clntwmv7t
2526
suffix := "chat/"
26-
if strings.HasPrefix("Embedding", meta.ActualModelName) {
27+
if strings.HasPrefix(meta.ActualModelName, "Embedding") {
28+
suffix = "embeddings/"
29+
}
30+
if strings.HasPrefix(meta.ActualModelName, "bge-large") {
31+
suffix = "embeddings/"
32+
}
33+
if strings.HasPrefix(meta.ActualModelName, "tao-8k") {
2734
suffix = "embeddings/"
2835
}
2936
switch meta.ActualModelName {
@@ -45,6 +52,12 @@ func (a *Adaptor) GetRequestURL(meta *util.RelayMeta) (string, error) {
4552
suffix += "bloomz_7b1"
4653
case "Embedding-V1":
4754
suffix += "embedding-v1"
55+
case "bge-large-zh":
56+
suffix += "bge_large_zh"
57+
case "bge-large-en":
58+
suffix += "bge_large_en"
59+
case "tao-8k":
60+
suffix += "tao_8k"
4861
default:
4962
suffix += meta.ActualModelName
5063
}

relay/channel/baidu/constants.go

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,4 +7,7 @@ var ModelList = []string{
77
"ERNIE-Speed",
88
"ERNIE-Bot-turbo",
99
"Embedding-V1",
10+
"bge-large-zh",
11+
"bge-large-en",
12+
"tao-8k",
1013
}

0 commit comments

Comments
 (0)