57 lines
1.9 KiB
Go
57 lines
1.9 KiB
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
"model-gateway/model/dto"
|
|
"model-gateway/service"
|
|
|
|
"gitea.redpowerfuture.com/red-future/common/beans"
|
|
)
|
|
|
|
// ModelManage 模型配置控制器
|
|
var ModelManage = new(modelManage)
|
|
|
|
type modelManage struct{}
|
|
|
|
// CreateModel 添加配置
|
|
func (c *modelManage) CreateModel(ctx context.Context, req *dto.CreateModelManageReq) (res *dto.CreateModelManageRes, err error) {
|
|
return service.ModelManage.Create(ctx, req)
|
|
}
|
|
|
|
// UpdateModel 更改配置
|
|
func (c *modelManage) UpdateModel(ctx context.Context, req *dto.UpdateModelManageReq) (res *beans.ResponseEmpty, err error) {
|
|
err = service.ModelManage.Update(ctx, req)
|
|
return
|
|
}
|
|
|
|
// DeleteModel 删除配置
|
|
func (c *modelManage) DeleteModel(ctx context.Context, req *dto.DeleteModelManageReq) (res *beans.ResponseEmpty, err error) {
|
|
err = service.ModelManage.Delete(ctx, req)
|
|
return
|
|
}
|
|
|
|
// GetModel 获取配置
|
|
func (c *modelManage) GetModel(ctx context.Context, req *dto.GetModelManageReq) (res *dto.GetModelManageRes, err error) {
|
|
return service.ModelManage.Get(ctx, req)
|
|
}
|
|
|
|
// ListModel 配置列表
|
|
func (c *modelManage) ListModel(ctx context.Context, req *dto.ListModelManageReq) (res *dto.ListModelManageRes, err error) {
|
|
return service.ModelManage.List(ctx, req)
|
|
}
|
|
|
|
// CheckChatModel 检查是否为聊天模型
|
|
func (c *modelManage) CheckChatModel(ctx context.Context, req *dto.CheckChatModelReq) (res *dto.CheckChatModelRes, err error) {
|
|
return service.ModelManage.CheckChatModel(ctx, req)
|
|
}
|
|
|
|
// ListType 模型类型列表
|
|
func (c *modelManage) ListType(ctx context.Context, req *dto.ModelTypeReq) (res *dto.ModelTypeRes, err error) {
|
|
return service.ModelManage.GetModelType(ctx, req)
|
|
}
|
|
|
|
// ListOperator 运营商列表
|
|
func (c *modelManage) ListOperator(ctx context.Context, req *dto.ModelSupplierReq) (res *dto.ModelSupplierRes, err error) {
|
|
return service.ModelManage.GetModelSupplier(ctx, req)
|
|
}
|