fix: fix baidu url check (#1143)
添加百度的另外3个向量模型【"bge-large-zh", "bge-large-en", "tao-8k", 】
This commit is contained in:
parent
8ede66a896
commit
66efabd5ae
@ -3,14 +3,15 @@ package baidu
|
|||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
"github.com/songquanpeng/one-api/relay/channel"
|
"github.com/songquanpeng/one-api/relay/channel"
|
||||||
"github.com/songquanpeng/one-api/relay/constant"
|
"github.com/songquanpeng/one-api/relay/constant"
|
||||||
"github.com/songquanpeng/one-api/relay/model"
|
"github.com/songquanpeng/one-api/relay/model"
|
||||||
"github.com/songquanpeng/one-api/relay/util"
|
"github.com/songquanpeng/one-api/relay/util"
|
||||||
"io"
|
|
||||||
"net/http"
|
|
||||||
"strings"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type Adaptor struct {
|
type Adaptor struct {
|
||||||
@ -23,7 +24,13 @@ func (a *Adaptor) Init(meta *util.RelayMeta) {
|
|||||||
func (a *Adaptor) GetRequestURL(meta *util.RelayMeta) (string, error) {
|
func (a *Adaptor) GetRequestURL(meta *util.RelayMeta) (string, error) {
|
||||||
// https://cloud.baidu.com/doc/WENXINWORKSHOP/s/clntwmv7t
|
// https://cloud.baidu.com/doc/WENXINWORKSHOP/s/clntwmv7t
|
||||||
suffix := "chat/"
|
suffix := "chat/"
|
||||||
if strings.HasPrefix("Embedding", meta.ActualModelName) {
|
if strings.HasPrefix(meta.ActualModelName, "Embedding") {
|
||||||
|
suffix = "embeddings/"
|
||||||
|
}
|
||||||
|
if strings.HasPrefix(meta.ActualModelName, "bge-large") {
|
||||||
|
suffix = "embeddings/"
|
||||||
|
}
|
||||||
|
if strings.HasPrefix(meta.ActualModelName, "tao-8k") {
|
||||||
suffix = "embeddings/"
|
suffix = "embeddings/"
|
||||||
}
|
}
|
||||||
switch meta.ActualModelName {
|
switch meta.ActualModelName {
|
||||||
@ -45,6 +52,12 @@ func (a *Adaptor) GetRequestURL(meta *util.RelayMeta) (string, error) {
|
|||||||
suffix += "bloomz_7b1"
|
suffix += "bloomz_7b1"
|
||||||
case "Embedding-V1":
|
case "Embedding-V1":
|
||||||
suffix += "embedding-v1"
|
suffix += "embedding-v1"
|
||||||
|
case "bge-large-zh":
|
||||||
|
suffix += "bge_large_zh"
|
||||||
|
case "bge-large-en":
|
||||||
|
suffix += "bge_large_en"
|
||||||
|
case "tao-8k":
|
||||||
|
suffix += "tao_8k"
|
||||||
default:
|
default:
|
||||||
suffix += meta.ActualModelName
|
suffix += meta.ActualModelName
|
||||||
}
|
}
|
||||||
|
@ -7,4 +7,7 @@ var ModelList = []string{
|
|||||||
"ERNIE-Speed",
|
"ERNIE-Speed",
|
||||||
"ERNIE-Bot-turbo",
|
"ERNIE-Bot-turbo",
|
||||||
"Embedding-V1",
|
"Embedding-V1",
|
||||||
|
"bge-large-zh",
|
||||||
|
"bge-large-en",
|
||||||
|
"tao-8k",
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user