diff --git a/controller/playground.go b/controller/playground.go index 37a5c7b0..10393250 100644 --- a/controller/playground.go +++ b/controller/playground.go @@ -66,5 +66,14 @@ func Playground(c *gin.Context) { } middleware.SetupContextForSelectedChannel(c, channel, playgroundRequest.Model) c.Set(constant.ContextKeyRequestStartTime, time.Now()) + + // Write user context to ensure acceptUnsetRatio is available + userId := c.GetInt("id") + userCache, err := model.GetUserCache(userId) + if err != nil { + openaiErr = service.OpenAIErrorWrapperLocal(err, "get_user_cache_failed", http.StatusInternalServerError) + return + } + userCache.WriteContext(c) Relay(c) }