Build lightweight AI agent admin

2026-06-08 18:14:59 +08:00
commit e164840f43
2530 changed files with 435693 additions and 0 deletions
@@ -0,0 +1,98 @@
+"""
+结束节点
+"""
+from typing import Any, Dict
+
+from ..base import BaseNode, NodeContext, NodeResult
+from ..registry import NodeRegistry
+
+
+@NodeRegistry.register
+class EndNode(BaseNode):
+    """
+    结束节点
+    
+    工作流的出口节点，输出最终结果
+    """
+    
+    node_type = 'end'
+    node_name = '结束'
+    node_category = 'basic'
+    node_icon = 'stop-circle'
+    node_description = '工作流的结束节点，输出最终结果'
+    
+    inputs = [
+        {
+            'name': 'output',
+            'type': 'any',
+            'description': '输出内容',
+        },
+    ]
+    
+    outputs = []
+    
+    def execute(self, context: NodeContext) -> NodeResult:
+        """执行结束节点"""
+        # 获取输出内容配置（前端保存的是字符串模板，如 "{{llm_response}}"）
+        output_template = self.config.get('output', '')
+        
+        if output_template and isinstance(output_template, str) and output_template.strip():
+            # 如果配置了输出模板，解析模板中的变量
+            output = context.resolve_template(output_template)
+        else:
+            # 没有配置输出模板，不输出任何内容
+            output = None
+        
+        # 处理输出变量（用于结构化输出）
+        # 只有当 output 不为 None 时才处理结构化输出
+        outputs_config = self.config.get('outputs', [])
+        if output is not None and outputs_config and isinstance(outputs_config, list):
+            # 如果定义了输出变量，构建结构化输出
+            structured_output = {}
+            for out_var in outputs_config:
+                if isinstance(out_var, dict):
+                    var_name = out_var.get('variable', '')
+                    if var_name:
+                        # 从上下文获取变量值
+                        structured_output[var_name] = context.get_variable(var_name, None)
+            
+            # 如果有结构化输出，合并到结果中
+            if structured_output:
+                if isinstance(output, dict):
+                    output = {**output, **structured_output}
+                else:
+                    output = {
+                        'result': output,
+                        **structured_output
+                    }
+        
+        return NodeResult(
+            success=True,
+            output=output,
+        )
+    
+    @classmethod
+    def get_config_schema(cls) -> Dict[str, Any]:
+        """获取配置 Schema"""
+        return {
+            'type': 'object',
+            'properties': {
+                'output': {
+                    'type': 'object',
+                    'title': '输出配置',
+                    'properties': {
+                        'type': {
+                            'type': 'string',
+                            'title': '输出类型',
+                            'enum': ['variable', 'template', 'previous'],
+                            'default': 'previous',
+                        },
+                        'value': {
+                            'type': 'string',
+                            'title': '输出值',
+                            'description': '变量名或模板',
+                        },
+                    },
+                },
+            },
+        }