后端没修完版

src/langbot/pkg/workflow/nodes/switch.py

@@ -0,0 +1,64 @@
+"""Switch Node - multi-way branch based on value
+
+Node metadata is loaded from: ../../templates/metadata/nodes/switch.yaml
+"""
+
+from __future__ import annotations
+
+from typing import Any, ClassVar
+
+from ..entities import ExecutionContext
+from ..node import WorkflowNode, workflow_node, NodePort, NodeConfig
+
+
+@workflow_node('switch')
+class SwitchNode(WorkflowNode):
+    """Switch node - multi-way branch based on value"""
+
+    type_name = "switch"
+    category = "control"
+    icon = "🔃"
+    name = "switch"
+    description = "switch"
+    name_zh = "多路分支"
+    name_en = "Switch"
+    description_zh = "根据多个条件分支工作流"
+    description_en = "Branch workflow based on multiple cases"
+
+    inputs: ClassVar[list[NodePort]] = []
+    outputs: ClassVar[list[NodePort]] = []
+    config_schema: ClassVar[list[NodeConfig]] = []
+
+    async def execute(self, inputs: dict[str, Any], context: ExecutionContext) -> dict[str, Any]:
+        expression = self.get_config("expression", "")
+        cases = self.get_config("cases", [])
+        input_data = inputs.get("input")
+
+        value = await self._evaluate_expression(expression, input_data, context)
+
+        for case in cases:
+            if str(case.get("value")) == str(value):
+                return {"matched_case": input_data, "default": None, "_matched_output": case.get("output")}
+
+        return {"matched_case": None, "default": input_data}
+
+    async def _evaluate_expression(self, expression: str, data: Any, context: ExecutionContext) -> Any:
+        if not expression:
+            return data
+
+        if expression.startswith("{{") and expression.endswith("}}"):
+            var_path = expression[2:-2].strip()
+            parts = var_path.split(".")
+
+            if parts[0] == "input":
+                result = data
+                for part in parts[1:]:
+                    if isinstance(result, dict):
+                        result = result.get(part)
+                    else:
+                        return None
+                return result
+            elif parts[0] == "variables":
+                return context.variables.get(".".join(parts[1:]))
+
+        return expression