From b50b43eb654a84abe1ba900d24617f6679223dde Mon Sep 17 00:00:00 2001
From: manjieqi <40858189+manjieqi@users.noreply.github.com>
Date: Fri, 5 Apr 2024 17:30:48 +0800
Subject: [PATCH] feat: update baidu model name & ratio (#1277)

---
 common/model-ratio.go            | 10 +++++-----
 relay/channel/baidu/adaptor.go   | 14 +++++++++++---
 relay/channel/baidu/constants.go | 10 +++++-----
 3 files changed, 21 insertions(+), 13 deletions(-)

diff --git a/common/model-ratio.go b/common/model-ratio.go
index 94607c92..97108c08 100644
--- a/common/model-ratio.go
+++ b/common/model-ratio.go
@@ -73,15 +73,15 @@ var ModelRatio = map[string]float64{
 	"claude-3-opus-20240229":   15.0 / 1000 * USD,
 	// https://cloud.baidu.com/doc/WENXINWORKSHOP/s/hlrk4akp7
 	"ERNIE-4.0-8K":       0.120 * RMB,
-	"ERNIE-Bot-8K-0922":  0.024 * RMB,
 	"ERNIE-3.5-8K":       0.012 * RMB,
-	"ERNIE-Lite-8K-0922": 0.008 * RMB,
-	"ERNIE-Speed-8K":     0.004 * RMB,
-	"ERNIE-3.5-4K-0205":  0.012 * RMB,
 	"ERNIE-3.5-8K-0205":  0.024 * RMB,
 	"ERNIE-3.5-8K-1222":  0.012 * RMB,
-	"ERNIE-Lite-8K":      0.003 * RMB,
+	"ERNIE-Bot-8K":       0.024 * RMB,
+	"ERNIE-3.5-4K-0205":  0.012 * RMB,
+	"ERNIE-Speed-8K":     0.004 * RMB,
 	"ERNIE-Speed-128K":   0.004 * RMB,
+	"ERNIE-Lite-8K-0922": 0.008 * RMB,
+	"ERNIE-Lite-8K-0308": 0.003 * RMB,
 	"ERNIE-Tiny-8K":      0.001 * RMB,
 	"BLOOMZ-7B":          0.004 * RMB,
 	"Embedding-V1":       0.002 * RMB,
diff --git a/relay/channel/baidu/adaptor.go b/relay/channel/baidu/adaptor.go
index 72302fdf..b7011bc7 100644
--- a/relay/channel/baidu/adaptor.go
+++ b/relay/channel/baidu/adaptor.go
@@ -44,17 +44,25 @@ func (a *Adaptor) GetRequestURL(meta *util.RelayMeta) (string, error) {
 		suffix += "eb-instant"
 	case "ERNIE-Speed":
 		suffix += "ernie_speed"
-	case "ERNIE-Bot-8K":
-		suffix += "ernie_bot_8k"
 	case "ERNIE-4.0-8K":
 		suffix += "completions_pro"
 	case "ERNIE-3.5-8K":
 		suffix += "completions"
+	case "ERNIE-3.5-8K-0205":
+		suffix += "ernie-3.5-8k-0205"
+	case "ERNIE-3.5-8K-1222":
+		suffix += "ernie-3.5-8k-1222"
+	case "ERNIE-Bot-8K":
+		suffix += "ernie_bot_8k"
+	case "ERNIE-3.5-4K-0205":
+		suffix += "ernie-3.5-4k-0205"
 	case "ERNIE-Speed-8K":
 		suffix += "ernie_speed"
 	case "ERNIE-Speed-128K":
 		suffix += "ernie-speed-128k"
-	case "ERNIE-Lite-8K":
+	case "ERNIE-Lite-8K-0922":
+		suffix += "eb-instant"
+	case "ERNIE-Lite-8K-0308":
 		suffix += "ernie-lite-8k"
 	case "ERNIE-Tiny-8K":
 		suffix += "ernie-tiny-8k"
diff --git a/relay/channel/baidu/constants.go b/relay/channel/baidu/constants.go
index ccdc25c3..f952adc6 100644
--- a/relay/channel/baidu/constants.go
+++ b/relay/channel/baidu/constants.go
@@ -2,15 +2,15 @@ package baidu
 
 var ModelList = []string{
 	"ERNIE-4.0-8K",
-	"ERNIE-Bot-8K-0922",
 	"ERNIE-3.5-8K",
-	"ERNIE-Lite-8K-0922",
-	"ERNIE-Speed-8K",
-	"ERNIE-3.5-4K-0205",
 	"ERNIE-3.5-8K-0205",
 	"ERNIE-3.5-8K-1222",
-	"ERNIE-Lite-8K",
+	"ERNIE-Bot-8K",
+	"ERNIE-3.5-4K-0205",
+	"ERNIE-Speed-8K",
 	"ERNIE-Speed-128K",
+	"ERNIE-Lite-8K-0922",
+	"ERNIE-Lite-8K-0308",
 	"ERNIE-Tiny-8K",
 	"BLOOMZ-7B",
 	"Embedding-V1",