🎨 优化项目结构 && 完善ai配置
This commit is contained in:
@@ -1,11 +1,9 @@
|
||||
package app
|
||||
|
||||
import (
|
||||
"strconv"
|
||||
|
||||
"git.echol.cn/loser/ai_proxy/server/global"
|
||||
"git.echol.cn/loser/ai_proxy/server/model/app/request"
|
||||
commonRequest "git.echol.cn/loser/ai_proxy/server/model/common/request"
|
||||
"git.echol.cn/loser/ai_proxy/server/model/app"
|
||||
"git.echol.cn/loser/ai_proxy/server/model/common/request"
|
||||
"git.echol.cn/loser/ai_proxy/server/model/common/response"
|
||||
"git.echol.cn/loser/ai_proxy/server/service"
|
||||
"git.echol.cn/loser/ai_proxy/server/utils"
|
||||
@@ -17,216 +15,163 @@ type AiPresetApi struct{}
|
||||
|
||||
var aiPresetService = service.ServiceGroupApp.AppServiceGroup.AiPresetService
|
||||
|
||||
// CreateAiPreset 创建AI预设
|
||||
// CreateAiPreset 创建预设
|
||||
// @Tags AiPreset
|
||||
// @Summary 创建AI预设
|
||||
// @Summary 创建预设
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param data body request.CreateAiPresetRequest true "预设信息"
|
||||
// @Success 200 {object} response.Response{data=app.AiPreset,msg=string} "创建成功"
|
||||
// @Router /app/preset [post]
|
||||
// @Param data body app.AiPreset true "预设信息"
|
||||
// @Success 200 {object} response.Response{msg=string} "创建成功"
|
||||
// @Router /aiPreset/createAiPreset [post]
|
||||
func (a *AiPresetApi) CreateAiPreset(c *gin.Context) {
|
||||
var req request.CreateAiPresetRequest
|
||||
err := c.ShouldBindJSON(&req)
|
||||
var preset app.AiPreset
|
||||
err := c.ShouldBindJSON(&preset)
|
||||
if err != nil {
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
preset.UserID = utils.GetUserID(c)
|
||||
|
||||
// 尝试获取用户ID,如果没有则使用0(公开访问)
|
||||
userId := uint(0)
|
||||
if id := utils.GetUserID(c); id > 0 {
|
||||
userId = id
|
||||
if err := aiPresetService.CreateAiPreset(&preset); err != nil {
|
||||
global.GVA_LOG.Error("创建失败!", zap.Error(err))
|
||||
response.FailWithMessage("创建失败", c)
|
||||
} else {
|
||||
response.OkWithMessage("创建成功", c)
|
||||
}
|
||||
preset, err := aiPresetService.CreateAiPreset(userId, &req)
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("创建预设失败!", zap.Error(err))
|
||||
response.FailWithMessage("创建预设失败", c)
|
||||
return
|
||||
}
|
||||
|
||||
response.OkWithData(preset, c)
|
||||
}
|
||||
|
||||
// DeleteAiPreset 删除AI预设
|
||||
// DeleteAiPreset 删除预设
|
||||
// @Tags AiPreset
|
||||
// @Summary 删除AI预设
|
||||
// @Summary 删除预设
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param id path uint true "预设ID"
|
||||
// @Param data body request.GetById true "ID"
|
||||
// @Success 200 {object} response.Response{msg=string} "删除成功"
|
||||
// @Router /app/preset/:id [delete]
|
||||
// @Router /aiPreset/deleteAiPreset [delete]
|
||||
func (a *AiPresetApi) DeleteAiPreset(c *gin.Context) {
|
||||
id, _ := strconv.ParseUint(c.Param("id"), 10, 32)
|
||||
// 尝试获取用户ID,如果没有则使用0(公开访问)
|
||||
userId := uint(0)
|
||||
if uid := utils.GetUserID(c); uid > 0 {
|
||||
userId = uid
|
||||
}
|
||||
|
||||
err := aiPresetService.DeleteAiPreset(uint(id), userId)
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("删除预设失败!", zap.Error(err))
|
||||
response.FailWithMessage("删除预设失败", c)
|
||||
return
|
||||
}
|
||||
|
||||
response.OkWithMessage("删除成功", c)
|
||||
}
|
||||
|
||||
// UpdateAiPreset 更新AI预设
|
||||
// @Tags AiPreset
|
||||
// @Summary 更新AI预设
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param data body request.UpdateAiPresetRequest true "预设信息"
|
||||
// @Success 200 {object} response.Response{data=app.AiPreset,msg=string} "更新成功"
|
||||
// @Router /app/preset [put]
|
||||
func (a *AiPresetApi) UpdateAiPreset(c *gin.Context) {
|
||||
var req request.UpdateAiPresetRequest
|
||||
err := c.ShouldBindJSON(&req)
|
||||
var reqId request.GetById
|
||||
err := c.ShouldBindJSON(&reqId)
|
||||
if err != nil {
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
userID := utils.GetUserID(c)
|
||||
|
||||
// 尝试获取用户ID,如果没有则使用0(公开访问)
|
||||
userId := uint(0)
|
||||
if id := utils.GetUserID(c); id > 0 {
|
||||
userId = id
|
||||
if err := aiPresetService.DeleteAiPreset(reqId.Uint(), userID); err != nil {
|
||||
global.GVA_LOG.Error("删除失败!", zap.Error(err))
|
||||
response.FailWithMessage("删除失败", c)
|
||||
} else {
|
||||
response.OkWithMessage("删除成功", c)
|
||||
}
|
||||
preset, err := aiPresetService.UpdateAiPreset(userId, &req)
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("更新预设失败!", zap.Error(err))
|
||||
response.FailWithMessage("更新预设失败", c)
|
||||
return
|
||||
}
|
||||
|
||||
response.OkWithData(preset, c)
|
||||
}
|
||||
|
||||
// GetAiPreset 获取AI预设详情
|
||||
// UpdateAiPreset 更新预设
|
||||
// @Tags AiPreset
|
||||
// @Summary 获取AI预设详情
|
||||
// @Summary 更新预设
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param id path uint true "预设ID"
|
||||
// @Success 200 {object} response.Response{data=app.AiPreset,msg=string} "获取成功"
|
||||
// @Router /app/preset/:id [get]
|
||||
func (a *AiPresetApi) GetAiPreset(c *gin.Context) {
|
||||
id, _ := strconv.ParseUint(c.Param("id"), 10, 32)
|
||||
// 尝试获取用户ID,如果没有则使用0(公开访问)
|
||||
userId := uint(0)
|
||||
if uid := utils.GetUserID(c); uid > 0 {
|
||||
userId = uid
|
||||
}
|
||||
|
||||
preset, err := aiPresetService.GetAiPreset(uint(id), userId)
|
||||
// @Param data body app.AiPreset true "预设信息"
|
||||
// @Success 200 {object} response.Response{msg=string} "更新成功"
|
||||
// @Router /aiPreset/updateAiPreset [put]
|
||||
func (a *AiPresetApi) UpdateAiPreset(c *gin.Context) {
|
||||
var preset app.AiPreset
|
||||
err := c.ShouldBindJSON(&preset)
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("获取预设失败!", zap.Error(err))
|
||||
response.FailWithMessage("获取预设失败", c)
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
userID := utils.GetUserID(c)
|
||||
|
||||
response.OkWithData(preset, c)
|
||||
if err := aiPresetService.UpdateAiPreset(&preset, userID); err != nil {
|
||||
global.GVA_LOG.Error("更新失败!", zap.Error(err))
|
||||
response.FailWithMessage("更新失败", c)
|
||||
} else {
|
||||
response.OkWithMessage("更新成功", c)
|
||||
}
|
||||
}
|
||||
|
||||
// GetAiPresetList 获取AI预设列表
|
||||
// FindAiPreset 查询预设
|
||||
// @Tags AiPreset
|
||||
// @Summary 获取AI预设列表
|
||||
// @Summary 查询预设
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param data query request.PageInfo true "分页信息"
|
||||
// @Param data query request.GetById true "ID"
|
||||
// @Success 200 {object} response.Response{data=app.AiPreset,msg=string} "查询成功"
|
||||
// @Router /aiPreset/findAiPreset [get]
|
||||
func (a *AiPresetApi) FindAiPreset(c *gin.Context) {
|
||||
var reqId request.GetById
|
||||
err := c.ShouldBindQuery(&reqId)
|
||||
if err != nil {
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
userID := utils.GetUserID(c)
|
||||
|
||||
if preset, err := aiPresetService.GetAiPreset(reqId.Uint(), userID); err != nil {
|
||||
global.GVA_LOG.Error("查询失败!", zap.Error(err))
|
||||
response.FailWithMessage("查询失败", c)
|
||||
} else {
|
||||
response.OkWithData(preset, c)
|
||||
}
|
||||
}
|
||||
|
||||
// GetAiPresetList 获取预设列表
|
||||
// @Tags AiPreset
|
||||
// @Summary 获取预设列表
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param data query request.PageInfo true "分页参数"
|
||||
// @Success 200 {object} response.Response{data=response.PageResult,msg=string} "获取成功"
|
||||
// @Router /app/preset/list [get]
|
||||
// @Router /aiPreset/getAiPresetList [get]
|
||||
func (a *AiPresetApi) GetAiPresetList(c *gin.Context) {
|
||||
var pageInfo commonRequest.PageInfo
|
||||
var pageInfo request.PageInfo
|
||||
err := c.ShouldBindQuery(&pageInfo)
|
||||
if err != nil {
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
userID := utils.GetUserID(c)
|
||||
|
||||
// 尝试获取用户ID,如果没有则使用0(公开访问)
|
||||
userId := uint(0)
|
||||
if id := utils.GetUserID(c); id > 0 {
|
||||
userId = id
|
||||
if list, total, err := aiPresetService.GetAiPresetList(pageInfo, userID); err != nil {
|
||||
global.GVA_LOG.Error("获取失败!", zap.Error(err))
|
||||
response.FailWithMessage("获取失败", c)
|
||||
} else {
|
||||
response.OkWithDetailed(response.PageResult{
|
||||
List: list,
|
||||
Total: total,
|
||||
Page: pageInfo.Page,
|
||||
PageSize: pageInfo.PageSize,
|
||||
}, "获取成功", c)
|
||||
}
|
||||
list, total, err := aiPresetService.GetAiPresetList(userId, pageInfo)
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("获取预设列表失败!", zap.Error(err))
|
||||
response.FailWithMessage("获取预设列表失败", c)
|
||||
return
|
||||
}
|
||||
|
||||
response.OkWithDetailed(response.PageResult{
|
||||
List: list,
|
||||
Total: total,
|
||||
Page: pageInfo.Page,
|
||||
PageSize: pageInfo.PageSize,
|
||||
}, "获取成功", c)
|
||||
}
|
||||
|
||||
// ImportAiPreset 导入AI预设(支持SillyTavern格式)
|
||||
// ImportAiPreset 导入预设
|
||||
// @Tags AiPreset
|
||||
// @Summary 导入AI预设
|
||||
// @Summary 导入预设(支持SillyTavern格式)
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param data body request.ImportAiPresetRequest true "导入数据"
|
||||
// @Success 200 {object} response.Response{data=app.AiPreset,msg=string} "导入成功"
|
||||
// @Router /app/preset/import [post]
|
||||
// @Param data body app.AiPreset true "预设JSON"
|
||||
// @Success 200 {object} response.Response{msg=string} "导入成功"
|
||||
// @Router /aiPreset/importAiPreset [post]
|
||||
func (a *AiPresetApi) ImportAiPreset(c *gin.Context) {
|
||||
var req request.ImportAiPresetRequest
|
||||
err := c.ShouldBindJSON(&req)
|
||||
var preset app.AiPreset
|
||||
err := c.ShouldBindJSON(&preset)
|
||||
if err != nil {
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
preset.UserID = utils.GetUserID(c)
|
||||
|
||||
// 尝试获取用户ID,如果没有则使用0(公开访问)
|
||||
userId := uint(0)
|
||||
if id := utils.GetUserID(c); id > 0 {
|
||||
userId = id
|
||||
if err := aiPresetService.CreateAiPreset(&preset); err != nil {
|
||||
global.GVA_LOG.Error("导入失败!", zap.Error(err))
|
||||
response.FailWithMessage("导入失败", c)
|
||||
} else {
|
||||
response.OkWithMessage("导入成功", c)
|
||||
}
|
||||
preset, err := aiPresetService.ImportAiPreset(userId, &req)
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("导入预设失败!", zap.Error(err))
|
||||
response.FailWithMessage("导入预设失败", c)
|
||||
return
|
||||
}
|
||||
|
||||
response.OkWithData(preset, c)
|
||||
}
|
||||
|
||||
// ExportAiPreset 导出AI预设
|
||||
// @Tags AiPreset
|
||||
// @Summary 导出AI预设
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param id path uint true "预设ID"
|
||||
// @Success 200 {object} map[string]interface{} "导出数据"
|
||||
// @Router /app/preset/:id/export [get]
|
||||
func (a *AiPresetApi) ExportAiPreset(c *gin.Context) {
|
||||
id, _ := strconv.ParseUint(c.Param("id"), 10, 32)
|
||||
// 尝试获取用户ID,如果没有则使用0(公开访问)
|
||||
userId := uint(0)
|
||||
if uid := utils.GetUserID(c); uid > 0 {
|
||||
userId = uid
|
||||
}
|
||||
|
||||
data, err := aiPresetService.ExportAiPreset(uint(id), userId)
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("导出预设失败!", zap.Error(err))
|
||||
response.FailWithMessage("导出预设失败", c)
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(200, data)
|
||||
}
|
||||
|
||||
@@ -2,7 +2,8 @@ package app
|
||||
|
||||
import (
|
||||
"git.echol.cn/loser/ai_proxy/server/global"
|
||||
"git.echol.cn/loser/ai_proxy/server/model/app/request"
|
||||
"git.echol.cn/loser/ai_proxy/server/model/app"
|
||||
"git.echol.cn/loser/ai_proxy/server/model/common/request"
|
||||
"git.echol.cn/loser/ai_proxy/server/model/common/response"
|
||||
"git.echol.cn/loser/ai_proxy/server/service"
|
||||
"git.echol.cn/loser/ai_proxy/server/utils"
|
||||
@@ -10,122 +11,141 @@ import (
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
type PresetBindingApi struct{}
|
||||
type AiPresetBindingApi struct{}
|
||||
|
||||
var presetBindingService = service.ServiceGroupApp.AppServiceGroup.PresetBindingService
|
||||
var aiPresetBindingService = service.ServiceGroupApp.AppServiceGroup.AiPresetBindingService
|
||||
|
||||
// CreateBinding 创建预设绑定
|
||||
// @Tags App
|
||||
// @Summary 创建预设绑定
|
||||
// CreateAiPresetBinding 创建绑定
|
||||
// @Tags AiPresetBinding
|
||||
// @Summary 创建绑定
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param data body request.CreateBindingRequest true "绑定信息"
|
||||
// @Param data body app.AiPresetBinding true "绑定信息"
|
||||
// @Success 200 {object} response.Response{msg=string} "创建成功"
|
||||
// @Router /app/binding [post]
|
||||
func (a *PresetBindingApi) CreateBinding(c *gin.Context) {
|
||||
var req request.CreateBindingRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
|
||||
err := presetBindingService.CreateBinding(&req)
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("创建绑定失败!", zap.Error(err))
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
|
||||
response.OkWithMessage("创建成功", c)
|
||||
}
|
||||
|
||||
// UpdateBinding 更新预设绑定
|
||||
// @Tags App
|
||||
// @Summary 更新预设绑定
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param data body request.UpdateBindingRequest true "绑定信息"
|
||||
// @Success 200 {object} response.Response{msg=string} "更新成功"
|
||||
// @Router /app/binding [put]
|
||||
func (a *PresetBindingApi) UpdateBinding(c *gin.Context) {
|
||||
var req request.UpdateBindingRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
|
||||
err := presetBindingService.UpdateBinding(&req)
|
||||
// @Router /aiPresetBinding/createAiPresetBinding [post]
|
||||
func (a *AiPresetBindingApi) CreateAiPresetBinding(c *gin.Context) {
|
||||
var binding app.AiPresetBinding
|
||||
err := c.ShouldBindJSON(&binding)
|
||||
if err != nil {
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
binding.UserID = utils.GetUserID(c)
|
||||
|
||||
response.OkWithMessage("更新成功", c)
|
||||
if err := aiPresetBindingService.CreateAiPresetBinding(&binding); err != nil {
|
||||
global.GVA_LOG.Error("创建失败!", zap.Error(err))
|
||||
response.FailWithMessage("创建失败", c)
|
||||
} else {
|
||||
response.OkWithMessage("创建成功", c)
|
||||
}
|
||||
}
|
||||
|
||||
// DeleteBinding 删除预设绑定
|
||||
// @Tags App
|
||||
// @Summary 删除预设绑定
|
||||
// DeleteAiPresetBinding 删除绑定
|
||||
// @Tags AiPresetBinding
|
||||
// @Summary 删除绑定
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param id path uint true "绑定ID"
|
||||
// @Param data body request.GetById true "ID"
|
||||
// @Success 200 {object} response.Response{msg=string} "删除成功"
|
||||
// @Router /app/binding/:id [delete]
|
||||
func (a *PresetBindingApi) DeleteBinding(c *gin.Context) {
|
||||
id, err := utils.StringToUint(c.Param("id"))
|
||||
if err != nil {
|
||||
response.FailWithMessage("无效的ID", c)
|
||||
return
|
||||
}
|
||||
|
||||
err = presetBindingService.DeleteBinding(id)
|
||||
// @Router /aiPresetBinding/deleteAiPresetBinding [delete]
|
||||
func (a *AiPresetBindingApi) DeleteAiPresetBinding(c *gin.Context) {
|
||||
var reqId request.GetById
|
||||
err := c.ShouldBindJSON(&reqId)
|
||||
if err != nil {
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
userID := utils.GetUserID(c)
|
||||
|
||||
response.OkWithMessage("删除成功", c)
|
||||
if err := aiPresetBindingService.DeleteAiPresetBinding(reqId.Uint(), userID); err != nil {
|
||||
global.GVA_LOG.Error("删除失败!", zap.Error(err))
|
||||
response.FailWithMessage("删除失败", c)
|
||||
} else {
|
||||
response.OkWithMessage("删除成功", c)
|
||||
}
|
||||
}
|
||||
|
||||
// GetBindingList 获取绑定列表
|
||||
// @Tags App
|
||||
// UpdateAiPresetBinding 更新绑定
|
||||
// @Tags AiPresetBinding
|
||||
// @Summary 更新绑定
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param data body app.AiPresetBinding true "绑定信息"
|
||||
// @Success 200 {object} response.Response{msg=string} "更新成功"
|
||||
// @Router /aiPresetBinding/updateAiPresetBinding [put]
|
||||
func (a *AiPresetBindingApi) UpdateAiPresetBinding(c *gin.Context) {
|
||||
var binding app.AiPresetBinding
|
||||
err := c.ShouldBindJSON(&binding)
|
||||
if err != nil {
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
userID := utils.GetUserID(c)
|
||||
|
||||
if err := aiPresetBindingService.UpdateAiPresetBinding(&binding, userID); err != nil {
|
||||
global.GVA_LOG.Error("更新失败!", zap.Error(err))
|
||||
response.FailWithMessage("更新失败", c)
|
||||
} else {
|
||||
response.OkWithMessage("更新成功", c)
|
||||
}
|
||||
}
|
||||
|
||||
// FindAiPresetBinding 查询绑定
|
||||
// @Tags AiPresetBinding
|
||||
// @Summary 查询绑定
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param data query request.GetById true "ID"
|
||||
// @Success 200 {object} response.Response{data=app.AiPresetBinding,msg=string} "查询成功"
|
||||
// @Router /aiPresetBinding/findAiPresetBinding [get]
|
||||
func (a *AiPresetBindingApi) FindAiPresetBinding(c *gin.Context) {
|
||||
var reqId request.GetById
|
||||
err := c.ShouldBindQuery(&reqId)
|
||||
if err != nil {
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
userID := utils.GetUserID(c)
|
||||
|
||||
if binding, err := aiPresetBindingService.GetAiPresetBinding(reqId.Uint(), userID); err != nil {
|
||||
global.GVA_LOG.Error("查询失败!", zap.Error(err))
|
||||
response.FailWithMessage("查询失败", c)
|
||||
} else {
|
||||
response.OkWithData(binding, c)
|
||||
}
|
||||
}
|
||||
|
||||
// GetAiPresetBindingList 获取绑定列表
|
||||
// @Tags AiPresetBinding
|
||||
// @Summary 获取绑定列表
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param page query int false "页码"
|
||||
// @Param pageSize query int false "每页数量"
|
||||
// @Param providerId query uint false "提供商ID"
|
||||
// @Param presetId query uint false "预设ID"
|
||||
// @Success 200 {object} response.Response{data=response.PageResult} "获取成功"
|
||||
// @Router /app/binding/list [get]
|
||||
func (a *PresetBindingApi) GetBindingList(c *gin.Context) {
|
||||
var req request.GetBindingListRequest
|
||||
if err := c.ShouldBindQuery(&req); err != nil {
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
|
||||
if req.Page == 0 {
|
||||
req.Page = 1
|
||||
}
|
||||
if req.PageSize == 0 {
|
||||
req.PageSize = 10
|
||||
}
|
||||
|
||||
list, total, err := presetBindingService.GetBindingList(&req)
|
||||
// @Param data query request.PageInfo true "分页参数"
|
||||
// @Success 200 {object} response.Response{data=response.PageResult,msg=string} "获取成功"
|
||||
// @Router /aiPresetBinding/getAiPresetBindingList [get]
|
||||
func (a *AiPresetBindingApi) GetAiPresetBindingList(c *gin.Context) {
|
||||
var pageInfo request.PageInfo
|
||||
err := c.ShouldBindQuery(&pageInfo)
|
||||
if err != nil {
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
userID := utils.GetUserID(c)
|
||||
|
||||
response.OkWithDetailed(response.PageResult{
|
||||
List: list,
|
||||
Total: total,
|
||||
Page: req.Page,
|
||||
PageSize: req.PageSize,
|
||||
}, "获取成功", c)
|
||||
if list, total, err := aiPresetBindingService.GetAiPresetBindingList(pageInfo, userID); err != nil {
|
||||
global.GVA_LOG.Error("获取失败!", zap.Error(err))
|
||||
response.FailWithMessage("获取失败", c)
|
||||
} else {
|
||||
response.OkWithDetailed(response.PageResult{
|
||||
List: list,
|
||||
Total: total,
|
||||
Page: pageInfo.Page,
|
||||
PageSize: pageInfo.PageSize,
|
||||
}, "获取成功", c)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,12 +1,12 @@
|
||||
package app
|
||||
|
||||
import (
|
||||
"strconv"
|
||||
|
||||
"git.echol.cn/loser/ai_proxy/server/global"
|
||||
"git.echol.cn/loser/ai_proxy/server/model/app/request"
|
||||
"git.echol.cn/loser/ai_proxy/server/model/app"
|
||||
"git.echol.cn/loser/ai_proxy/server/model/common/request"
|
||||
"git.echol.cn/loser/ai_proxy/server/model/common/response"
|
||||
"git.echol.cn/loser/ai_proxy/server/service"
|
||||
"git.echol.cn/loser/ai_proxy/server/utils"
|
||||
"github.com/gin-gonic/gin"
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
@@ -15,171 +15,137 @@ type AiProviderApi struct{}
|
||||
|
||||
var aiProviderService = service.ServiceGroupApp.AppServiceGroup.AiProviderService
|
||||
|
||||
// CreateAiProvider 创建AI提供商
|
||||
// CreateAiProvider 创建提供商
|
||||
// @Tags AiProvider
|
||||
// @Summary 创建AI提供商
|
||||
// @Summary 创建提供商
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param data body request.CreateAiProviderRequest true "提供商信息"
|
||||
// @Success 200 {object} response.Response{data=app.AiProvider,msg=string} "创建成功"
|
||||
// @Router /app/provider [post]
|
||||
// @Param data body app.AiProvider true "提供商信息"
|
||||
// @Success 200 {object} response.Response{msg=string} "创建成功"
|
||||
// @Router /aiProvider/createAiProvider [post]
|
||||
func (a *AiProviderApi) CreateAiProvider(c *gin.Context) {
|
||||
var req request.CreateAiProviderRequest
|
||||
err := c.ShouldBindJSON(&req)
|
||||
var provider app.AiProvider
|
||||
err := c.ShouldBindJSON(&provider)
|
||||
if err != nil {
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
provider.UserID = utils.GetUserID(c)
|
||||
|
||||
provider, err := aiProviderService.CreateAiProvider(&req)
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("创建提供商失败!", zap.Error(err))
|
||||
response.FailWithMessage("创建提供商失败", c)
|
||||
return
|
||||
if err := aiProviderService.CreateAiProvider(&provider); err != nil {
|
||||
global.GVA_LOG.Error("创建失败!", zap.Error(err))
|
||||
response.FailWithMessage("创建失败", c)
|
||||
} else {
|
||||
response.OkWithMessage("创建成功", c)
|
||||
}
|
||||
|
||||
response.OkWithData(provider, c)
|
||||
}
|
||||
|
||||
// DeleteAiProvider 删除AI提供商
|
||||
// DeleteAiProvider 删除提供商
|
||||
// @Tags AiProvider
|
||||
// @Summary 删除AI提供商
|
||||
// @Summary 删除提供商
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param id path uint true "提供商ID"
|
||||
// @Param data body request.GetById true "ID"
|
||||
// @Success 200 {object} response.Response{msg=string} "删除成功"
|
||||
// @Router /app/provider/:id [delete]
|
||||
// @Router /aiProvider/deleteAiProvider [delete]
|
||||
func (a *AiProviderApi) DeleteAiProvider(c *gin.Context) {
|
||||
id, _ := strconv.ParseUint(c.Param("id"), 10, 32)
|
||||
|
||||
err := aiProviderService.DeleteAiProvider(uint(id))
|
||||
var reqId request.GetById
|
||||
err := c.ShouldBindJSON(&reqId)
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("删除提供商失败!", zap.Error(err))
|
||||
response.FailWithMessage("删除提供商失败", c)
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
userID := utils.GetUserID(c)
|
||||
|
||||
response.OkWithMessage("删除成功", c)
|
||||
if err := aiProviderService.DeleteAiProvider(reqId.Uint(), userID); err != nil {
|
||||
global.GVA_LOG.Error("删除失败!", zap.Error(err))
|
||||
response.FailWithMessage("删除失败", c)
|
||||
} else {
|
||||
response.OkWithMessage("删除成功", c)
|
||||
}
|
||||
}
|
||||
|
||||
// UpdateAiProvider 更新AI提供商
|
||||
// UpdateAiProvider 更新提供商
|
||||
// @Tags AiProvider
|
||||
// @Summary 更新AI提供商
|
||||
// @Summary 更新提供商
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param data body request.UpdateAiProviderRequest true "提供商信息"
|
||||
// @Success 200 {object} response.Response{data=app.AiProvider,msg=string} "更新成功"
|
||||
// @Router /app/provider [put]
|
||||
// @Param data body app.AiProvider true "提供商信息"
|
||||
// @Success 200 {object} response.Response{msg=string} "更新成功"
|
||||
// @Router /aiProvider/updateAiProvider [put]
|
||||
func (a *AiProviderApi) UpdateAiProvider(c *gin.Context) {
|
||||
var req request.UpdateAiProviderRequest
|
||||
err := c.ShouldBindJSON(&req)
|
||||
var provider app.AiProvider
|
||||
err := c.ShouldBindJSON(&provider)
|
||||
if err != nil {
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
userID := utils.GetUserID(c)
|
||||
|
||||
provider, err := aiProviderService.UpdateAiProvider(&req)
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("更新提供商失败!", zap.Error(err))
|
||||
response.FailWithMessage("更新提供商失败", c)
|
||||
return
|
||||
if err := aiProviderService.UpdateAiProvider(&provider, userID); err != nil {
|
||||
global.GVA_LOG.Error("更新失败!", zap.Error(err))
|
||||
response.FailWithMessage("更新失败", c)
|
||||
} else {
|
||||
response.OkWithMessage("更新成功", c)
|
||||
}
|
||||
|
||||
response.OkWithData(provider, c)
|
||||
}
|
||||
|
||||
// GetAiProvider 获取AI提供商详情
|
||||
// FindAiProvider 查询提供商
|
||||
// @Tags AiProvider
|
||||
// @Summary 获取AI提供商详情
|
||||
// @Summary 查询提供商
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param id path uint true "提供商ID"
|
||||
// @Success 200 {object} response.Response{data=app.AiProvider,msg=string} "获取成功"
|
||||
// @Router /app/provider/:id [get]
|
||||
func (a *AiProviderApi) GetAiProvider(c *gin.Context) {
|
||||
id, _ := strconv.ParseUint(c.Param("id"), 10, 32)
|
||||
|
||||
provider, err := aiProviderService.GetAiProvider(uint(id))
|
||||
// @Param data query request.GetById true "ID"
|
||||
// @Success 200 {object} response.Response{data=app.AiProvider,msg=string} "查询成功"
|
||||
// @Router /aiProvider/findAiProvider [get]
|
||||
func (a *AiProviderApi) FindAiProvider(c *gin.Context) {
|
||||
var reqId request.GetById
|
||||
err := c.ShouldBindQuery(&reqId)
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("获取提供商失败!", zap.Error(err))
|
||||
response.FailWithMessage("获取提供商失败", c)
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
userID := utils.GetUserID(c)
|
||||
|
||||
response.OkWithData(provider, c)
|
||||
if provider, err := aiProviderService.GetAiProvider(reqId.Uint(), userID); err != nil {
|
||||
global.GVA_LOG.Error("查询失败!", zap.Error(err))
|
||||
response.FailWithMessage("查询失败", c)
|
||||
} else {
|
||||
response.OkWithData(provider, c)
|
||||
}
|
||||
}
|
||||
|
||||
// GetAiProviderList 获取AI提供商列表
|
||||
// GetAiProviderList 获取提供商列表
|
||||
// @Tags AiProvider
|
||||
// @Summary 获取AI提供商列表
|
||||
// @Summary 获取提供商列表
|
||||
// @Security ApiKeyAuth
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Success 200 {object} response.Response{data=[]app.AiProvider,msg=string} "获取成功"
|
||||
// @Router /app/provider/list [get]
|
||||
// @Param data query request.PageInfo true "分页参数"
|
||||
// @Success 200 {object} response.Response{data=response.PageResult,msg=string} "获取成功"
|
||||
// @Router /aiProvider/getAiProviderList [get]
|
||||
func (a *AiProviderApi) GetAiProviderList(c *gin.Context) {
|
||||
list, err := aiProviderService.GetAiProviderList()
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("获取提供商列表失败!", zap.Error(err))
|
||||
response.FailWithMessage("获取提供商列表失败", c)
|
||||
return
|
||||
}
|
||||
|
||||
response.OkWithData(list, c)
|
||||
}
|
||||
|
||||
// TestConnection 测试连接
|
||||
// @Tags AiProvider
|
||||
// @Summary 测试AI提供商连接
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param data body request.TestConnectionRequest true "连接信息"
|
||||
// @Success 200 {object} response.Response{data=response.TestConnectionResponse,msg=string} "测试成功"
|
||||
// @Router /app/provider/test [post]
|
||||
func (a *AiProviderApi) TestConnection(c *gin.Context) {
|
||||
var req request.TestConnectionRequest
|
||||
err := c.ShouldBindJSON(&req)
|
||||
var pageInfo request.PageInfo
|
||||
err := c.ShouldBindQuery(&pageInfo)
|
||||
if err != nil {
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
userID := utils.GetUserID(c)
|
||||
|
||||
result, err := aiProviderService.TestConnection(&req)
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("测试连接失败!", zap.Error(err))
|
||||
response.FailWithMessage("测试连接失败", c)
|
||||
return
|
||||
if list, total, err := aiProviderService.GetAiProviderList(pageInfo, userID); err != nil {
|
||||
global.GVA_LOG.Error("获取失败!", zap.Error(err))
|
||||
response.FailWithMessage("获取失败", c)
|
||||
} else {
|
||||
response.OkWithDetailed(response.PageResult{
|
||||
List: list,
|
||||
Total: total,
|
||||
Page: pageInfo.Page,
|
||||
PageSize: pageInfo.PageSize,
|
||||
}, "获取成功", c)
|
||||
}
|
||||
|
||||
response.OkWithData(result, c)
|
||||
}
|
||||
|
||||
// GetModels 获取模型列表
|
||||
// @Tags AiProvider
|
||||
// @Summary 获取AI提供商的模型列表
|
||||
// @accept application/json
|
||||
// @Produce application/json
|
||||
// @Param data body request.GetModelsRequest true "提供商信息"
|
||||
// @Success 200 {object} response.Response{data=[]response.ModelInfo,msg=string} "获取成功"
|
||||
// @Router /app/provider/models [post]
|
||||
func (a *AiProviderApi) GetModels(c *gin.Context) {
|
||||
var req request.GetModelsRequest
|
||||
err := c.ShouldBindJSON(&req)
|
||||
if err != nil {
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
|
||||
models, err := aiProviderService.GetModels(&req)
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("获取模型列表失败!", zap.Error(err))
|
||||
response.FailWithMessage(err.Error(), c)
|
||||
return
|
||||
}
|
||||
|
||||
response.OkWithData(models, c)
|
||||
}
|
||||
|
||||
@@ -1,8 +1,8 @@
|
||||
package app
|
||||
|
||||
type ApiGroup struct {
|
||||
AiPresetApi AiPresetApi
|
||||
AiProviderApi AiProviderApi
|
||||
AiProxyApi AiProxyApi
|
||||
PresetBindingApi PresetBindingApi
|
||||
AiProxyApi
|
||||
AiPresetApi
|
||||
AiProviderApi
|
||||
AiPresetBindingApi
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user