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

fix: '/v1/models' #474

Browse Source
This commit is contained in:
CalciumIon 2024-09-17 22:39:58 +08:00
parent e6df0ed20c
commit 02f5137781
1 changed files with 13 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
controller/model.go
View File

@ -146,49 +146,22 @@ func ListModels(c *gin.Context) {
}) })
return return
} }
models := model.GetGroupModels(user.Group)
userOpenAiModels := make([]dto.OpenAIModels, 0) userOpenAiModels := make([]dto.OpenAIModels, 0)
permission := getPermission() permission := getPermission()
for _, s := range models {
modelLimitEnable := c.GetBool("token_model_limit_enabled") if _, ok := openAIModelsMap[s]; ok {
if modelLimitEnable { userOpenAiModels = append(userOpenAiModels, openAIModelsMap[s])
s, ok := c.Get("token_model_limit")
var tokenModelLimit map[string]bool
if ok {
tokenModelLimit = s.(map[string]bool)
} else { } else {
tokenModelLimit = map[string]bool{} userOpenAiModels = append(userOpenAiModels, dto.OpenAIModels{
} Id: s,
for allowModel, _ := range tokenModelLimit { Object: "model",
if _, ok := openAIModelsMap[allowModel]; ok { Created: 1626777600,
userOpenAiModels = append(userOpenAiModels, openAIModelsMap[allowModel]) OwnedBy: "custom",
} else { Permission: permission,
userOpenAiModels = append(userOpenAiModels, dto.OpenAIModels{ Root: s,
Id: allowModel, Parent: nil,
Object: "model", })
Created: 1626777600,
OwnedBy: "custom",
Permission: permission,
Root: allowModel,
Parent: nil,
})
}
}
} else {
models := model.GetGroupModels(user.Group)
for _, s := range models {
if _, ok := openAIModelsMap[s]; ok {
userOpenAiModels = append(userOpenAiModels, openAIModelsMap[s])
} else {
userOpenAiModels = append(userOpenAiModels, dto.OpenAIModels{
Id: s,
Object: "model",
Created: 1626777600,
OwnedBy: "custom",
Permission: permission,
Root: s,
Parent: nil,
})
}
} }
} }
c.JSON(200, gin.H{ c.JSON(200, gin.H{