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,84 @@
+"""
+模板渲染节点
+"""
+import logging
+from typing import Any, Dict
+
+from ..base import BaseNode, NodeContext, NodeResult
+from ..registry import NodeRegistry
+
+logger = logging.getLogger(__name__)
+
+
+@NodeRegistry.register
+class TemplateNode(BaseNode):
+    """
+    模板渲染节点
+    
+    使用变量渲染模板字符串
+    """
+    
+    node_type = 'template'
+    node_name = '模板'
+    node_category = 'data'
+    node_icon = 'file-text'
+    node_description = '使用变量渲染模板字符串'
+    
+    inputs = [
+        {
+            'name': 'template',
+            'type': 'string',
+            'description': '模板字符串',
+        },
+    ]
+    
+    outputs = [
+        {
+            'name': 'result',
+            'type': 'string',
+            'description': '渲染结果',
+        },
+    ]
+    
+    def execute(self, context: NodeContext) -> NodeResult:
+        """执行模板渲染"""
+        try:
+            template = self.config.get('template', '')
+            output_variable = self.config.get('output_variable', 'template_result')
+            
+            # 渲染模板
+            result = context.resolve_template(template)
+            
+            return NodeResult(
+                success=True,
+                output=result,
+                output_variables={output_variable: result},
+            )
+            
+        except Exception as e:
+            logger.exception(f'模板节点执行失败: {e}')
+            return NodeResult(
+                success=False,
+                error=str(e),
+            )
+    
+    @classmethod
+    def get_config_schema(cls) -> Dict[str, Any]:
+        """获取配置 Schema"""
+        return {
+            'type': 'object',
+            'properties': {
+                'template': {
+                    'type': 'string',
+                    'title': '模板',
+                    'description': '支持变量引用，如 {{variable_name}}',
+                    'format': 'textarea',
+                },
+                'output_variable': {
+                    'type': 'string',
+                    'title': '输出变量名',
+                    'default': 'template_result',
+                },
+            },
+            'required': ['template'],
+        }