ChatGPT/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-10-14 06:20:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: /v1/models now only return available models

Browse Source
This commit is contained in:
JustSong
2024-04-04 02:44:59 +08:00
parent dc7aaf2de5
commit 8b9813d63b
2 changed files with 36 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
middleware/auth.go
View File

@@ -115,7 +115,8 @@ func TokenAuth() func(c *gin.Context) {
}
c.Set("request_model", requestModel)
if token.Models != nil && *token.Models != "" {
if !isModelInList(requestModel, *token.Models) {
c.Set("available_models", *token.Models)
if requestModel != "" && !isModelInList(requestModel, *token.Models) {
abortWithMessage(c, http.StatusForbidden, fmt.Sprintf("该令牌无权使用模型：%s", requestModel))
return
}