xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-04 16:23:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: api key manage page funciton is ready

Browse Source
This commit is contained in:
RockYang
2024-01-04 10:48:04 +08:00
parent 79f6a43019
commit 21f2622a4b
6 changed files with 133 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
api/handler/admin/api_key_handler.go
View File

@@ -29,6 +29,9 @@ func (h *ApiKeyHandler) Save(c *gin.Context) {
Platform string `json:"platform"`
Type string `json:"type"`
Value string `json:"value"`
ApiURL string `json:"api_url"`
Enabled bool `json:"enabled"`
UseProxy bool `json:"use_proxy"`
}
if err := c.ShouldBindJSON(&data); err != nil {
resp.ERROR(c, types.InvalidArgs)
@@ -42,6 +45,9 @@ func (h *ApiKeyHandler) Save(c *gin.Context) {
apiKey.Platform = data.Platform
apiKey.Value = data.Value
apiKey.Type = data.Type
apiKey.ApiURL = data.ApiURL
apiKey.Enabled = data.Enabled
apiKey.UseProxy = data.UseProxy
res := h.db.Save(&apiKey)
if res.Error != nil {
resp.ERROR(c, "更新数据库失败！")
@@ -80,6 +86,26 @@ func (h *ApiKeyHandler) List(c *gin.Context) {
resp.SUCCESS(c, keys)
}
func (h *ApiKeyHandler) Set(c *gin.Context) {
var data struct {
Id uint `json:"id"`
Filed string `json:"filed"`
Value interface{} `json:"value"`
}
if err := c.ShouldBindJSON(&data); err != nil {
resp.ERROR(c, types.InvalidArgs)
return
}
res := h.db.Model(&model.ApiKey{}).Where("id = ?", data.Id).Update(data.Filed, data.Value)
if res.Error != nil {
resp.ERROR(c, "更新数据库失败！")
return
}
resp.SUCCESS(c)
}
func (h *ApiKeyHandler) Remove(c *gin.Context) {
id := h.GetInt(c, "id", 0)

1
api/main.go
View File

@@ -255,6 +255,7 @@ func main() {
group := s.Engine.Group("/api/admin/apikey/")
group.POST("save", h.Save)
group.GET("list", h.List)
group.POST("set", h.Set)
group.GET("remove", h.Remove)
}),
fx.Invoke(func(s *core.AppServer, h *admin.UserHandler) {

3
api/store/model/api_key.go
View File

@@ -6,5 +6,8 @@ type ApiKey struct {
Platform string
Type string // 用途 chat => 聊天，img => 绘图
Value string // API Key 的值
ApiURL string // 当前 KEY 的 API 地址
Enabled bool // 是否启用
UseProxy bool // 是否使用代理访问 API URL
LastUsedAt int64 // 最后使用时间
}

5
api/store/vo/api_key.go
View File

@@ -5,6 +5,9 @@ type ApiKey struct {
BaseVo
Platform string `json:"platform"`
Type string `json:"type"`
Value string `json:"value"` // API Key 的值
Value string `json:"value"` // API Key 的值
ApiURL string `json:"api_url"`
Enabled bool `json:"enabled"`
UseProxy bool `json:"use_proxy"`
LastUsedAt int64 `json:"last_used_at"` // 最后使用时间
}

3
database/update-v3.2.4.sql
View File

@@ -2,4 +2,5 @@ ALTER TABLE `chatgpt_users` ADD `nickname` VARCHAR(30) NOT NULL COMMENT '昵称'
ALTER TABLE `chatgpt_rewards` ADD `exchange` VARCHAR(255) NOT NULL COMMENT '兑换详情（json）' AFTER `status`;
ALTER TABLE `chatgpt_api_keys` ADD `api_url` VARCHAR(255) NULL COMMENT 'API 地址' AFTER `last_used_at`, ADD `enabled` TINYINT(1) NULL COMMENT '是否启用' AFTER `api_url`;
ALTER TABLE `chatgpt_api_keys` DROP INDEX `value`;
ALTER TABLE `chatgpt_mj_jobs` ADD UNIQUE(`task_id`);
ALTER TABLE `chatgpt_mj_jobs` ADD UNIQUE(`task_id`);
ALTER TABLE `chatgpt_api_keys` ADD `use_proxy` TINYINT(1) NULL COMMENT '是否使用代理访问' AFTER `enabled`;

114
web/src/views/admin/ApiKey.vue
View File

@@ -15,10 +15,9 @@
<el-tag v-else-if="scope.row.type === 'img'" type="success">绘图</el-tag>
</template>
</el-table-column>
<el-table-column label="创建时间">
<el-table-column prop="use_proxy" label="使用代理">
<template #default="scope">
<span>{{ dateFormat(scope.row['created_at']) }}</span>
<el-switch v-model="scope.row['use_proxy']" @change="set('use_proxy',scope.row)"/>
</template>
</el-table-column>
@@ -28,6 +27,11 @@
<el-tag v-else>未使用</el-tag>
</template>
</el-table-column>
<el-table-column prop="enabled" label="启用状态">
<template #default="scope">
<el-switch v-model="scope.row['enabled']" @change="set('enabled',scope.row)"/>
</template>
</el-table-column>
<el-table-column label="操作" width="180">
<template #default="scope">
@@ -51,31 +55,51 @@
:closable="false"
show-icon
style="margin-bottom: 10px; font-size:14px;">
<p><b>注意：</b>如果是百度文心一言平台，需要用竖线（|）将 API Key 和 Secret Key 串接起来填入！</p>
<p><b>注意：</b>如果是讯飞星火大模型，需要用竖线（|）将 APPID, APIKey 和 APISecret 按照顺序串接起来填入！</p>
<p><b>注意：</b>如果是百度文心一言平台，API-KEY 为 APIKey|SecretKey，中间用竖线（|）连接</p>
<p><b>注意：</b>如果是讯飞星火大模型，API-KEY 为 AppId|APIKey|APISecret，中间用竖线（|）连接</p>
</el-alert>
<el-form :model="item" label-width="120px" ref="formRef" :rules="rules">
<el-form-item label="所属平台：" prop="platform">
<el-select v-model="item.platform" placeholder="请选择平台">
<el-select v-model="item.platform" placeholder="请选择平台" @change="changePlatform">
<el-option v-for="item in platforms" :value="item.value" :label="item.name" :key="item.value">{{
item.name
}}
</el-option>
</el-select>
</el-form-item>
<el-form-item label="API KEY：" prop="value">
<el-input v-model="item.value" autocomplete="off"/>
</el-form-item>
<el-form-item label="用途：" prop="type">
<el-select v-model="item.type" placeholder="请选择用途">
<el-select v-model="item.type" placeholder="请选择用途" @change="changePlatform">
<el-option v-for="item in types" :value="item.value" :label="item.name" :key="item.value">{{
item.name
}}
</el-option>
</el-select>
</el-form-item>
<el-form-item label="API KEY：" prop="value">
<el-input v-model="item.value" autocomplete="off"/>
</el-form-item>
<el-form-item label="API URL：" prop="api_url">
<el-input v-model="item.api_url" autocomplete="off"
placeholder="如果你用了第三方的 API 中转，这里填写中转地址"/>
</el-form-item>
<el-form-item label="使用代理：" prop="use_proxy">
<el-switch v-model="item.use_proxy"/>
<el-tooltip
effect="dark"
content="是否使用代理访问 API URL，OpenAI 官方API需要开启代理访问"
raw-content
placement="right"
>
<el-icon>
<InfoFilled/>
</el-icon>
</el-tooltip>
</el-form-item>
<el-form-item label="启用状态：" prop="enable">
<el-switch v-model="item.enabled"/>
</el-form-item>
</el-form>
<template #footer>
@@ -93,7 +117,7 @@ import {reactive, ref} from "vue";
import {httpGet, httpPost} from "@/utils/http";
import {ElMessage} from "element-plus";
import {dateFormat, disabledDate, removeArrayItem} from "@/utils/libs";
import {Plus} from "@element-plus/icons-vue";
import {InfoFilled, Plus} from "@element-plus/icons-vue";
// 变量定义
const items = ref([])
@@ -108,11 +132,32 @@ const loading = ref(true)
const formRef = ref(null)
const title = ref("")
const platforms = ref([
{name: "【OpenAI】ChatGPT", value: "OpenAI"},
{name: "【讯飞】星火大模型", value: "XunFei"},
{name: "【清华智普】ChatGLM", value: "ChatGLM"},
{name: "【百度】文心一言", value: "Baidu"},
{name: "【微软】Azure", value: "Azure"},
{
name: "【OpenAI】ChatGPT",
value: "OpenAI",
api_url: "https://api.fast-tunnel.one/v1/chat/completions",
img_url: "https://api.openai.com/v1/images/generations"
},
{
name: "【讯飞】星火大模型",
value: "XunFei",
api_url: "wss://spark-api.xf-yun.com/{version}/chat"
},
{
name: "【清华智普】ChatGLM",
value: "ChatGLM",
api_url: "https://open.bigmodel.cn/api/paas/v3/model-api/{model}/sse-invoke"
},
{
name: "【百度】文心一言",
value: "Baidu",
api_url: "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/{model}"
},
{
name: "【微软】Azure",
value: "Azure",
api_url: "https://chat-bot-api.openai.azure.com/openai/deployments/{model}/chat/completions?api-version=2023-05-15"
},
])
const types = ref([
{name: "聊天", value: "chat"},
@@ -176,6 +221,33 @@ const remove = function (row) {
ElMessage.error("删除失败：" + e.message)
})
}
const set = (filed, row) => {
httpPost('/api/admin/apikey/set', {id: row.id, filed: filed, value: row[filed]}).then(() => {
ElMessage.success("操作成功！")
}).catch(e => {
ElMessage.error("操作失败：" + e.message)
})
}
const changePlatform = () => {
let platform = null
for (let v of platforms.value) {
if (v.value === item.value.platform) {
platform = v
break
}
}
if (platform !== null) {
if (item.value.type === "chat") {
item.value.api_url = platform.api_url
} else if (platform.img_url) {
item.value.api_url = platform.img_url
}
}
}
</script>
<style lang="stylus" scoped>
@@ -194,6 +266,14 @@ const remove = function (row) {
.el-select {
width: 100%
}
}
.el-form {
.el-form-item__content {
.el-icon {
padding-left 10px;
}
}
}
</style>