feat: 完成input_context节点

2026-09-22 19:12:31 +08:00
parent 9bddbc3aa5
commit 4d46336bc2
14 changed files with 229 additions and 14 deletions
@@ -0,0 +1,60 @@
+"""输入解析与校验，可直接运行本文件修改示例进行调试"""
+
+from app.processor.base import BaseNode
+from app.processor.state import QueryState
+from app.schemas.input_context import InputContext
+
+
+class InputNode(BaseNode):
+    """纯输入节点，不依赖模型配置；校验失败保留 ValidationError"""
+
+    name = "input_node"
+
+    def process(self, state: QueryState) -> QueryState:
+        # 深拷贝隔离原始请求，后续修改上下文不会改变提交时的快照
+        context = InputContext.model_validate(state["request"]).model_copy(deep=True)
+        updated_state: QueryState = {**state, "input_context": context}
+        return updated_state
+
+
+if __name__ == "__main__":
+    import json
+
+    # 1. 创建 state：可将 screen 替换成前端的完整大屏 JSON
+    state: QueryState = {
+        "request": {
+            "prompt": "把销售额统计向右移动 50px",
+            "screen": {
+                "canvas": {
+                    "width": 1920,
+                    "height": 1080,
+                    "backgroundColor": "#ffffff10",
+                },
+                "nodes": [
+                    {
+                        "id": "sales-chart",
+                        "type": "bar-chart",
+                        "name": "销售额统计",
+                        "layout": {"x": 26, "y": 34.8, "width": 400, "height": 260},
+                        "props": {"option": {"title": {"text": "销售额统计"}}},
+                    }
+                ],
+                "dataSource": [],
+            },
+            "context": {"selected_node_ids": ["sales-chart"]},
+            "attachments": [],
+        }
+    }
+
+    # 2. 执行节点
+    node = InputNode()
+    final_state = node.process(state)
+
+    # 3. 打印完整状态
+    printable_state = {
+        **final_state,
+        "input_context": final_state["input_context"].model_dump(
+            mode="json", exclude_unset=True
+        ),
+    }
+    print(json.dumps(printable_state, ensure_ascii=False, indent=2))