xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-19 22:53:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: channel listing

Browse Source
This commit is contained in:
ckt1031
2023-07-24 00:15:58 +08:00
parent 657de89219
commit 082dd33ffa
3 changed files with 16 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
controller/model.go
View File

@@ -332,22 +332,24 @@ func init() {
}
}
func ListModels(c *gin.Context) {
func ListModels(c *gin.Context, isChannel bool) {
userId := c.GetInt("id")
userGroup, _ := model.CacheGetUserGroup(userId)
// Only return the models that the channel and user have access to
var modalList []OpenAIModels
for _, modalData := range openAIModels {
channel, err := model.CacheGetRandomSatisfiedChannel(userGroup, modalData.Id)
if !isChannel {
for _, modalData := range openAIModels {
channel, err := model.CacheGetRandomSatisfiedChannel(userGroup, modalData.Id)
if err != nil {
continue
}
if err != nil {
continue
}
if channel != nil {
modalList = append(modalList, modalData)
if channel != nil {
modalList = append(modalList, modalData)
}
}
}