diff --git a/CHANGELOG.md b/CHANGELOG.md index 50356463..b050ecd6 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -8,6 +8,7 @@ - 功能优化:登录,注册页面允许替换用户自己的 Logo 和 Title - Bug 修复:修复 OpenAI 实时语音通话没有检测用户算力不足的 Bug - 功能新增:管理后台增加算力日志查询功能,支持按用户,按模型,按日期,按类型查询算力日志 +- 功能优化:支持为模型绑定 Dalle 和 chat 类型的 API KEY ## v4.1.8 diff --git a/api/handler/admin/api_key_handler.go b/api/handler/admin/api_key_handler.go index b6454252..8daa0990 100644 --- a/api/handler/admin/api_key_handler.go +++ b/api/handler/admin/api_key_handler.go @@ -16,6 +16,7 @@ import ( "geekai/store/vo" "geekai/utils" "geekai/utils/resp" + "strings" "github.com/gin-gonic/gin" "gorm.io/gorm" @@ -71,16 +72,18 @@ func (h *ApiKeyHandler) Save(c *gin.Context) { resp.SUCCESS(c, keyVo) } +// List 获取 API KEY 列表 func (h *ApiKeyHandler) List(c *gin.Context) { status := h.GetBool(c, "status") - t := h.GetTrim(c, "type") + t := c.Query("type") session := h.DB.Session(&gorm.Session{}) if status { session = session.Where("enabled", true) } if t != "" { - session = session.Where("type", t) + types := strings.Split(t, "|") + session = session.Where("type IN ?", types) } var items []model.ApiKey diff --git a/web/src/views/admin/ChatModel.vue b/web/src/views/admin/ChatModel.vue index 05706e87..617b0d9a 100644 --- a/web/src/views/admin/ChatModel.vue +++ b/web/src/views/admin/ChatModel.vue @@ -130,6 +130,7 @@ {{ v.name }} + {{ v.type }} {{ substr(v.api_url, 50) }} @@ -194,7 +195,7 @@ const type = ref([ // 获取 API KEY const apiKeys = ref([]); -httpGet("/api/admin/apikey/list?type=chat") +httpGet("/api/admin/apikey/list?type=chat|dalle") .then((res) => { apiKeys.value = res.data; })