52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
"""目录生成相关节点
|
|
|
|
包含目录生成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"
|
|
] |