xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-17 16:06:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: playground group

Browse Source
This commit is contained in:
1808837298@qq.com 2024-10-10 13:39:09 +08:00
parent e298f2e5a4
commit 6a8a4bcf65
2 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
common/user_groups.go
View File

@ -21,3 +21,8 @@ func UpdateUserUsableGroupsByJSONString(jsonStr string) error {
UserUsableGroups = make(map[string]string) UserUsableGroups = make(map[string]string)
return json.Unmarshal([]byte(jsonStr), &UserUsableGroups) return json.Unmarshal([]byte(jsonStr), &UserUsableGroups)
} }
func GroupInUserUsableGroups(groupName string) bool {
_, ok := UserUsableGroups[groupName]
return ok
}

8
controller/relay.go
View File

@ -68,9 +68,15 @@ func Playground(c *gin.Context) {
} }
c.Set("original_model", playgroundRequest.Model) c.Set("original_model", playgroundRequest.Model)
group := playgroundRequest.Group group := playgroundRequest.Group
userGroup := c.GetString("group")
if group == "" { if group == "" {
group = c.GetString("group") group = userGroup
} else { } else {
if !common.GroupInUserUsableGroups(group) && group != userGroup {
openaiErr = service.OpenAIErrorWrapperLocal(errors.New("无权访问该分组"), "group_not_allowed", http.StatusForbidden)
return
}
c.Set("group", group) c.Set("group", group)
} }
c.Set("token_name", "playground-"+group) c.Set("token_name", "playground-"+group)