"""目录生成相关节点

包含目录生成Agent的所有节点实现和相关组件。
"""

# 核心节点
from .group_criteria import GroupCriteriaNode
from .generate_first_level import GenerateFirstLevelNode
from .generate_sub_chapters import GenerateSubChaptersNode
from .review_structure import ReviewStructureNode
from .apply_suggestions import ApplyReviewSuggestionsNode
from .adjust_chapters import AdjustChaptersNode
from .finalize_chapters import FinalizeChaptersNode
from .user_feedback import UserFeedbackNode
from .optimize_with_feedback import OptimizeWithFeedbackNode
from .prepare_tech_consistency_prompt import PrepareTechConsistencyPromptNode

# 辅助组件
from .factories import ChapterFactory
from .category_manager import CategoryManager
from .llm_helper import LLMHelper
from .base_mixins import (
    NodeStateManager,
    ErrorHandlerMixin,
    WorkflowUtilsMixin,
    TocNodeBase
)

__all__ = [
    # 核心节点
    "GroupCriteriaNode",
    "GenerateFirstLevelNode",
    "GenerateSubChaptersNode",
    "ReviewStructureNode",
    "ApplyReviewSuggestionsNode",
    "AdjustChaptersNode",
    "FinalizeChaptersNode",
    "UserFeedbackNode",
    "OptimizeWithFeedbackNode",
    "PrepareTechConsistencyPromptNode",

    # 辅助组件
    "ChapterFactory",
    "CategoryManager",
    "LLMHelper",

    # 基础类和混入
    "NodeStateManager",
    "ErrorHandlerMixin",
    "WorkflowUtilsMixin",
    "TocNodeBase"
]