🎨 优化项目结构 && 完善ai配置
This commit is contained in:
@@ -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)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user