V4.6.5-alpha (#609)

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

@@ -2,32 +2,39 @@ export enum FlowNodeInputTypeEnum {
   systemInput = 'systemInput', // history, userChatInput, variableInput
 
   input = 'input', // one line input
-  textarea = 'textarea',
   numberInput = 'numberInput',
   select = 'select',
   slider = 'slider',
-  custom = 'custom',
   target = 'target', // data input
   switch = 'switch',
+  textarea = 'textarea',
+
+  addInputParam = 'addInputParam', // params input
+
   selectApp = 'selectApp',
+
   // chat special input
   aiSettings = 'aiSettings',
 
-  // model select
+  // ai model select
   selectChatModel = 'selectChatModel',
   selectCQModel = 'selectCQModel',
+  selectExtractModel = 'selectExtractModel',
 
   // dataset special input
   selectDataset = 'selectDataset',
   selectDatasetParamsModal = 'selectDatasetParamsModal',
 
-  hidden = 'hidden'
+  hidden = 'hidden',
+  custom = 'custom'
 }
 
 export enum FlowNodeOutputTypeEnum {
   answer = 'answer',
   source = 'source',
-  hidden = 'hidden'
+  hidden = 'hidden',
+
+  addOutputParam = 'addOutputParam'
 }
 
 export enum FlowNodeTypeEnum {
@@ -45,7 +52,10 @@ export enum FlowNodeTypeEnum {
   pluginModule = 'pluginModule',
   pluginInput = 'pluginInput',
   pluginOutput = 'pluginOutput',
+  textEditor = 'textEditor',
 
   // abandon
   variable = 'variable'
 }
+
+export const EDGE_TYPE = 'smoothstep';

packages/global/core/module/node/type.d.ts

@@ -1,6 +1,7 @@
 import { FlowNodeInputTypeEnum, FlowNodeOutputTypeEnum, FlowNodeTypeEnum } from './constant';
-import { ModuleDataTypeEnum, ModuleInputKeyEnum, ModuleOutputKeyEnum } from '../constants';
+import { ModuleIOValueTypeEnum, ModuleInputKeyEnum, ModuleOutputKeyEnum } from '../constants';
 import { SelectedDatasetType } from '../api';
+import { EditInputFieldMap, EditOutputFieldMap } from './type';
 
 export type FlowNodeChangeProps = {
   moduleId: string;
@@ -20,27 +21,34 @@ export type FlowNodeChangeProps = {
 };
 
 export type FlowNodeInputItemType = {
+  valueType?: `${ModuleIOValueTypeEnum}`; // data type
+  type: `${FlowNodeInputTypeEnum}`; // Node Type. Decide on a render style
   key: `${ModuleInputKeyEnum}` | string;
-  type: `${FlowNodeInputTypeEnum}`; // Decide on a render style
   value?: any;
-  valueType?: `${ModuleDataTypeEnum}`; // data type
   label: string;
   description?: string;
   required?: boolean;
+
   edit?: boolean; // Whether to allow editing
-  connected?: boolean; // unConnected field will be deleted
+  editField?: EditInputFieldMap;
+  defaultEditField?: EditNodeFieldType;
+
+  connected?: boolean; // There are incoming data
 
   showTargetInApp?: boolean;
   showTargetInPlugin?: boolean;
 
-  placeholder?: string; // input,textarea
-  list?: { label: string; value: any }[]; // select
-  step?: number; // slider max?: number;
-  max?: number;
-  min?: number;
-  markList?: { label: string; value: any }[]; // slider
+  hideInApp?: boolean;
+  hideInPlugin?: boolean;
 
-  plusField?: boolean; // plus system will show
+  placeholder?: string; // input,textarea
+
+  list?: { label: string; value: any }[]; // select
+
+  markList?: { label: string; value: any }[]; // slider
+  step?: number; // slider
+  max?: number; // slider, number input
+  min?: number; // slider, number input
 };
 
 export type FlowNodeOutputTargetItemType = {
@@ -48,15 +56,41 @@ export type FlowNodeOutputTargetItemType = {
   key: string;
 };
 export type FlowNodeOutputItemType = {
-  key: `${ModuleOutputKeyEnum}` | string;
-  label?: string;
-  edit?: boolean;
-  description?: string;
-  valueType?: `${ModuleDataTypeEnum}`;
   type?: `${FlowNodeOutputTypeEnum}`;
+  key: `${ModuleOutputKeyEnum}` | string;
+  valueType?: `${ModuleIOValueTypeEnum}`;
+
+  label?: string;
+  description?: string;
+
+  edit?: boolean;
+  editField?: EditOutputFieldMap;
+  defaultEditField?: EditNodeFieldType;
+
   targets: FlowNodeOutputTargetItemType[];
 };
 
+/* --------------- edit field ------------------- */
+export type EditInputFieldMap = EditOutputFieldMap & {
+  inputType?: boolean;
+  required?: boolean;
+};
+export type EditOutputFieldMap = {
+  name?: boolean;
+  key?: boolean;
+  description?: boolean;
+  dataType?: boolean;
+};
+export type EditNodeFieldType = {
+  inputType?: `${FlowNodeInputTypeEnum}`; // input type
+  outputType?: `${FlowNodeOutputTypeEnum}`;
+  required?: boolean;
+  key?: string;
+  label?: string;
+  description?: string;
+  valueType?: `${ModuleIOValueTypeEnum}`;
+};
+
 /* ------------- item type --------------- */
 /* ai chat modules props */
 export type AIChatModuleProps = {