🎨 优化项目结构 && 完善ai配置

This commit is contained in:
2026-03-03 15:39:23 +08:00
parent 557c865948
commit 2714e63d2a
585 changed files with 62223 additions and 100018 deletions

View File

@@ -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)
}

View File

@@ -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)
}
}

View File

@@ -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)
}

View File

@@ -1,8 +1,8 @@
package app
type ApiGroup struct {
AiPresetApi AiPresetApi
AiProviderApi AiProviderApi
AiProxyApi AiProxyApi
PresetBindingApi PresetBindingApi
AiProxyApi
AiPresetApi
AiProviderApi
AiPresetBindingApi
}