376 lines
16 KiB
Python
376 lines
16 KiB
Python
"""AI 写作 LangGraph Graph，含 4 个用户干预点。"""
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
|
||
from langchain_core.runnables import RunnableConfig
|
||
from langgraph.graph import END, StateGraph
|
||
from langgraph.types import interrupt
|
||
|
||
from .state import AIWritingState
|
||
from .nodes.intent_parser import intent_parser_node
|
||
from .nodes.researcher import researcher_node
|
||
from .nodes.writer_outline import writer_outline_node
|
||
from .nodes.writer_draft import writer_draft_node
|
||
from .nodes.editor import editor_node
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
async def _mark_session_done(config: RunnableConfig | None) -> None:
|
||
"""把业务表行 status 写成 ``done``。
|
||
|
||
pause_draft_confirm action=finalize / pause_review_confirm force_finalize 时调，
|
||
让 LangGraph Server 路径下新会话有正确终态（老路径靠 ``_stream_graph_segment``
|
||
finally 块写终态，新路径完全不走它）。失败不影响 graph 流程。
|
||
"""
|
||
if not config:
|
||
return
|
||
thread_id = (config.get("configurable") or {}).get("thread_id")
|
||
if not thread_id:
|
||
return
|
||
from deerflow.persistence.ai_writing_sessions import get_default_store
|
||
from deerflow.runtime.user_context import AUTO
|
||
|
||
repo = get_default_store()
|
||
if repo is None:
|
||
return
|
||
try:
|
||
await repo.update(thread_id, user_id=AUTO, status="done")
|
||
except Exception:
|
||
logger.exception("ai_writing: mark_session_done failed (non-fatal) thread_id=%s", thread_id)
|
||
|
||
# ── 干预节点 ──────────────────────────────────────────────────────────────────
|
||
|
||
async def pause_material_confirm(state: AIWritingState) -> dict:
|
||
"""PAUSE-1：素材确认"""
|
||
pkg = state["material_package"]
|
||
user_input: dict = interrupt({
|
||
"pause_point": "material_confirm",
|
||
"message": "素材收集专家已完成素材搜集，请确认素材后继续。",
|
||
"materials": pkg["materials"],
|
||
"keywords": pkg["keywords"],
|
||
"summary": pkg["summary"],
|
||
})
|
||
|
||
action = user_input.get("action", "confirm")
|
||
|
||
if action == "confirm":
|
||
approved_ids = set(user_input.get("approvedMaterialIds") or [m["id"] for m in pkg["materials"]])
|
||
filtered = [m for m in pkg["materials"] if m["id"] in approved_ids]
|
||
updated_pkg = {**pkg, "materials": filtered}
|
||
return {
|
||
"material_package": updated_pkg,
|
||
"last_user_intervention": user_input,
|
||
"status": "writing_outline",
|
||
"progress_events": [{"type": "resumed", "agent_name": "系统", "message": "用户已确认素材，开始规划大纲"}],
|
||
}
|
||
|
||
# re_search：重新搜索
|
||
extra_kws = user_input.get("extraKeywords") or []
|
||
return {
|
||
"last_user_intervention": {**user_input, "_extra_keywords": extra_kws},
|
||
"status": "researching",
|
||
"progress_events": [{"type": "resumed", "agent_name": "系统", "message": "重新搜索中..."}],
|
||
}
|
||
|
||
|
||
def _normalize_outline(outline: dict) -> dict:
|
||
"""将前端回传的 camelCase 大纲统一为后端 snake_case 格式。"""
|
||
if not outline:
|
||
return outline
|
||
sections = []
|
||
for s in outline.get("sections", []):
|
||
sections.append({
|
||
"section_title": s.get("section_title") or s.get("sectionTitle", ""),
|
||
"key_points": s.get("key_points") or s.get("keyPoints", []),
|
||
"material_ids": s.get("material_ids") or s.get("materialIds", []),
|
||
})
|
||
return {"title": outline.get("title", ""), "sections": sections}
|
||
|
||
|
||
def _check_outline_material_support(outline: dict) -> str:
|
||
"""轻量启发式：统计未关联任何素材的章节，返回预警文案（空串=无需预警）。
|
||
|
||
仅做提示、不拦截流程；真正的素材不足会在写作节点被检出并触发求助暂停。
|
||
"""
|
||
sections = (outline or {}).get("sections", [])
|
||
weak = [
|
||
(s.get("section_title") or s.get("sectionTitle") or "")
|
||
for s in sections
|
||
if not (s.get("material_ids") or s.get("materialIds"))
|
||
]
|
||
weak = [t for t in weak if t]
|
||
if not weak:
|
||
return ""
|
||
return (
|
||
f"有 {len(weak)} 个章节未关联检索素材（{('、'.join(weak))}）。"
|
||
"严格依据素材模式下，这些章节在写作时可能因素材不足而暂停向你求助。"
|
||
)
|
||
|
||
|
||
async def pause_outline_confirm(state: AIWritingState) -> dict:
|
||
"""PAUSE-2：大纲确认"""
|
||
user_input: dict = interrupt({
|
||
"pause_point": "outline_confirm",
|
||
"message": "作家已完成大纲规划，请确认或编辑大纲后继续。",
|
||
"outline": state["current_outline"],
|
||
})
|
||
|
||
action = user_input.get("action", "confirm")
|
||
|
||
if action == "confirm":
|
||
raw_outline = user_input.get("editedOutline") or state["current_outline"]
|
||
edited = _normalize_outline(raw_outline) if raw_outline else state["current_outline"]
|
||
events: list[dict] = [{"type": "resumed", "agent_name": "系统", "message": "用户已确认大纲，开始写作"}]
|
||
# 阶段四：严格模式下对大纲做轻量素材充分性预警（仅提示，不拦截流程）
|
||
if state.get("writing_mode") == "strict":
|
||
warning = _check_outline_material_support(edited)
|
||
if warning:
|
||
events.append({
|
||
"type": "notice", "agent_name": "系统",
|
||
"title": "素材充分性提示", "detail": warning,
|
||
})
|
||
return {
|
||
"current_outline": edited,
|
||
"last_user_intervention": user_input,
|
||
"status": "writing_draft",
|
||
"progress_events": events,
|
||
}
|
||
|
||
# re_outline
|
||
feedback = user_input.get("outlineFeedback", "")
|
||
updates: dict = {
|
||
"last_user_intervention": {**user_input, "_feedback": feedback},
|
||
"status": "writing_outline",
|
||
"progress_events": [{"type": "resumed", "agent_name": "系统", "message": f"重新规划大纲（{feedback[:30]}）"}],
|
||
}
|
||
# 用户在卡片里手动改过大纲 → 以手动版作为本次重规划的编辑基准
|
||
raw_edited = user_input.get("editedOutline")
|
||
if raw_edited:
|
||
updates["current_outline"] = _normalize_outline(raw_edited)
|
||
return updates
|
||
|
||
|
||
async def pause_draft_confirm(state: AIWritingState, config: RunnableConfig) -> dict:
|
||
"""PAUSE-3：草稿确认"""
|
||
draft = state["current_draft"]
|
||
user_input: dict = interrupt({
|
||
"pause_point": "draft_confirm",
|
||
"message": "草稿已完成，请确认或给出修改意见。",
|
||
"draft_title": draft["title"],
|
||
"revision_count": draft["revision_count"],
|
||
"review_result": state.get("review_result"),
|
||
})
|
||
|
||
action = user_input.get("action", "to_editor")
|
||
|
||
if action == "finalize":
|
||
await _mark_session_done(config)
|
||
return {
|
||
"last_user_intervention": user_input,
|
||
"status": "done",
|
||
"progress_events": [{"type": "resumed", "agent_name": "系统", "message": "用户确认定稿"}],
|
||
}
|
||
|
||
if action == "user_revise":
|
||
notes = user_input.get("userRevisionNotes", "")
|
||
return {
|
||
"last_user_intervention": {**user_input, "_user_notes": notes},
|
||
"revision_count": state.get("revision_count", 0) + 1,
|
||
"status": "revising",
|
||
"progress_events": [{"type": "resumed", "agent_name": "系统", "message": f"按用户意见重写（{notes[:30]}）"}],
|
||
}
|
||
|
||
# to_editor
|
||
return {
|
||
"last_user_intervention": user_input,
|
||
"status": "reviewing",
|
||
"progress_events": [{"type": "resumed", "agent_name": "系统", "message": "提交编辑审核"}],
|
||
}
|
||
|
||
|
||
async def pause_section_help(state: AIWritingState) -> dict:
|
||
"""素材不足求助点：严格模式下作家发现章节缺素材，让用户逐章决定如何处理。
|
||
|
||
用户对每个被阻塞章节选择：
|
||
- supplement：补充检索素材（整体回到检索流程重走）
|
||
- delete：删除该章节
|
||
- loose：按通用知识续写该章节
|
||
"""
|
||
blocked = state.get("blocked_sections") or []
|
||
user_input: dict = interrupt({
|
||
"pause_point": "section_help",
|
||
"message": "部分章节缺少足够素材支撑写作，请决定如何处理。",
|
||
"blocked_sections": blocked,
|
||
})
|
||
|
||
# sectionDecisions: [{"sectionTitle": "...", "decision": "supplement|delete|loose"}]
|
||
by_title: dict = {}
|
||
for d in user_input.get("sectionDecisions") or []:
|
||
title = d.get("sectionTitle") or d.get("section_title")
|
||
if title:
|
||
by_title[title] = d.get("decision", "loose")
|
||
|
||
# 任一章节选「补充检索」→ 整体回到检索，重走「检索 → 素材确认 → 大纲 → 写作」
|
||
supplement_reasons = [
|
||
b["reason"] for b in blocked if by_title.get(b["section_title"]) == "supplement"
|
||
]
|
||
if supplement_reasons:
|
||
return {
|
||
# supplement_search：补充检索（保留已有素材、追加新素材），完成后直接回写作、不重排大纲
|
||
"last_user_intervention": {**user_input, "action": "supplement_search", "_extra_keywords": supplement_reasons},
|
||
"status": "researching",
|
||
"blocked_sections": None,
|
||
"progress_events": [{"type": "resumed", "agent_name": "系统", "message": "补充检索素材中…"}],
|
||
}
|
||
|
||
# 否则：删除标记章节、记录「通用知识续写」章节，直接回写作
|
||
outline = state.get("current_outline") or {"title": "", "sections": []}
|
||
kept, loose_titles = [], []
|
||
for s in outline.get("sections", []):
|
||
title = s.get("section_title") or s.get("sectionTitle", "")
|
||
decision = by_title.get(title)
|
||
if decision == "delete":
|
||
continue
|
||
if decision == "loose":
|
||
loose_titles.append(title)
|
||
kept.append(s)
|
||
# 防御：用户把所有章节都删了 → 保留原大纲并全部按通用知识写，避免空文章
|
||
if not kept:
|
||
kept = outline.get("sections", [])
|
||
loose_titles = [s.get("section_title") or s.get("sectionTitle", "") for s in kept]
|
||
return {
|
||
"current_outline": {**outline, "sections": kept},
|
||
"last_user_intervention": {**user_input, "_loose_sections": loose_titles},
|
||
"status": "writing_draft",
|
||
"blocked_sections": None,
|
||
"progress_events": [{"type": "resumed", "agent_name": "系统", "message": "已按你的选择继续写作…"}],
|
||
}
|
||
|
||
|
||
async def pause_review_confirm(state: AIWritingState, config: RunnableConfig) -> dict:
|
||
"""PAUSE-4：编辑打回确认"""
|
||
user_input: dict = interrupt({
|
||
"pause_point": "review_confirm",
|
||
"message": "编辑认为文章需要修改，请决定下一步操作。",
|
||
"review_result": state["review_result"],
|
||
"revision_count": state.get("revision_count", 0),
|
||
"max_revisions": state.get("max_revisions", 3),
|
||
})
|
||
|
||
if user_input.get("overrideVerdict") or user_input.get("action") == "force_finalize":
|
||
await _mark_session_done(config)
|
||
return {
|
||
"last_user_intervention": user_input,
|
||
"status": "done",
|
||
"progress_events": [{"type": "resumed", "agent_name": "系统", "message": "用户强制定稿"}],
|
||
}
|
||
|
||
action = user_input.get("action", "accept_review")
|
||
extra_notes = user_input.get("userRevisionNotes", "")
|
||
return {
|
||
"last_user_intervention": {**user_input, "_extra_notes": extra_notes, "action": action},
|
||
"revision_count": state.get("revision_count", 0) + 1,
|
||
"status": "revising",
|
||
"progress_events": [{"type": "resumed", "agent_name": "系统", "message": "按编辑意见重写"}],
|
||
}
|
||
|
||
|
||
# ── 条件路由 ──────────────────────────────────────────────────────────────────
|
||
|
||
def after_researcher(state: AIWritingState) -> str:
|
||
"""检索完成后：补充检索（素材不足求助）直接回写作，否则去素材确认。"""
|
||
return "writer_draft" if state.get("status") == "writing_draft" else "pause_material"
|
||
|
||
|
||
def after_material_pause(state: AIWritingState) -> str:
|
||
return "researcher" if state["status"] == "researching" else "writer_outline"
|
||
|
||
|
||
def after_outline_pause(state: AIWritingState) -> str:
|
||
return "writer_outline" if state["status"] == "writing_outline" else "writer_draft"
|
||
|
||
|
||
def after_writer_draft(state: AIWritingState) -> str:
|
||
"""写作完成后：素材不足则去求助暂停点，否则去草稿确认。"""
|
||
return "pause_section_help" if state["status"] == "awaiting_section_help" else "pause_draft"
|
||
|
||
|
||
def after_section_help_pause(state: AIWritingState) -> str:
|
||
"""求助暂停点之后：补充检索回 researcher，否则回写作。"""
|
||
return "researcher" if state["status"] == "researching" else "writer_draft"
|
||
|
||
|
||
def after_draft_pause(state: AIWritingState) -> str:
|
||
status = state["status"]
|
||
if status == "done":
|
||
return "end"
|
||
if status == "revising":
|
||
return "writer_draft"
|
||
return "editor"
|
||
|
||
|
||
def after_review(state: AIWritingState) -> str:
|
||
review = state.get("review_result") or {}
|
||
if review.get("verdict") == "pass":
|
||
return "pause_draft"
|
||
rev_count = state.get("revision_count", 0)
|
||
max_rev = state.get("max_revisions", 3)
|
||
if rev_count >= max_rev:
|
||
return "pause_draft" # 超出上限，强制到草稿确认
|
||
return "pause_review"
|
||
|
||
|
||
def after_review_pause(state: AIWritingState) -> str:
|
||
return "end" if state["status"] == "done" else "writer_draft"
|
||
|
||
|
||
|
||
# ── LangGraph Server 入口 ────────────────────────────────────────────────────
|
||
#
|
||
# compile 出裸 graph（不带 checkpointer），由 LangGraph Server 注入主 checkpointer
|
||
# （langgraph.json 里 ``make_checkpointer`` 配的那个）。AI 写作的 thread 状态与
|
||
# 主聊天共享同一份 saver —— LangGraph schema 用 ``(thread_id, checkpoint_ns)`` 做
|
||
# 主键，UUID 不会撞。
|
||
def make_ai_writing_graph(config=None):
|
||
"""LangGraph Server graph factory：不接收 checkpointer，由 Server 注入。
|
||
|
||
Args:
|
||
config: LangGraph Server 调用工厂时注入的 RunnableConfig（暂未消费）。
|
||
"""
|
||
g = StateGraph(AIWritingState)
|
||
|
||
g.add_node("intent_parser", intent_parser_node)
|
||
g.add_node("researcher", researcher_node)
|
||
g.add_node("pause_material", pause_material_confirm)
|
||
g.add_node("writer_outline", writer_outline_node)
|
||
g.add_node("pause_outline", pause_outline_confirm)
|
||
g.add_node("writer_draft", writer_draft_node)
|
||
g.add_node("pause_section_help", pause_section_help)
|
||
g.add_node("pause_draft", pause_draft_confirm)
|
||
g.add_node("editor", editor_node)
|
||
g.add_node("pause_review", pause_review_confirm)
|
||
|
||
g.set_entry_point("intent_parser")
|
||
g.add_edge("intent_parser", "researcher")
|
||
g.add_conditional_edges("researcher", after_researcher,
|
||
{"pause_material": "pause_material", "writer_draft": "writer_draft"})
|
||
g.add_conditional_edges("pause_material", after_material_pause,
|
||
{"researcher": "researcher", "writer_outline": "writer_outline"})
|
||
g.add_edge("writer_outline", "pause_outline")
|
||
g.add_conditional_edges("pause_outline", after_outline_pause,
|
||
{"writer_outline": "writer_outline", "writer_draft": "writer_draft"})
|
||
g.add_conditional_edges("writer_draft", after_writer_draft,
|
||
{"pause_section_help": "pause_section_help", "pause_draft": "pause_draft"})
|
||
g.add_conditional_edges("pause_section_help", after_section_help_pause,
|
||
{"researcher": "researcher", "writer_draft": "writer_draft"})
|
||
g.add_conditional_edges("pause_draft", after_draft_pause,
|
||
{"end": END, "writer_draft": "writer_draft", "editor": "editor"})
|
||
g.add_conditional_edges("editor", after_review,
|
||
{"pause_draft": "pause_draft", "pause_review": "pause_review"})
|
||
g.add_conditional_edges("pause_review", after_review_pause,
|
||
{"end": END, "writer_draft": "writer_draft"})
|
||
|
||
return g.compile()
|