package entity

import (
	"ai-agent/workflow/consts/node"

	"gitea.redpowerfuture.com/red-future/common/beans"
)

type NodePrompt struct {
	beans.SQLBaseDO `orm:",inherit"` // 嵌入基础字段：Id, TenantId, Creator, CreatedAt, Updater, UpdatedAt, DeletedAt

	NodeType   node.NodeType   `orm:"node_type" json:"nodeType"`
	Prompt     string          `orm:"prompt" json:"prompt"`
	SourceType node.SourceType `orm:"source_type" json:"sourceType"`
}

type nodePromptCol struct {
	beans.SQLBaseCol
	NodeType   string
	Prompt     string
	SourceType string
}

var NodePromptCol = nodePromptCol{
	SQLBaseCol: beans.DefSQLBaseCol,
	NodeType:   "node_type",
	Prompt:     "prompt",
	SourceType: "source_type",
}