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

Merge branch 'patch/models-api'

Browse Source
This commit is contained in:
Laisky.Cai
2025-02-27 01:12:38 +00:00
parent 5905a7f295 54635ca4fe
commit 155a3aac73
3 changed files with 57 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
controller/model.go
View File

@@ -152,7 +152,7 @@ func ListModels(c *gin.Context) {
}
// Get available models with their channel names
availableAbilities, err := model.GetGroupModelsV2(c.Request.Context(), userGroup)
availableAbilities, err := model.CacheGetGroupModelsV2(c.Request.Context(), userGroup)
if err != nil {
middleware.AbortWithError(c, http.StatusBadRequest, err)
return
@@ -217,7 +217,8 @@ func GetUserAvailableModels(c *gin.Context) {
})
return
}
models, err := model.CacheGetGroupModels(ctx, userGroup)
models, err := model.CacheGetGroupModelsV2(ctx, userGroup)
if err != nil {
c.JSON(http.StatusOK, gin.H{
"success": false,
@@ -225,10 +226,20 @@ func GetUserAvailableModels(c *gin.Context) {
})
return
}
var modelNames []string
modelsMap := map[string]bool{}
for _, model := range models {
modelsMap[model.Model] = true
}
for modelName := range modelsMap {
modelNames = append(modelNames, modelName)
}
c.JSON(http.StatusOK, gin.H{
"success": true,
"message": "",
"data": models,
"data": modelNames,
})
return
}