From 54088bc6646f4e54b473ba852b5df8e731bc09ae Mon Sep 17 00:00:00 2001
From: "1808837298@qq.com" <1808837298@qq.com>
Date: Sun, 3 Mar 2024 15:53:48 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E6=B7=BB=E5=8A=A0?=
 =?UTF-8?q?=E5=92=8C=E7=BC=96=E8=BE=91=E6=B8=A0=E9=81=93=E6=97=A0=E5=8F=AF?=
 =?UTF-8?q?=E9=80=89=E6=8B=A9=E6=A8=A1=E5=9E=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 common/model-ratio.go | 1 +
 controller/model.go   | 7 +++++++
 router/api-router.go  | 2 +-
 3 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/common/model-ratio.go b/common/model-ratio.go
index 846ec8a..648f9fc 100644
--- a/common/model-ratio.go
+++ b/common/model-ratio.go
@@ -83,6 +83,7 @@ var ModelRatio = map[string]float64{
 	"SparkDesk-v1.1":            1.2858, // ￥0.018 / 1k tokens
 	"SparkDesk-v2.1":            1.2858, // ￥0.018 / 1k tokens
 	"SparkDesk-v3.1":            1.2858, // ￥0.018 / 1k tokens
+	"SparkDesk-v3.5":            1.2858, // ￥0.018 / 1k tokens
 	"360GPT_S2_V9":              0.8572, // ¥0.012 / 1k tokens
 	"embedding-bert-512-v1":     0.0715, // ¥0.001 / 1k tokens
 	"embedding_s1_v1":           0.0715, // ¥0.001 / 1k tokens
diff --git a/controller/model.go b/controller/model.go
index 8909de4..38c6c46 100644
--- a/controller/model.go
+++ b/controller/model.go
@@ -129,6 +129,13 @@ func ListModels(c *gin.Context) {
 	})
 }
 
+func ChannelListModels(c *gin.Context) {
+	c.JSON(200, gin.H{
+		"object": "list",
+		"data":   openAIModels,
+	})
+}
+
 func RetrieveModel(c *gin.Context) {
 	modelId := c.Param("model")
 	if model, ok := openAIModelsMap[modelId]; ok {
diff --git a/router/api-router.go b/router/api-router.go
index 04d3490..48d62fa 100644
--- a/router/api-router.go
+++ b/router/api-router.go
@@ -73,7 +73,7 @@ func SetApiRouter(router *gin.Engine) {
 		{
 			channelRoute.GET("/", controller.GetAllChannels)
 			channelRoute.GET("/search", controller.SearchChannels)
-			channelRoute.GET("/models", controller.ListModels)
+			channelRoute.GET("/models", controller.ChannelListModels)
 			channelRoute.GET("/:id", controller.GetChannel)
 			channelRoute.GET("/test", controller.TestAllChannels)
 			channelRoute.GET("/test/:id", controller.TestChannel)