sladro
c1292fcacc
Co-authored-by: factory-droid[bot] <138933559+factory-droid[bot]@users.noreply.github.com>
97 lines
3.4 KiB
Python
97 lines
3.4 KiB
Python
"""最终确定章节结构的节点
|
||
|
||
基于AI审查结果最终确定章节结构，并确保包含必要的标准章节。
|
||
"""
|
||
|
||
import logging
|
||
from typing import Dict, List, Any
|
||
|
||
from ..base import BaseNode, NodeContext
|
||
from ...tools.parser import DocumentChapter
|
||
from .factories import ChapterFactory
|
||
from .base_mixins import TocNodeBase
|
||
from .category_manager import CategoryManager
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class FinalizeChaptersNode(BaseNode, TocNodeBase):
|
||
"""最终确定章节结构的节点"""
|
||
|
||
@property
|
||
def name(self) -> str:
|
||
return "finalize_chapters"
|
||
|
||
@property
|
||
def description(self) -> str:
|
||
return "最终确定章节结构"
|
||
|
||
def execute(self, state: Dict[str, Any], context: NodeContext) -> Dict[str, Any]:
|
||
"""执行章节最终确定"""
|
||
return self.safe_execute(self._do_finalize_chapters, state, "最终确定章节结构")
|
||
|
||
def _do_finalize_chapters(self, state: Dict[str, Any]) -> Dict[str, Any]:
|
||
"""执行实际的章节最终确定逻辑"""
|
||
# 验证必需字段
|
||
if not self.validate_required_fields(state, ["preliminary_chapters"]):
|
||
raise ValueError("缺少初步章节数据")
|
||
|
||
# 获取调整后的章节（如果有）或原始章节
|
||
adjusted_chapters = state.get("adjusted_chapters", state["preliminary_chapters"])
|
||
|
||
# 应用最终确定逻辑（主要是确保标准章节存在）
|
||
final_chapters = adjusted_chapters
|
||
|
||
# 确保标准章节存在
|
||
final_chapters = self._ensure_standard_chapters(final_chapters)
|
||
|
||
self._apply_saved_guidance(final_chapters, state.get("chapter_guidance_map"))
|
||
|
||
self.log_step_info("finalize_chapters", f"最终生成{len(final_chapters)}个章节")
|
||
|
||
return self._update_state(state,
|
||
final_chapters=final_chapters,
|
||
should_continue=True) # 让后续user_feedback节点决定是否继续
|
||
|
||
|
||
def _ensure_standard_chapters(self, chapters: List[DocumentChapter]) -> List[DocumentChapter]:
|
||
"""确保包含标准章节
|
||
|
||
Args:
|
||
chapters: 当前章节列表
|
||
|
||
Returns:
|
||
包含标准章节的完整列表
|
||
"""
|
||
final_chapters = chapters.copy()
|
||
|
||
# 确保评标索引表存在（作为第一章）
|
||
has_index = any("评标索引" in ch.title for ch in final_chapters)
|
||
if not has_index:
|
||
index_chapter = ChapterFactory.create_standard_chapter(
|
||
"evaluation_index",
|
||
"评标索引表（技术评分完全对应）",
|
||
1
|
||
)
|
||
final_chapters.insert(0, index_chapter)
|
||
logger.info("添加评标索引表章节")
|
||
|
||
# 可以在此添加其他必要的标准章节
|
||
# 例如：封面、目录、声明等
|
||
|
||
return final_chapters
|
||
|
||
def _apply_saved_guidance(self,
|
||
chapters: List[DocumentChapter],
|
||
guidance_map: Dict[str, Any] | None) -> None:
|
||
if not guidance_map:
|
||
return
|
||
|
||
for chapter in chapters:
|
||
category_key = CategoryManager.infer_category_for_chapter(chapter)
|
||
guidance = guidance_map.get(category_key)
|
||
if guidance:
|
||
chapter.guidance = guidance
|
||
|
||
if chapter.children:
|
||
self._apply_saved_guidance(chapter.children, guidance_map) |