59 lines
2.6 KiB
Go
59 lines
2.6 KiB
Go
|
package entity
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"ai-agent/workflow/consts/node"
|
|||
|
|
|
|||
|
|
"gitea.com/red-future/common/beans"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// NodeExecution 节点执行记录
|
|||
|
|
// 记录每个节点的入参、出参、token消耗、执行状态等信息
|
|||
|
|
type NodeExecution struct {
|
|||
|
|
beans.SQLBaseDO `orm:",inherit"` // 嵌入基础字段：Id, TenantId, Creator, CreatedAt, Updater, UpdatedAt, DeletedAt
|
|||
|
|
|
|||
|
|
FlowExecutionId int64 `orm:"flow_execution_id" json:"flowExecutionId" description:"流程执行ID"`
|
|||
|
|
NodeId string `orm:"node_id" json:"nodeId" description:"节点ID"`
|
|||
|
|
NodeName string `orm:"node_name" json:"nodeName" description:"节点名称"`
|
|||
|
|
NodeGroupId string `orm:"node_group_id" json:"nodeGroupId" description:"节点组ID"`
|
|||
|
|
InputParams map[string]interface{} `orm:"input_params" json:"inputParams" description:"节点输入参数"`
|
|||
|
|
OutputParams map[string]interface{} `orm:"output_params" json:"outputParams" description:"节点输出参数"`
|
|||
|
|
PromptTokens int `orm:"prompt_tokens" json:"promptTokens" description:"提示词token消耗"`
|
|||
|
|
CompletionTokens int `orm:"completion_tokens" json:"completionTokens" description:"补全token消耗"`
|
|||
|
|
TotalTokens int `orm:"total_tokens" json:"totalTokens" description:"总token消耗"`
|
|||
|
|
Status node.NodeExecutionStatus `orm:"status" json:"status" description:"执行状态:1-运行中,2-成功,3-失败,4-暂停,5-等待执行"`
|
|||
|
|
DurationMs int64 `orm:"duration_ms" json:"durationMs" description:"执行时长（毫秒）"`
|
|||
|
|
ErrorMessage string `orm:"error_message" json:"errorMessage" description:"错误信息"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type nodeExecutionCol struct {
|
|||
|
|
beans.SQLBaseCol
|
|||
|
|
FlowExecutionId string
|
|||
|
|
NodeId string
|
|||
|
|
NodeName string
|
|||
|
|
NodeGroupId string
|
|||
|
|
InputParams string
|
|||
|
|
OutputParams string
|
|||
|
|
PromptTokens string
|
|||
|
|
CompletionTokens string
|
|||
|
|
TotalTokens string
|
|||
|
|
Status string
|
|||
|
|
DurationMs string
|
|||
|
|
ErrorMessage string
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var NodeExecutionCol = nodeExecutionCol{
|
|||
|
|
SQLBaseCol: beans.DefSQLBaseCol,
|
|||
|
|
FlowExecutionId: "flow_execution_id",
|
|||
|
|
NodeId: "node_id",
|
|||
|
|
NodeName: "node_name",
|
|||
|
|
NodeGroupId: "node_group_id",
|
|||
|
|
InputParams: "input_params",
|
|||
|
|
OutputParams: "output_params",
|
|||
|
|
PromptTokens: "prompt_tokens",
|
|||
|
|
CompletionTokens: "completion_tokens",
|
|||
|
|
TotalTokens: "total_tokens",
|
|||
|
|
Status: "status",
|
|||
|
|
DurationMs: "duration_ms",
|
|||
|
|
ErrorMessage: "error_message",
|
|||
|
|
}
|