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() }