From 4116d8d59aeb1a7cc51a29ad70f8475084a9d564 Mon Sep 17 00:00:00 2001
From: jiaguobao <779188083@qq.com>
Date: Wed, 13 Mar 2024 16:43:00 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E7=99=BE=E5=BA=A6=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E4=B8=ADURL=E7=9A=84=E5=AD=97=E7=AC=A6=E4=B8=B2?=
 =?UTF-8?q?=E5=88=A4=E6=96=ADbug=EF=BC=9B=20=E6=B7=BB=E5=8A=A0=E7=99=BE?=
 =?UTF-8?q?=E5=BA=A6=E7=9A=84=E5=8F=A6=E5=A4=963=E4=B8=AA=E5=90=91?=
 =?UTF-8?q?=E9=87=8F=E6=A8=A1=E5=9E=8B=E3=80=90"bge-large-zh",=20=09"bge-l?=
 =?UTF-8?q?arge-en",=20=09"tao-8k",=20=E3=80=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 relay/channel/baidu/adaptor.go   | 21 +++++++++++++++++----
 relay/channel/baidu/constants.go |  3 +++
 2 files changed, 20 insertions(+), 4 deletions(-)

diff --git a/relay/channel/baidu/adaptor.go b/relay/channel/baidu/adaptor.go
index 1a96997a..2d2e24f6 100644
--- a/relay/channel/baidu/adaptor.go
+++ b/relay/channel/baidu/adaptor.go
@@ -3,14 +3,15 @@ package baidu
 import (
 	"errors"
 	"fmt"
+	"io"
+	"net/http"
+	"strings"
+
 	"github.com/gin-gonic/gin"
 	"github.com/songquanpeng/one-api/relay/channel"
 	"github.com/songquanpeng/one-api/relay/constant"
 	"github.com/songquanpeng/one-api/relay/model"
 	"github.com/songquanpeng/one-api/relay/util"
-	"io"
-	"net/http"
-	"strings"
 )
 
 type Adaptor struct {
@@ -23,7 +24,13 @@ func (a *Adaptor) Init(meta *util.RelayMeta) {
 func (a *Adaptor) GetRequestURL(meta *util.RelayMeta) (string, error) {
 	// https://cloud.baidu.com/doc/WENXINWORKSHOP/s/clntwmv7t
 	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/"
 	}
 	switch meta.ActualModelName {
@@ -45,6 +52,12 @@ func (a *Adaptor) GetRequestURL(meta *util.RelayMeta) (string, error) {
 		suffix += "bloomz_7b1"
 	case "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:
 		suffix += meta.ActualModelName
 	}
diff --git a/relay/channel/baidu/constants.go b/relay/channel/baidu/constants.go
index 0fa8f2d6..45a4e901 100644
--- a/relay/channel/baidu/constants.go
+++ b/relay/channel/baidu/constants.go
@@ -7,4 +7,7 @@ var ModelList = []string{
 	"ERNIE-Speed",
 	"ERNIE-Bot-turbo",
 	"Embedding-V1",
+	"bge-large-zh",
+	"bge-large-en",
+	"tao-8k",
 }