parser

type BaseNodeParser struct {
	// contains filtered or unexported fields
}

type ClassifierNodeParser struct {
	*BaseNodeParser
}

type CodeNodeParser struct {
	*BaseNodeParser
}

type ConditionNodeParser struct {
	*BaseNodeParser
}

type DifyApp struct {
	Name                string `yaml:"name" json:"name"`
	Description         string `yaml:"description" json:"description"`
	Icon                string `yaml:"icon" json:"icon"`
	IconBackground      string `yaml:"icon_background" json:"icon_background"`
	Mode                string `yaml:"mode" json:"mode"`
	UseIconAsAnswerIcon bool   `yaml:"use_icon_as_answer_icon" json:"use_icon_as_answer_icon"`
}

type DifyCase struct {
	CaseID          string          `yaml:"case_id" json:"case_id"`
	ID              string          `yaml:"id" json:"id"`
	LogicalOperator string          `yaml:"logical_operator" json:"logical_operator"`
	Conditions      []DifyCondition `yaml:"conditions" json:"conditions"`
}

type DifyClass struct {
	ID   string `yaml:"id" json:"id"`
	Name string `yaml:"name" json:"name"`
}

type DifyCondition struct {
	ID                 string   `yaml:"id" json:"id"`
	VariableSelector   []string `yaml:"variable_selector" json:"variable_selector"`
	ComparisonOperator string   `yaml:"comparison_operator" json:"comparison_operator"`
	Value              string   `yaml:"value" json:"value"`
	VarType            string   `yaml:"varType" json:"varType"`
}

type DifyContext struct {
	Enabled          bool     `yaml:"enabled" json:"enabled"`
	VariableSelector []string `yaml:"variable_selector,omitempty" json:"variable_selector,omitempty"`
}

type DifyDSL struct {
	App      DifyApp      `yaml:"app" json:"app"`
	Kind     string       `yaml:"kind" json:"kind"`
	Version  string       `yaml:"version" json:"version"`
	Workflow DifyWorkflow `yaml:"workflow" json:"workflow"`
}

type DifyEdge struct {
	ID           string        `yaml:"id" json:"id"`
	Source       string        `yaml:"source" json:"source"`
	Target       string        `yaml:"target" json:"target"`
	SourceHandle string        `yaml:"sourceHandle,omitempty" json:"sourceHandle,omitempty"`
	TargetHandle string        `yaml:"targetHandle,omitempty" json:"targetHandle,omitempty"`
	Type         string        `yaml:"type" json:"type"`
	Data         *DifyEdgeData `yaml:"data,omitempty" json:"data,omitempty"`
	Selected     bool          `yaml:"selected,omitempty" json:"selected,omitempty"`
	ZIndex       int           `yaml:"zIndex,omitempty" json:"zIndex,omitempty"`
}

type DifyEdgeData struct {
	IsInLoop      bool   `yaml:"isInLoop,omitempty" json:"isInLoop,omitempty"`
	IsInIteration bool   `yaml:"isInIteration,omitempty" json:"isInIteration,omitempty"`
	IterationID   string `yaml:"iteration_id,omitempty" json:"iteration_id,omitempty"`
	SourceType    string `yaml:"sourceType,omitempty" json:"sourceType,omitempty"`
	TargetType    string `yaml:"targetType,omitempty" json:"targetType,omitempty"`
}

type DifyFeatureConfig struct {
	Enabled bool `yaml:"enabled" json:"enabled"`
}

type DifyFeatures struct {
	FileUpload                    *DifyFileUploadConfig   `yaml:"file_upload,omitempty" json:"file_upload,omitempty"`
	OpeningStatement              string                  `yaml:"opening_statement,omitempty" json:"opening_statement,omitempty"`
	RetrieverResource             *DifyRetrieverConfig    `yaml:"retriever_resource,omitempty" json:"retriever_resource,omitempty"`
	SensitiveWordAvoidance        *DifyFeatureConfig      `yaml:"sensitive_word_avoidance,omitempty" json:"sensitive_word_avoidance,omitempty"`
	SpeechToText                  *DifyFeatureConfig      `yaml:"speech_to_text,omitempty" json:"speech_to_text,omitempty"`
	SuggestedQuestions            []string                `yaml:"suggested_questions,omitempty" json:"suggested_questions,omitempty"`
	SuggestedQuestionsAfterAnswer *DifyFeatureConfig      `yaml:"suggested_questions_after_answer,omitempty" json:"suggested_questions_after_answer,omitempty"`
	TextToSpeech                  *DifyTextToSpeechConfig `yaml:"text_to_speech,omitempty" json:"text_to_speech,omitempty"`
}

type DifyFileUploadConfig struct {
	Enabled                  bool              `yaml:"enabled" json:"enabled"`
	AllowedFileExtensions    []string          `yaml:"allowed_file_extensions,omitempty" json:"allowed_file_extensions,omitempty"`
	AllowedFileTypes         []string          `yaml:"allowed_file_types,omitempty" json:"allowed_file_types,omitempty"`
	AllowedFileUploadMethods []string          `yaml:"allowed_file_upload_methods,omitempty" json:"allowed_file_upload_methods,omitempty"`
	FileUploadConfig         *DifyUploadLimits `yaml:"fileUploadConfig,omitempty" json:"fileUploadConfig,omitempty"`
	Image                    *DifyImageConfig  `yaml:"image,omitempty" json:"image,omitempty"`
	NumberLimits             int               `yaml:"number_limits,omitempty" json:"number_limits,omitempty"`
}

type DifyGraph struct {
	Edges    []DifyEdge    `yaml:"edges" json:"edges"`
	Nodes    []DifyNode    `yaml:"nodes" json:"nodes"`
	Viewport *DifyViewport `yaml:"viewport,omitempty" json:"viewport,omitempty"`
}

type DifyImageConfig struct {
	Enabled         bool     `yaml:"enabled" json:"enabled"`
	NumberLimits    int      `yaml:"number_limits" json:"number_limits"`
	TransferMethods []string `yaml:"transfer_methods" json:"transfer_methods"`
}

type DifyModel struct {
	Provider         string                 `yaml:"provider" json:"provider"`
	Name             string                 `yaml:"name" json:"name"`
	Mode             string                 `yaml:"mode" json:"mode"`
	CompletionParams map[string]interface{} `yaml:"completion_params,omitempty" json:"completion_params,omitempty"`
}

type DifyNode struct {
	ID               string        `yaml:"id" json:"id"`
	Type             string        `yaml:"type" json:"type"`
	Position         DifyPosition  `yaml:"position" json:"position"`
	PositionAbsolute *DifyPosition `yaml:"positionAbsolute,omitempty" json:"positionAbsolute,omitempty"`
	Width            float64       `yaml:"width,omitempty" json:"width,omitempty"`
	Height           float64       `yaml:"height,omitempty" json:"height,omitempty"`
	Selected         bool          `yaml:"selected,omitempty" json:"selected,omitempty"`
	Draggable        bool          `yaml:"draggable,omitempty" json:"draggable,omitempty"`
	Selectable       bool          `yaml:"selectable,omitempty" json:"selectable,omitempty"`
	SourcePosition   string        `yaml:"sourcePosition,omitempty" json:"sourcePosition,omitempty"`
	TargetPosition   string        `yaml:"targetPosition,omitempty" json:"targetPosition,omitempty"`
	ZIndex           int           `yaml:"zIndex,omitempty" json:"zIndex,omitempty"`
	ParentID         string        `yaml:"parentId,omitempty" json:"parentId,omitempty"`
	Extent           string        `yaml:"extent,omitempty" json:"extent,omitempty"`
	Data             DifyNodeData  `yaml:"data" json:"data"`
}

type DifyNodeData struct {
	Type      string         `yaml:"type" json:"type"`
	Title     string         `yaml:"title" json:"title"`
	Desc      string         `yaml:"desc,omitempty" json:"desc,omitempty"`
	Selected  bool           `yaml:"selected,omitempty" json:"selected,omitempty"`
	Variables []DifyVariable `yaml:"variables,omitempty" json:"variables,omitempty"`
	Outputs   interface{}    `yaml:"outputs,omitempty" json:"outputs,omitempty"` // Supports []DifyOutput and map[string]interface{}

	// LLM node specific fields
	Model          *DifyModel   `yaml:"model,omitempty" json:"model,omitempty"`
	PromptTemplate []DifyPrompt `yaml:"prompt_template,omitempty" json:"prompt_template,omitempty"`
	Context        *DifyContext `yaml:"context,omitempty" json:"context,omitempty"`
	Vision         *DifyVision  `yaml:"vision,omitempty" json:"vision,omitempty"`

	// Code node specific fields
	Code         string `yaml:"code,omitempty" json:"code,omitempty"`
	CodeLanguage string `yaml:"code_language,omitempty" json:"code_language,omitempty"`

	// Condition node specific fields
	Cases []DifyCase `yaml:"cases,omitempty" json:"cases,omitempty"`

	// Classifier node specific fields
	Classes               []DifyClass `yaml:"classes,omitempty" json:"classes,omitempty"`
	Instruction           string      `yaml:"instruction,omitempty" json:"instruction,omitempty"`
	Instructions          string      `yaml:"instructions,omitempty" json:"instructions,omitempty"`
	QueryVariableSelector []string    `yaml:"query_variable_selector,omitempty" json:"query_variable_selector,omitempty"`
	Topics                []string    `yaml:"topics,omitempty" json:"topics,omitempty"`

	// Iteration node specific fields
	ErrorHandleMode   string   `yaml:"error_handle_mode,omitempty" json:"error_handle_mode,omitempty"`
	IsParallel        bool     `yaml:"is_parallel,omitempty" json:"is_parallel,omitempty"`
	IteratorInputType string   `yaml:"iterator_input_type,omitempty" json:"iterator_input_type,omitempty"`
	IteratorSelector  []string `yaml:"iterator_selector,omitempty" json:"iterator_selector,omitempty"`
	OutputSelector    []string `yaml:"output_selector,omitempty" json:"output_selector,omitempty"`
	OutputType        string   `yaml:"output_type,omitempty" json:"output_type,omitempty"`
	ParallelNums      int      `yaml:"parallel_nums,omitempty" json:"parallel_nums,omitempty"`
	StartNodeID       string   `yaml:"start_node_id,omitempty" json:"start_node_id,omitempty"`

	// Iteration-related identifiers
	IsInIteration bool   `yaml:"isInIteration,omitempty" json:"isInIteration,omitempty"`
	IsInLoop      bool   `yaml:"isInLoop,omitempty" json:"isInLoop,omitempty"`
	IterationID   string `yaml:"iteration_id,omitempty" json:"iteration_id,omitempty"`
}

type DifyOutput struct {
	ValueSelector []string `yaml:"value_selector" json:"value_selector"`
	ValueType     string   `yaml:"value_type" json:"value_type"`
	Variable      string   `yaml:"variable" json:"variable"`
}

type DifyParser struct {
	*common.BaseParser
	// contains filtered or unexported fields
}

type DifyPosition struct {
	X float64 `yaml:"x" json:"x"`
	Y float64 `yaml:"y" json:"y"`
}

type DifyPrompt struct {
	ID   string `yaml:"id" json:"id"`
	Role string `yaml:"role" json:"role"`
	Text string `yaml:"text" json:"text"`
}

type DifyRetrieverConfig struct {
	Enabled bool `yaml:"enabled" json:"enabled"`
}

type DifyTextToSpeechConfig struct {
	Enabled  bool   `yaml:"enabled" json:"enabled"`
	Language string `yaml:"language,omitempty" json:"language,omitempty"`
	Voice    string `yaml:"voice,omitempty" json:"voice,omitempty"`
}

type DifyUploadLimits struct {
	AudioFileSizeLimit      int `yaml:"audio_file_size_limit" json:"audio_file_size_limit"`
	BatchCountLimit         int `yaml:"batch_count_limit" json:"batch_count_limit"`
	FileSizeLimit           int `yaml:"file_size_limit" json:"file_size_limit"`
	ImageFileSizeLimit      int `yaml:"image_file_size_limit" json:"image_file_size_limit"`
	VideoFileSizeLimit      int `yaml:"video_file_size_limit" json:"video_file_size_limit"`
	WorkflowFileUploadLimit int `yaml:"workflow_file_upload_limit" json:"workflow_file_upload_limit"`
}

type DifyVariable struct {
	Label         string   `yaml:"label" json:"label"`
	Variable      string   `yaml:"variable" json:"variable"`
	Type          string   `yaml:"type" json:"type"`
	ValueType     string   `yaml:"value_type,omitempty" json:"value_type,omitempty"`
	ValueSelector []string `yaml:"value_selector,omitempty" json:"value_selector,omitempty"`
	Required      bool     `yaml:"required" json:"required"`
	MaxLength     int      `yaml:"max_length,omitempty" json:"max_length,omitempty"`
	Options       []string `yaml:"options,omitempty" json:"options,omitempty"`
}

type DifyViewport struct {
	X    float64 `yaml:"x" json:"x"`
	Y    float64 `yaml:"y" json:"y"`
	Zoom float64 `yaml:"zoom" json:"zoom"`
}

type DifyVision struct {
	Enabled bool `yaml:"enabled" json:"enabled"`
}

type DifyWorkflow struct {
	ConversationVariables []interface{} `yaml:"conversation_variables" json:"conversation_variables"`
	EnvironmentVariables  []interface{} `yaml:"environment_variables" json:"environment_variables"`
	Features              DifyFeatures  `yaml:"features" json:"features"`
	Graph                 DifyGraph     `yaml:"graph" json:"graph"`
}

type EndNodeParser struct {
	*BaseNodeParser
	// contains filtered or unexported fields
}

type IterationNodeParser struct {
	*BaseNodeParser
}

type IterationStartNodeParser struct {
	*BaseNodeParser
}

type LLMNodeParser struct {
	*BaseNodeParser
}

type NodeParser interface {
	// GetSupportedType returns the supported node type.
	GetSupportedType() string

	// ParseNode parses a node.
	ParseNode(difyNode DifyNode) (*models.Node, error)

	// ValidateNode validates node data.
	ValidateNode(difyNode DifyNode) error
}

type NodeParserFactory interface {
	// CreateParser creates a parser.
	CreateParser(nodeType string, variableRefSystem *models.VariableReferenceSystem) (NodeParser, error)

	// GetSupportedTypes returns supported node types.
	GetSupportedTypes() []string
}

type ParserFactory struct {
	// contains filtered or unexported fields
}

type StartNodeParser struct {
	*BaseNodeParser
}