From a100baf57ff6c3124032af55d1822608ceca78a7 Mon Sep 17 00:00:00 2001
From: feitianbubu <feitianbubu@qq.com>
Date: Wed, 16 Jul 2025 23:41:31 +0800
Subject: [PATCH] fix: playground chat

---
 controller/playground.go |  5 ++---
 middleware/auth.go       | 12 ++++++++++++
 2 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/controller/playground.go b/controller/playground.go
index e071d12e..9421a665 100644
--- a/controller/playground.go
+++ b/controller/playground.go
@@ -65,9 +65,8 @@ func Playground(c *gin.Context) {
 		Group:  group,
 	}
 	_ = middleware.SetupContextForToken(c, tempToken)
-	_, err = getChannel(c, group, playgroundRequest.Model, 0)
-	if err != nil {
-		newAPIError = types.NewError(err, types.ErrorCodeGetChannelFailed)
+	_, newAPIError = getChannel(c, group, playgroundRequest.Model, 1)
+	if newAPIError != nil {
 		return
 	}
 	//middleware.SetupContextForSelectedChannel(c, channel, playgroundRequest.Model)
diff --git a/middleware/auth.go b/middleware/auth.go
index 47d033a9..b9c80920 100644
--- a/middleware/auth.go
+++ b/middleware/auth.go
@@ -123,6 +123,18 @@ func authHelper(c *gin.Context, minRole int) {
 	c.Set("id", id)
 	c.Set("group", session.Get("group"))
 	c.Set("use_access_token", useAccessToken)
+
+	userCache, err := model.GetUserCache(id.(int))
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		c.Abort()
+		return
+	}
+	userCache.WriteContext(c)
+
 	c.Next()
 }