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",
}