19904408334
e548f20b6e
在流程执行和节点执行中新增 TotalFee 和 TokenInfo 字段，用于记录每次模型调用的详细计费数据；流程执行完成时汇总所有节点的 token 消耗和费用；重构 token 更新逻辑，从响应字段提取改为使用回调返回的 billingData；优化 URL 后缀提取方法以正确处理带查询参数的链接。
77 lines
3.4 KiB
Go
77 lines
3.4 KiB
Go
package node
|
|
|
|
import (
|
|
"ai-agent/workflow/consts/node"
|
|
flowDto "ai-agent/workflow/model/dto/flow"
|
|
"ai-agent/workflow/model/entity"
|
|
|
|
"gitea.redpowerfuture.com/red-future/common/beans"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
)
|
|
|
|
// CreateNodeExecutionReq 创建节点执行记录请求
|
|
type CreateNodeExecutionReq struct {
|
|
g.Meta `path:"/create" method:"post" tags:"节点执行记录" summary:"创建节点执行记录" dc:"创建节点执行记录"`
|
|
FlowExecutionId int64 `json:"flowExecutionId" v:"required#流程执行ID不能为空"`
|
|
NodeId string `json:"nodeId" v:"required#节点ID不能为空"`
|
|
NodeName string `json:"nodeName"`
|
|
NodeGroupId string `json:"nodeGroupId"`
|
|
Status node.NodeExecutionStatus `json:"status"`
|
|
InputParams *flowDto.NodeExecutionInput `json:"inputParams"`
|
|
InputParamsPath string
|
|
OutputParams *flowDto.NodeExecutionInput `json:"outputParams"`
|
|
OutputParamsPath string
|
|
}
|
|
|
|
type CreateNodeExecutionRes struct {
|
|
Id int64 `json:"id,string"`
|
|
}
|
|
|
|
// UpdateNodeExecutionReq 更新节点执行记录请求
|
|
type UpdateNodeExecutionReq struct {
|
|
g.Meta `path:"/update" method:"put" tags:"节点执行记录" summary:"更新节点执行记录" dc:"更新节点执行记录状态和结果"`
|
|
Id int64 `json:"id" v:"required#ID不能为空"`
|
|
InputParams *flowDto.NodeExecutionInput `json:"inputParams"`
|
|
InputParamsPath string
|
|
OutputParams *flowDto.NodeExecutionInput `json:"outputParams"`
|
|
OutputParamsPath string
|
|
PromptTokens int `json:"promptTokens"`
|
|
CompletionTokens int `json:"completionTokens"`
|
|
TotalTokens int `json:"totalTokens"`
|
|
TokenInfo []map[string]any `json:"tokenInfo"`
|
|
Status node.NodeExecutionStatus `json:"status"`
|
|
DurationMs int64 `json:"durationMs"`
|
|
ErrorMessage string `json:"errorMessage"`
|
|
}
|
|
|
|
// DeleteNodeExecutionReq 删除节点执行记录请求
|
|
type DeleteNodeExecutionReq struct {
|
|
g.Meta `path:"/delete" method:"delete" tags:"节点执行记录" summary:"删除节点执行记录" dc:"删除节点执行记录"`
|
|
Id int64 `json:"id" v:"required#ID不能为空"`
|
|
}
|
|
|
|
// GetNodeExecutionReq 根据ID查询节点执行记录请求
|
|
type GetNodeExecutionReq struct {
|
|
g.Meta `path:"/get" method:"get" tags:"节点执行记录" summary:"查询节点执行记录详情" dc:"根据ID查询节点执行记录详情"`
|
|
Id int64 `json:"id" v:"required#ID不能为空"`
|
|
}
|
|
|
|
// ListNodeExecutionByFlowReq 查询流程下所有节点执行记录请求
|
|
type ListNodeExecutionByFlowReq struct {
|
|
g.Meta `path:"/listByFlow" method:"get" tags:"节点执行记录" summary:"查询流程节点执行列表" dc:"查询指定流程执行下的所有节点执行记录"`
|
|
Page *beans.Page `json:"page"`
|
|
FlowExecutionId int64 `json:"flowExecutionId" v:"required#流程执行ID不能为空"`
|
|
NodeGroupId string `json:"nodeGroupId"`
|
|
}
|
|
|
|
// NodeExecutionResp 节点执行记录响应
|
|
type NodeExecutionResp struct {
|
|
*entity.NodeExecution
|
|
}
|
|
|
|
// ListNodeExecutionResp 节点执行记录列表响应
|
|
type ListNodeExecutionResp struct {
|
|
List []*entity.NodeExecution `json:"list"`
|
|
Total int `json:"total"`
|
|
}
|