From 0cf53ac5ff226b4831f8d915a769967385b8c927 Mon Sep 17 00:00:00 2001
From: "1808837298@qq.com" <1808837298@qq.com>
Date: Thu, 26 Sep 2024 01:49:35 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20Playground=E7=9B=B8=E5=85=B3=E6=8E=A5?=
 =?UTF-8?q?=E5=8F=A3=E7=A6=81=E7=94=A8AccessToken?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 controller/relay.go | 8 ++++++--
 middleware/auth.go  | 1 +
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/controller/relay.go b/controller/relay.go
index 6a1ddb2..412f951 100644
--- a/controller/relay.go
+++ b/controller/relay.go
@@ -49,6 +49,12 @@ func Playground(c *gin.Context) {
 		}
 	}()
 
+	useAccessToken := c.GetBool("use_access_token")
+	if useAccessToken {
+		openaiErr = service.OpenAIErrorWrapperLocal(errors.New("暂不支持使用 access token"), "access_token_not_supported", http.StatusBadRequest)
+		return
+	}
+
 	playgroundRequest := &dto.PlayGroundRequest{}
 	err := common.UnmarshalBodyReusable(c, playgroundRequest)
 	if err != nil {
@@ -67,8 +73,6 @@ func Playground(c *gin.Context) {
 	} else {
 		c.Set("group", group)
 	}
-	log.Printf("group: %s", group)
-	log.Printf("model: %s", playgroundRequest.Model)
 	channel, err := model.CacheGetRandomSatisfiedChannel(group, playgroundRequest.Model, 0)
 	if err != nil {
 		openaiErr = service.OpenAIErrorWrapperLocal(err, "get_playground_channel_failed", http.StatusInternalServerError)
diff --git a/middleware/auth.go b/middleware/auth.go
index 2735f8a..76f2b6b 100644
--- a/middleware/auth.go
+++ b/middleware/auth.go
@@ -122,6 +122,7 @@ func authHelper(c *gin.Context, minRole int) {
 	c.Set("role", role)
 	c.Set("id", id)
 	c.Set("group", session.Get("group"))
+	c.Set("use_access_token", useAccessToken)
 	c.Next()
 }