qhd
03c95c3601
- 新增 ActivePull 实体、DAO、DTO 及 Service，支持主动拉取任务管理 - 新增 ComposeCallback、VideoCallback、HttpNodeCallback 多类型回调接口 - FlowExecution 增加 NodeGroupId 和 TotalTokens 字段，支持节点组追踪与 Token 统计 - ExecutedNodes 结构由字符串列表改为包含执行状态的节点对象列表 - 重构回调通知机制，统一 Notify 函数调用 - 优化输出项类型判断逻辑，新增文件类型标识
289 lines
12 KiB
Go
289 lines
12 KiB
Go
package flow
|
||
|
||
import (
|
||
"ai-agent/workflow/consts/flow"
|
||
"ai-agent/workflow/consts/node"
|
||
"ai-agent/workflow/model/entity"
|
||
|
||
"gitea.com/red-future/common/beans"
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"github.com/gogf/gf/v2/os/gtime"
|
||
)
|
||
|
||
// NodeExecutionInput 节点执行入参（包含配置+表单架构）
|
||
type NodeExecutionInput struct {
|
||
Config *entity.FlowNode `json:"config"` // 节点配置
|
||
Global *FlowExecutionInput `json:"global"`
|
||
NodeExecutionId int64 `json:"nodeExecutionId"`
|
||
}
|
||
|
||
// ExecutedNode 已执行节点记录，包含节点ID和执行状态
|
||
type ExecutedNode struct {
|
||
NodeId string `json:"nodeId"`
|
||
Status node.NodeExecutionStatus `json:"status"` // 执行状态：成功/失败
|
||
}
|
||
|
||
// FlowExecutionInput 工作流执行入参（全程不变）
|
||
type FlowExecutionInput struct {
|
||
NodeGroupId string `json:"nodeGroupId"`
|
||
IsDialogue bool `json:"isDialogue"`
|
||
ExecutionId int64 `json:"executionId"`
|
||
ConfigMap map[string]*entity.FlowNode `json:"configMap"`
|
||
SessionId string `json:"sessionId" dc:"会话ID"`
|
||
Desc string `json:"desc"`
|
||
SkillName string `json:"skillName"`
|
||
FileUrl []string `json:"fileUrl"`
|
||
ExecutedNodes []ExecutedNode `json:"executedNodes"` // 已执行节点列表，包含执行状态
|
||
}
|
||
|
||
type GetIsChatModelRes struct {
|
||
Model struct {
|
||
ModelName string `json:"modelName"`
|
||
ResponseBody map[string]any `json:"responseBody"`
|
||
}
|
||
}
|
||
|
||
type GetModelInfoReq struct {
|
||
ModelName string `json:"modelName"`
|
||
}
|
||
|
||
type GetModelInfoRes struct {
|
||
Model struct {
|
||
LastFrame string `json:"lastFrame"`
|
||
ResponseTokenField string `json:"responseTokenField"`
|
||
ResponseMapping map[string]any `json:"responseMapping"`
|
||
ResponseBody string `json:"responseBody"`
|
||
//QueryConfig struct {
|
||
// ResponseType string `json:"responseType"`
|
||
// CallbackUrl string `json:"callbackUrl"`
|
||
// Method string `json:"method"`
|
||
// Url string `json:"url"`
|
||
// Headers map[string]any `json:"headers"`
|
||
// Body map[string]any `json:"body"`
|
||
// Response []map[string]any `json:"response"`
|
||
// ResponseBody string `json:"responseBody"`
|
||
// ResponseTokenField string `json:"responseTokenField"`
|
||
//} `json:"queryConfig"`
|
||
} `json:"model"`
|
||
}
|
||
|
||
type ComposeMessagesReq struct {
|
||
BuildType int `json:"buildType"`
|
||
ModelName string `json:"modelName"`
|
||
SkillName string `json:"skillName"`
|
||
CallbackUrl string `json:"callbackUrl"`
|
||
Form []map[string]any `json:"form"`
|
||
UserForm []map[string]any `json:"userForm"`
|
||
Consult []Consult `json:"consult"`
|
||
SessionId string `json:"sessionId" dc:"会话ID"`
|
||
NodeId string `json:"nodeId"`
|
||
Cause string `json:"cause"`
|
||
}
|
||
|
||
type Consult struct {
|
||
Type string `json:"type"`
|
||
Url string `json:"url"`
|
||
}
|
||
|
||
type ComposeMessagesRes struct {
|
||
TaskId string `json:"taskId"`
|
||
}
|
||
|
||
type VideoConcatReq struct {
|
||
VideoUrls []string `json:"video_urls"`
|
||
Method string `json:"method"`
|
||
Upload bool `json:"upload"`
|
||
CallbackUrl string `json:"callback_url"`
|
||
}
|
||
|
||
type VideoConcatRes struct {
|
||
TaskId string `json:"taskId"`
|
||
}
|
||
|
||
type ModelGatewayReq struct {
|
||
ModelName string `json:"modelName"`
|
||
ModelKey string `json:"modelKey"`
|
||
BizName string `json:"bizName"`
|
||
CallbackUrl string `json:"callbackUrl"`
|
||
InputRef string `json:"inputRef"`
|
||
RequestPayload map[string]any `json:"requestPayload"`
|
||
EpicycleId int64 `json:"epicycleId" dc:"轮次ID"`
|
||
}
|
||
|
||
type ModelGatewayRes struct {
|
||
TaskId string `json:"taskId"`
|
||
}
|
||
|
||
type ComposeCallbackReq struct {
|
||
g.Meta `path:"/composeCallBack" method:"post" tags:"提示词处理" summary:"提示词 回调" dc:"提示词 成功后 GET 回调：callbackUrl/{bizName}"`
|
||
TaskId string `json:"taskId"`
|
||
Status string `json:"status"`
|
||
Messages struct {
|
||
TotalRounds int `json:"total_rounds"` // 总轮数
|
||
Rounds []map[string]any `json:"rounds"` // 每轮详情（动态类型）
|
||
} `json:"messages,omitempty"`
|
||
EpicycleId int64 `json:"epicycleId"`
|
||
ErrorMsg string `json:"errorMsg,omitempty"`
|
||
}
|
||
|
||
type ModelCallbackReq struct {
|
||
g.Meta `path:"/modelCallback" method:"post" tags:"提示词处理" summary:"model-gateway 回调" dc:"model-gateway 成功后 GET 回调：callbackUrl/{bizName}"`
|
||
TaskId string `p:"task_id" json:"task_id" v:"required#task_id不能为空" dc:"网关任务ID"`
|
||
State int `p:"state" json:"state" dc:"网关任务状态"`
|
||
OssFile string `p:"oss_file" json:"oss_file" dc:"结果文件地址"`
|
||
FileType string `p:"file_type" json:"file_type" dc:"结果文件类型"`
|
||
Messages map[string]any `json:"messages"`
|
||
ErrorMsg string `json:"error_msg"`
|
||
}
|
||
|
||
type VideoCallbackReq struct {
|
||
g.Meta `path:"/videoCallback" method:"post" tags:"视频处理" summary:"media 回调" dc:"media 成功后 GET 回调：callbackUrl/{bizName}"`
|
||
TaskId string `json:"taskId"`
|
||
FileURL string `json:"fileUrl"`
|
||
}
|
||
|
||
//=============================================================================
|
||
|
||
// 原始入参结构体
|
||
type Word struct {
|
||
Confidence float64 `json:"confidence"`
|
||
StartTime float64 `json:"startTime"`
|
||
EndTime float64 `json:"endTime"`
|
||
Word string `json:"word"`
|
||
}
|
||
type Sentence struct {
|
||
EndTime float64 `json:"endTime"`
|
||
StartTime float64 `json:"startTime"`
|
||
Text string `json:"text"`
|
||
Words []Word `json:"words"`
|
||
}
|
||
type InputData struct {
|
||
Data struct {
|
||
Sentences []Sentence `json:"sentences"`
|
||
} `json:"data"`
|
||
}
|
||
|
||
// 输出目标结构体（对应截图subtitles格式）
|
||
type Subtitle struct {
|
||
Start float64 `json:"start"`
|
||
End float64 `json:"end"`
|
||
Text string `json:"text"`
|
||
}
|
||
|
||
//==============================================================================
|
||
|
||
type ExecuteReq struct {
|
||
g.Meta `path:"/execute" method:"post" tags:"任务管理" summary:"执行任务" dc:"执行任务"`
|
||
|
||
FlowId int64 `json:"flowId" dc:"用户流程ID"`
|
||
FlowName string `json:"flowName"`
|
||
NodeGroupId string `json:"nodeGroupId"`
|
||
FlowContent *entity.FlowInfo `json:"flowContent" description:"流程内容"`
|
||
NodeInputParams []*entity.FlowNode `json:"nodeInputParams" description:"节点输入参数"`
|
||
SessionId string `json:"sessionId" dc:"会话ID"`
|
||
Desc string `json:"desc"`
|
||
SkillName string `json:"skillName"`
|
||
FileUrl []string `json:"fileUrl"`
|
||
ResultUrl string `json:"resultUrl"`
|
||
}
|
||
|
||
type ExecuteRes struct {
|
||
Id int64 `json:"id,string" dc:"执行记录ID，用于查询执行状态和结果"`
|
||
}
|
||
|
||
type CancelReq struct {
|
||
g.Meta `path:"/cancel" method:"post" tags:"任务管理" summary:"取消任务" dc:"取消任务"`
|
||
|
||
SessionId string `json:"sessionId" dc:"会话ID"`
|
||
}
|
||
|
||
type CreateFlowExecutionReq struct {
|
||
FlowUserId int64 `json:"flowUserId" description:"流程ID"`
|
||
FlowName string `json:"flowName"`
|
||
NodeGroupId string `json:"nodeGroupId"`
|
||
TriggerType flow.FlowExecutionTriggerType `json:"triggerType" description:"触发类型"`
|
||
DurationMs int64 `json:"durationMs" description:"执行时长（毫秒）"`
|
||
Status flow.FlowExecutionStatus `json:"status" description:"状态:1-运行中,2-成功,3-失败"`
|
||
FlowContent *entity.FlowInfo `json:"flowContent" description:"流程内容"`
|
||
NodeInputParams []*entity.FlowNode `json:"nodeInputParams" description:"节点输入参数"`
|
||
OutputParams []map[string]interface{} `json:"outputParams" description:"输出参数"`
|
||
ErrorMessage string `json:"errorMessage" description:"错误信息"`
|
||
TraceId string `json:"traceId" description:"跟踪ID"`
|
||
SessionId string `json:"sessionId" dc:"会话ID"`
|
||
}
|
||
|
||
type CreateFlowExecutionRes struct {
|
||
Id int64 `json:"id,string"`
|
||
}
|
||
|
||
type UpdateFlowExecutionReq struct {
|
||
Id int64 `json:"id" v:"required#ID不能为空"`
|
||
NodeGroupId string `json:"nodeGroupId"`
|
||
DurationMs int64 `json:"durationMs" description:"执行时长（毫秒）"`
|
||
Status flow.FlowExecutionStatus `json:"status" description:"状态:1-运行中,2-成功,3-失败"`
|
||
OutputParams []map[string]interface{} `json:"outputParams" description:"输出参数"`
|
||
ErrorMessage string `json:"errorMessage" description:"错误信息"`
|
||
TraceId string `json:"traceId" description:"跟踪ID"`
|
||
}
|
||
|
||
type GetFlowExecutionReq struct {
|
||
g.Meta `path:"/get" method:"get" tags:"任务管理" summary:"获取任务详情" dc:"获取任务详情"`
|
||
|
||
Id int64 `json:"id" v:"required#ID不能为空"`
|
||
SessionId string `json:"sessionId" dc:"会话ID"`
|
||
}
|
||
|
||
type ListFlowExecutionReq struct {
|
||
g.Meta `path:"/list" method:"get" tags:"任务管理" summary:"任务列表" dc:"任务列表"`
|
||
|
||
Page *beans.Page `json:"page"`
|
||
Creator string `json:"creator"`
|
||
}
|
||
|
||
type ListFlowExecutionRes struct {
|
||
List []*VOFlowExecution `json:"list"`
|
||
Total int `json:"total"`
|
||
}
|
||
|
||
type VOFlowExecution struct {
|
||
Id int64 `json:"id,string" dc:"id"`
|
||
FlowUserId int64 `json:"flowUserId,string" description:"流程ID"`
|
||
FlowName string `json:"flowName"`
|
||
TriggerType flow.FlowExecutionTriggerType `json:"triggerType" description:"触发类型"`
|
||
DurationMs int64 `json:"durationMs" description:"执行时长（毫秒）"`
|
||
Status flow.FlowExecutionStatus `json:"status" description:"状态:1-运行中,2-成功,3-失败"`
|
||
FlowContent *entity.FlowInfo `json:"flowContent" description:"流程内容"`
|
||
NodeInputParams []*entity.FlowNode `json:"nodeInputParams" description:"节点输入参数"`
|
||
OutputParams []map[string]interface{} `json:"outputParams" description:"输出参数"`
|
||
ErrorMessage string `json:"errorMessage" description:"错误信息"`
|
||
TraceId string `json:"traceId" description:"跟踪ID"`
|
||
SessionId string `json:"sessionId" dc:"会话ID"`
|
||
CreatedAt *gtime.Time `json:"createdAt" dc:"创建时间"`
|
||
UpdatedAt *gtime.Time `json:"updatedAt" dc:"更新时间"`
|
||
ImgAddressPrefix string `json:"imgAddressPrefix"`
|
||
}
|
||
|
||
// ========== 核心：构建树状结构 ==========
|
||
// 定义树结构
|
||
type OutputItem struct {
|
||
Timestamp string `json:"timestamp" description:"时间戳key"`
|
||
Content string `json:"content" description:"内容值"`
|
||
Type string `json:"type" description:"类型"`
|
||
Label string `json:"label" description:"后缀+数字标号"`
|
||
}
|
||
type FlowNode struct {
|
||
FlowName string `json:"flowName" description:"流程名称"`
|
||
Id int64 `json:"Id,string" description:"任务ID"`
|
||
SessionId string `json:"sessionId" description:"会话ID"`
|
||
Items []OutputItem `json:"items" description:"输出项列表"`
|
||
}
|
||
type DateNode struct {
|
||
CreateDate string `json:"createDate" description:"创建日期"`
|
||
Flows []FlowNode `json:"flows" description:"流程列表"`
|
||
}
|
||
|
||
type ListFlowExecutionTreeRes struct {
|
||
Tree []DateNode `json:"tree"`
|
||
ImgAddressPrefix string `json:"imgAddressPrefix"`
|
||
}
|