feat:新增代码注释

This commit is contained in:
2026-09-29 19:21:29 +08:00
parent e332259b07
commit ff8adba01b
13 changed files with 293 additions and 20 deletions
+16 -2
View File
@@ -17,30 +17,37 @@ class ClarifyNode(BaseNode):
name = "clarify_node"
async def process(self, state: QueryState) -> QueryState:
# 浅拷贝状态,保存澄清问题和等待状态时不直接修改传入 state 的顶层字段
updated_state: QueryState = state.copy()
# 获取上游确认的意图,区分意图不明确和具体需求缺少信息两种澄清场景
intent = state["intent"]
# 整理意图和输入上下文,让模型结合用户原话、大屏及已有澄清记录生成问题
payload = {
"intent": intent,
"input_context": state["input_context"].model_dump(mode="json"),
}
# 如果意图已明确,补充需求节点的结果,围绕尚缺的信息提问
if intent != "unclear":
# 以需求节点的问题为依据补齐选项,不重新判断意图
payload["requirement"] = state["requirement"].model_dump(mode="json")
llm_client = get_llm(Clarification)
# 系统消息提供稳定的澄清规则,用户消息承载本次需要分析的具体资料
messages = [
SystemMessage(content=CLARIFY_SYSTEM_PROMPT),
HumanMessage(content=json.dumps(payload, ensure_ascii=False)),
]
# 格式错误重试一次
# 最多调用两次,仅在输出结构或解析失败时重试一次,其他异常继续向上抛出
for attempt in range(2):
try:
# 异步生成澄清问题和候选项,成功后立即退出重试循环
clarification = await llm_client.ainvoke(messages)
break
except (ValidationError, OutputParserException):
# 如果首次生成格式不合法,追加明确的格式要求,再调用一次模型
if attempt == 0:
# 要求重新生成完整问题,避免截断选项改变原意
messages.append(
@@ -52,17 +59,24 @@ class ClarifyNode(BaseNode):
)
)
)
# 进入下一次尝试,使用包含格式纠正要求的消息列表重新生成
continue
# 连续格式失败时退回自由回答,需求阶段沿用已确定的问题
# 意图不明确时不读取需求,避免使用尚未生成或不适用的需求结果
requirement = state.get("requirement") if intent != "unclear" else None
# 优先使用需求节点已有的澄清问题,没有可用问题时使用通用询问
question = (
requirement.clarification_question if requirement else None
) or "你希望我对当前大屏做什么?可以描述想修改的内容,或想了解的问题"
# 只构建问题,不提供候选项,让用户自由补充,避免继续依赖模型格式修正
clarification = Clarification(question=question)
# 保存模型生成或兜底构建的澄清内容,供后续暂停流程展示给用户
updated_state["clarification"] = clarification
# 在进入 interrupt 前保存等待状态,暂停中的节点尚不会返回更新
updated_state["status"] = "waiting_for_input"
# 返回澄清内容和等待状态,实际暂停由后续调用 interrupt 的节点完成
return updated_state