red-future/model-gateway
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(service): 重构模型网关服务结构

Browse Source
This commit is contained in:
王利钊
2026-06-11 17:58:49 +08:00
parent afd60caf56
commit 1c6c9bae14
34 changed files with 784 additions and 1223 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
model/dto/model_gateway_logs_stat_dto.go Normal file
View File

@@ -0,0 +1,20 @@
package dto
import "github.com/gogf/gf/v2/frame/g"
// ListModelStatReq 统计列表
type ListModelStatReq struct {
g.Meta `path:"/listModelStat" method:"get" tags:"统计" summary:"模型请求统计列表" dc:"按天统计模型请求次数，支持分页与条件筛选"`
PageNum int `p:"pageNum" json:"pageNum" dc:"页码（默认1）"`
PageSize int `p:"pageSize" json:"pageSize" dc:"每页条数（默认10）"`
StartDay string `p:"startDay" json:"startDay" dc:"开始日期（YYYY-MM-DD，可选）"`
EndDay string `p:"endDay" json:"endDay" dc:"结束日期（YYYY-MM-DD，可选）"`
TenantID *int64 `p:"tenantId" json:"tenantId" dc:"租户ID（可选）"`
Creator string `p:"creator" json:"creator" dc:"创建人（可选，模糊匹配）"`
ModelName string `p:"modelName" json:"modelName" dc:"模型名称（可选，模糊匹配）"`
}
type ListModelStatRes struct {
List any `json:"list" dc:"列表数据"`
Total int64 `json:"total" dc:"总数"`
}

2
model/dto/model_dto.go → model/dto/model_gateway_models_dto.go
View File

@@ -103,7 +103,7 @@ type GetModelReq struct {
}
type GetModelRes struct {
Model *entity.AsynchModel `json:"model" dc:"模型配置详情"`
Model *entity.ModelGatewayModel `json:"model" dc:"模型配置详情"`
}
// ListModelReq 配置列表

24
model/dto/task_dto.go → model/dto/model_gateway_task_dto.go
View File

@@ -1,6 +1,8 @@
package dto
import "github.com/gogf/gf/v2/frame/g"
import (
"github.com/gogf/gf/v2/frame/g"
)
// CreateTaskReq 创建异步任务
type CreateTaskReq struct {
@@ -8,7 +10,6 @@ type CreateTaskReq struct {
ModelName string `p:"modelName" json:"modelName" v:"required#modelName不能为空" dc:"模型名称"`
BizName string `p:"bizName" json:"bizName" dc:"业务名称（调用方模块/系统，用于统计）"`
CallbackUrl string `p:"callbackUrl" json:"callbackUrl" dc:"回调地址（可选，用于后续业务通知）"`
InputRef string `p:"inputRef" json:"inputRef" dc:"输入引用（如OSS/文件引用等）"`
RequestPayload map[string]any `p:"requestPayload" json:"requestPayload" dc:"请求负载（透传给模型服务）"`
EpicycleId int64 `json:"epicycleId" dc:"轮次ID"`
BuildType int64 `json:"buildType" dc:"构建类型：1-提示词构建 2-节点构建"`
@@ -68,9 +69,10 @@ type GetTaskBatchReq struct {
}
type GetTaskBatchItem struct {
TaskID string `json:"taskId" dc:"任务ID"`
State int `json:"state" dc:"任务状态"`
OssFile string `json:"ossFile" dc:"结果文件OSS地址"`
TaskID string `json:"taskId" dc:"任务ID"`
State int `json:"state" dc:"任务状态"`
OssFile string `json:"ossFile" dc:"结果文件OSS地址"`
TextResult map[string]any `json:"textResult" dc:"文本结果"`
}
type GetTaskBatchRes struct {
@@ -83,8 +85,9 @@ type ListTaskReq struct {
PageNum int `p:"pageNum" json:"pageNum" dc:"页码（默认1）"`
PageSize int `p:"pageSize" json:"pageSize" dc:"每页条数（默认10）"`
ModelName string `p:"modelName" json:"modelName" dc:"模型名称（模糊匹配）"`
BizName string `p:"bizName" json:"bizName" dc:"业务名称"`
TaskID string `p:"taskId" json:"taskId" dc:"任务ID（模糊匹配）"`
State *int `p:"state" json:"state" dc:"任务状态（0/1/2/3/4，可选）"`
State int `p:"state" json:"state" dc:"任务状态（0/1/2/3/4，可选）"`
}
type ListTaskRes struct {
@@ -102,12 +105,3 @@ type RunWorkReq struct {
type RunWorkRes struct {
Claimed int `json:"claimed" dc:"本次抢占并处理的任务数"`
}
// CleanWorkReq 手动触发 cleaner 执行一次（由上层定时任务调用）
type CleanWorkReq struct {
g.Meta `path:"/cleanWork" method:"post" tags:"任务管理" summary:"执行一次Cleaner" dc:"手动触发一次清理/重试（用于由上层定时任务控制）"`
}
type CleanWorkRes struct {
Ok bool `json:"ok" dc:"是否执行成功"`
}

20
model/dto/stat_dto.go
View File

@@ -1,20 +0,0 @@
package dto
import "github.com/gogf/gf/v2/frame/g"
// ListModelStatReq 统计列表
type ListModelStatReq struct {
g.Meta `path:"/listModelStat" method:"get" tags:"统计" summary:"模型请求统计列表" dc:"按天统计模型请求次数，支持分页与条件筛选"`
PageNum int `p:"pageNum" json:"pageNum" dc:"页码（默认1）"`
PageSize int `p:"pageSize" json:"pageSize" dc:"每页条数（默认10）"`
StartDay string `p:"startDay" json:"startDay" dc:"开始日期（YYYY-MM-DD，可选）"`
EndDay string `p:"endDay" json:"endDay" dc:"结束日期（YYYY-MM-DD，可选）"`
TenantID *int64 `p:"tenantId" json:"tenantId" dc:"租户ID（可选）"`
Creator string `p:"creator" json:"creator" dc:"创建人（可选，模糊匹配）"`
ModelName string `p:"modelName" json:"modelName" dc:"模型名称（可选，模糊匹配）"`
}
type ListModelStatRes struct {
List any `json:"list" dc:"列表数据"`
Total int64 `json:"total" dc:"总数"`
}