feat: 接通 Planner → Action → Validator

app/processor/nodes/action_node.py

@@ -1,20 +1,23 @@
-"""将简单修改需求转换为动作列表"""
+"""将简单修改或新增规划转换为动作列表"""
 
 import json
 
-from langchain_core.messages import HumanMessage, SystemMessage
 from langchain_core.exceptions import OutputParserException
+from langchain_core.messages import HumanMessage, SystemMessage
 from pydantic import BaseModel, ValidationError
 
+from app.actions.plan import assemble_plan_actions, plan_block_reason, plan_materials
 from app.clients.llm import get_llm
 from app.processor.base import BaseNode
 from app.processor.state import QueryState
 from app.prompt.action import ACTION_SYSTEM_PROMPT
-from app.schemas.actions import ActionBatch
 from app.schemas.action_validation import ActionValidationError, schema_errors
+from app.schemas.actions import ActionBatch, AddAction
 
 
 def can_generate_actions(state: QueryState) -> bool:
+    if state.get("screen_plan") is not None:
+        return plan_block_reason(state) is None
     requirement = state["requirement"]
     scope = state["change_scope"]
     # 当前只接入三种已有组件操作，后续能力逐步扩展
@@ -42,6 +45,18 @@ class ActionNode(BaseNode):
 
     async def process(self, state: QueryState) -> QueryState:
         updated_state: QueryState = state.copy()
+        if state.get("screen_plan") is not None and (
+            reason := plan_block_reason(state)
+        ):
+            updated_state.update(
+                actions=None,
+                validation_passed=False,
+                validation_errors=[],
+                simulated_screen=None,
+                failure_reason=reason,
+                status="failed",
+            )
+            return updated_state
         if not can_generate_actions(state):
             raise ValueError(
                 "ActionNode 当前只支持无需规划和资源的 move、resize、update 需求"
@@ -51,6 +66,9 @@ class ActionNode(BaseNode):
             "requirement": state["requirement"].model_dump(mode="json"),
             "change_scope": state["change_scope"].model_dump(mode="json"),
         }
+        if state.get("screen_plan") is not None:
+            payload["screen_plan"] = state["screen_plan"].model_dump(mode="json")
+            payload["component_materials"] = plan_materials(state)
         if state.get("repair_plan") is not None:
             payload.update(
                 repair_plan=state["repair_plan"].model_dump(mode="json"),
@@ -79,9 +97,14 @@ class ActionNode(BaseNode):
                     HumanMessage(content=json.dumps(payload, ensure_ascii=False)),
                 ]
             )
-            batch = ActionBatch.model_validate(
-                result.model_dump() if isinstance(result, BaseModel) else result
-            )
+            raw = result.model_dump() if isinstance(result, BaseModel) else result
+            if isinstance(result, ActionBatch) and state.get("screen_plan") is not None:
+                for source, target in zip(result.actions, raw["actions"]):
+                    if isinstance(source, AddAction):
+                        target["node"] = source.node.model_dump(exclude_unset=True)
+            batch = ActionBatch.model_validate(raw)
+            if state.get("screen_plan") is not None:
+                batch = assemble_plan_actions(state, batch)
         except ValidationError as error:
             updated_state["validation_passed"] = False
             updated_state["validation_errors"] = schema_errors(error)
@@ -102,12 +125,16 @@ class ActionNode(BaseNode):
 
 if __name__ == "__main__":
     import asyncio
+    import sys
 
     from pydantic import BaseModel
 
+    from app.processor.nodes.component_resource_node import ComponentResourceNode
     from app.processor.nodes.input_node import InputNode
+    from app.processor.nodes.validator_node import ValidatorNode
     from app.schemas.change_scope import ChangeScope
     from app.schemas.requirement import Requirement
+    from app.schemas.screen_plan import ScreenPlan
 
     async def main():
         state = InputNode().process(
@@ -151,8 +178,47 @@ if __name__ == "__main__":
             needs_planning=False,
             reason="明确的组件位移",
         )
+        # 加 --add 测试规划到新增动作，默认仍测试简单移动
+        if "--add" in sys.argv:
+            state = InputNode().process(
+                {
+                    "request": {
+                        "prompt": "保留已有组件，在顶部新增销售总览标题",
+                        "screen": state["input_context"].screen.model_dump(mode="json"),
+                    }
+                }
+            )
+            state["requirement"] = Requirement(
+                intent="modify_screen",
+                goal="新增标题",
+                needs_clarification=False,
+                changes=[{"target_scope": "screen", "operation": "add"}],
+            )
+            state["change_scope"] = ChangeScope(
+                scope="local",
+                needs_planning=True,
+                reason="顶部新增标题",
+            )
+            state = ComponentResourceNode().process(state)
+            state["screen_plan"] = ScreenPlan(
+                goal="新增标题",
+                design_summary="顶部标题",
+                layout_description="保留原图表",
+                components=[
+                    {
+                        "plan_id": "heading",
+                        "disposition": "add",
+                        "component_type": "text",
+                        "purpose": "显示销售总览",
+                        "design": {"text": "销售总览"},
+                        "layout": {"x": 100, "y": 40, "width": 500, "height": 60},
+                    }
+                ],
+            )
         node = ActionNode()
         final_state = await node.process(state)
+        if "--add" in sys.argv and final_state.get("actions") is not None:
+            final_state = ValidatorNode().process(final_state)
         print(
             json.dumps(
                 final_state,