Sandbox (#1610) (#1611)

packages/global/core/workflow/constants.ts

@@ -108,7 +108,11 @@ export enum NodeInputKeyEnum {
   ifElseList = 'ifElseList',
 
   // variable update
-  updateList = 'updateList'
+  updateList = 'updateList',
+
+  // code
+  code = 'code',
+  codeType = 'codeType' // js|py
 }
 
 export enum NodeOutputKeyEnum {
@@ -121,6 +125,7 @@ export enum NodeOutputKeyEnum {
   error = 'error',
   text = 'system_text',
   addOutputParam = 'system_addOutputParam',
+  rawResponse = 'system_rawResponse',
 
   // dataset
   datasetQuoteQA = 'quoteQA',

packages/global/core/workflow/node/constant.ts

@@ -113,7 +113,8 @@ export enum FlowNodeTypeEnum {
   stopTool = 'stopTool',
   lafModule = 'lafModule',
   ifElseNode = 'ifElseNode',
-  variableUpdate = 'variableUpdate'
+  variableUpdate = 'variableUpdate',
+  code = 'code'
 }
 
 export const EDGE_TYPE = 'default';

packages/global/core/workflow/runtime/type.d.ts

@@ -31,6 +31,8 @@ export type DispatchNodeResponseType = {
   runningTime?: number;
   query?: string;
   textOutput?: string;
+  customInputs?: Record<string, any>;
+  customOutputs?: Record<string, any>;
 
   // bill
   tokens?: number;

packages/global/core/workflow/template/constants.ts

@@ -22,6 +22,7 @@ import type { FlowNodeTemplateType } from '../type';
 import { LafModule } from './system/laf';
 import { IfElseNode } from './system/ifElse/index';
 import { VariableUpdateNode } from './system/variableUpdate';
+import { CodeNode } from './system/sandbox';
 
 /* app flow module templates */
 export const appSystemModuleTemplates: FlowNodeTemplateType[] = [
@@ -40,7 +41,8 @@ export const appSystemModuleTemplates: FlowNodeTemplateType[] = [
   AiQueryExtension,
   LafModule,
   IfElseNode,
-  VariableUpdateNode
+  VariableUpdateNode,
+  CodeNode
 ];
 /* plugin flow module templates */
 export const pluginSystemModuleTemplates: FlowNodeTemplateType[] = [
@@ -59,7 +61,8 @@ export const pluginSystemModuleTemplates: FlowNodeTemplateType[] = [
   AiQueryExtension,
   LafModule,
   IfElseNode,
-  VariableUpdateNode
+  VariableUpdateNode,
+  CodeNode
 ];
 
 /* all module */
@@ -84,5 +87,6 @@ export const moduleTemplatesFlat: FlowNodeTemplateType[] = [
   AiQueryExtension,
   LafModule,
   IfElseNode,
-  VariableUpdateNode
+  VariableUpdateNode,
+  CodeNode
 ];

packages/global/core/workflow/template/system/sandbox/constants.ts

@@ -0,0 +1,7 @@
+export const JS_TEMPLATE = `function main({data1, data2}){
+    
+    return {
+        result: data1,
+        data2
+    }
+}`;

packages/global/core/workflow/template/system/sandbox/index.ts

@@ -0,0 +1,72 @@
+import {
+  FlowNodeTemplateTypeEnum,
+  NodeInputKeyEnum,
+  NodeOutputKeyEnum,
+  WorkflowIOValueTypeEnum
+} from '../../../constants';
+import {
+  FlowNodeInputTypeEnum,
+  FlowNodeOutputTypeEnum,
+  FlowNodeTypeEnum
+} from '../../../node/constant';
+import { FlowNodeTemplateType } from '../../../type';
+import { getHandleConfig } from '../../utils';
+import { Input_Template_DynamicInput } from '../../input';
+import { Output_Template_AddOutput } from '../../output';
+import { JS_TEMPLATE } from './constants';
+
+export const CodeNode: FlowNodeTemplateType = {
+  id: FlowNodeTypeEnum.code,
+  templateType: FlowNodeTemplateTypeEnum.tools,
+  flowNodeType: FlowNodeTypeEnum.code,
+  sourceHandle: getHandleConfig(true, true, true, true),
+  targetHandle: getHandleConfig(true, true, true, true),
+  avatar: '/imgs/workflow/code.svg',
+  name: '代码运行',
+  intro: '执行一段简单的脚本代码，通常用于进行复杂的数据处理。',
+  showStatus: true,
+  version: '482',
+  inputs: [
+    {
+      ...Input_Template_DynamicInput,
+      description: '这些变量会作为代码的运行的输入参数',
+      editField: {
+        key: true,
+        valueType: true
+      }
+    },
+    {
+      key: NodeInputKeyEnum.codeType,
+      renderTypeList: [FlowNodeInputTypeEnum.hidden],
+      label: '',
+      value: 'js'
+    },
+    {
+      key: NodeInputKeyEnum.code,
+      renderTypeList: [FlowNodeInputTypeEnum.custom],
+      label: '',
+      value: JS_TEMPLATE
+    }
+  ],
+  outputs: [
+    {
+      ...Output_Template_AddOutput,
+      description: '将代码中 return 的对象作为输出，传递给后续的节点'
+    },
+    {
+      id: NodeOutputKeyEnum.rawResponse,
+      key: NodeOutputKeyEnum.rawResponse,
+      label: '完整响应数据',
+      valueType: WorkflowIOValueTypeEnum.object,
+      type: FlowNodeOutputTypeEnum.static
+    },
+    {
+      id: NodeOutputKeyEnum.error,
+      key: NodeOutputKeyEnum.error,
+      label: '运行错误',
+      description: '代码运行错误信息，成功时返回空',
+      valueType: WorkflowIOValueTypeEnum.object,
+      type: FlowNodeOutputTypeEnum.static
+    }
+  ]
+};