v4.5.2 (#439)

packages/global/core/dataset/utils.ts

@@ -1,5 +1,6 @@
 import { DatasetCollectionTypeEnum } from './constant';
 import { getFileIcon } from '../../common/file/icon';
+import { strIsLink } from '../../common/string/tools';
 
 export function getCollectionIcon(
   type: `${DatasetCollectionTypeEnum}` = DatasetCollectionTypeEnum.file,
@@ -7,9 +8,11 @@ export function getCollectionIcon(
 ) {
   if (type === DatasetCollectionTypeEnum.folder) {
     return '/imgs/files/folder.svg';
-  } else if (type === DatasetCollectionTypeEnum.link) {
+  }
+  if (type === DatasetCollectionTypeEnum.link) {
     return '/imgs/files/link.svg';
-  } else if (type === DatasetCollectionTypeEnum.virtual) {
+  }
+  if (type === DatasetCollectionTypeEnum.virtual) {
     if (name === '手动录入') {
       return '/imgs/files/manual.svg';
     } else if (name === '手动标注') {
@@ -19,3 +22,25 @@ export function getCollectionIcon(
   }
   return getFileIcon(name);
 }
+export function getSourceNameIcon({
+  sourceName,
+  sourceId
+}: {
+  sourceName: string;
+  sourceId?: string;
+}) {
+  if (strIsLink(sourceId)) {
+    return '/imgs/files/link.svg';
+  }
+  const fileIcon = getFileIcon(sourceName, '');
+  if (fileIcon) {
+    return fileIcon;
+  }
+
+  if (sourceName === '手动录入') {
+    return '/imgs/files/manual.svg';
+  } else if (sourceName === '手动标注') {
+    return '/imgs/files/mark.svg';
+  }
+  return '/imgs/files/collection.svg';
+}

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

@@ -0,0 +1,59 @@
+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',
+  chatInput = 'chatInput',
+  selectApp = 'selectApp',
+  // chat special input
+  aiSettings = 'aiSettings',
+  maxToken = 'maxToken',
+  selectChatModel = 'selectChatModel',
+  // dataset special input
+  selectDataset = 'selectDataset',
+  hidden = 'hidden'
+}
+
+export enum FlowNodeOutputTypeEnum {
+  answer = 'answer',
+  source = 'source',
+  hidden = 'hidden'
+}
+
+export enum FlowNodeTypeEnum {
+  empty = 'empty',
+  variable = 'variable',
+  userGuide = 'userGuide',
+  questionInput = 'questionInput',
+  historyNode = 'historyNode',
+  chatNode = 'chatNode',
+  datasetSearchNode = 'datasetSearchNode',
+  answerNode = 'answerNode',
+  classifyQuestion = 'classifyQuestion',
+  contentExtract = 'contentExtract',
+  httpRequest = 'httpRequest',
+  runApp = 'app',
+  pluginModule = 'pluginModule',
+  pluginInput = 'pluginInput',
+  pluginOutput = 'pluginOutput'
+}
+
+export enum FlowNodeSpecialInputKeyEnum {
+  'answerText' = 'text',
+  'agents' = 'agents', // cq agent key
+  'pluginId' = 'pluginId'
+}
+
+export enum FlowNodeValTypeEnum {
+  'string' = 'string',
+  'number' = 'number',
+  'boolean' = 'boolean',
+  'chatHistory' = 'chatHistory',
+  'datasetQuote' = 'datasetQuote',
+  'any' = 'any'
+}

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

@@ -0,0 +1,57 @@
+import {
+  FlowNodeInputTypeEnum,
+  FlowNodeOutputTypeEnum,
+  FlowNodeValTypeEnum,
+  FlowNodeTypeEnum
+} from './constant';
+
+export type FlowNodeChangeProps = {
+  moduleId: string;
+  type:
+    | 'attr' // key: attr, value: new value
+    | 'updateInput' // key: update input key, value: new input value
+    | 'replaceInput' // key: old input key, value: new input value
+    | 'addInput' // key: null, value: new input value
+    | 'delInput' // key: delete input key, value: null
+    | 'updateOutput' // key: update output key, value: new output value
+    | 'replaceOutput' // key: old output key, value: new output value
+    | 'addOutput' // key: null, value: new output value
+    | 'delOutput'; // key: delete output key, value: null
+  key?: string;
+  value?: any;
+  index?: number;
+};
+
+export type FlowNodeInputItemType = {
+  key: string; // 字段名
+  value?: any;
+  valueType?: `${FlowNodeValTypeEnum}`;
+  type: `${FlowNodeInputTypeEnum}`;
+  label: string;
+  edit?: boolean;
+  connected?: boolean;
+  description?: string;
+  placeholder?: string;
+  max?: number;
+  min?: number;
+  step?: number;
+  required?: boolean;
+  list?: { label: string; value: any }[];
+  markList?: { label: string; value: any }[];
+  customData?: () => any;
+  valueCheck?: (value: any) => boolean;
+};
+
+export type FlowNodeOutputTargetItemType = {
+  moduleId: string;
+  key: string;
+};
+export type FlowNodeOutputItemType = {
+  key: string; // 字段名
+  label?: string;
+  edit?: boolean;
+  description?: string;
+  valueType?: `${FlowNodeValTypeEnum}`;
+  type?: `${FlowNodeOutputTypeEnum}`;
+  targets: FlowNodeOutputTargetItemType[];
+};

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

@@ -0,0 +1,44 @@
+import { FlowNodeTypeEnum } from './node/constant';
+import { FlowNodeInputItemType, FlowNodeOutputItemType } from './node/type';
+
+export type FlowModuleTemplateType = {
+  id: string;
+  flowType: `${FlowNodeTypeEnum}`; // unique
+  logo?: string;
+  name: string;
+  description?: string;
+  intro?: string;
+  showStatus?: boolean; // chatting response step status
+  inputs: FlowNodeInputItemType[];
+  outputs: FlowNodeOutputItemType[];
+};
+export type FlowModuleItemType = FlowModuleTemplateType & {
+  moduleId: string;
+};
+export type SystemModuleTemplateType = {
+  label: string;
+  list: FlowModuleTemplateType[];
+}[];
+
+export type ModuleItemType = {
+  name: string;
+  logo?: string;
+  intro?: string;
+  description?: string;
+  moduleId: string;
+  position?: {
+    x: number;
+    y: number;
+  };
+  flowType: `${FlowNodeTypeEnum}`;
+  showStatus?: boolean;
+  inputs: FlowNodeInputItemType[];
+  outputs: FlowNodeOutputItemType[];
+};
+
+/* function type */
+export type SelectAppItemType = {
+  id: string;
+  name: string;
+  logo: string;
+};

packages/global/core/module/utils.ts

@@ -0,0 +1,47 @@
+import {
+  FlowNodeInputTypeEnum,
+  FlowNodeSpecialInputKeyEnum,
+  FlowNodeTypeEnum
+} from './node/constant';
+import { FlowNodeInputItemType, FlowNodeOutputItemType } from './node/type';
+import { ModuleItemType } from './type';
+
+export function getPluginTemplatePluginIdInput(pluginId: string) {
+  return {
+    key: FlowNodeSpecialInputKeyEnum.pluginId,
+    type: FlowNodeInputTypeEnum.hidden,
+    label: 'pluginId',
+    value: pluginId,
+    connected: true
+  };
+}
+
+export function formatPluginIOModules(
+  pluginId: string,
+  modules: ModuleItemType[]
+): {
+  inputs: FlowNodeInputItemType[];
+  outputs: FlowNodeOutputItemType[];
+} {
+  const pluginInput = modules.find((module) => module.flowType === FlowNodeTypeEnum.pluginInput);
+  const customOutput = modules.find((module) => module.flowType === FlowNodeTypeEnum.pluginOutput);
+
+  return {
+    inputs: pluginInput
+      ? [
+          getPluginTemplatePluginIdInput(pluginId),
+          ...pluginInput.inputs.map((item) => ({
+            ...item,
+            edit: false,
+            connected: false
+          }))
+        ]
+      : [],
+    outputs: customOutput
+      ? customOutput.outputs.map((item) => ({
+          ...item,
+          edit: false
+        }))
+      : []
+  };
+}

packages/global/core/plugin/constants.ts

@@ -0,0 +1,28 @@
+import { ModuleItemType } from '../module/type';
+
+export const defaultModules: ModuleItemType[] = [
+  {
+    moduleId: 'fph4s3',
+    name: '自定义输出',
+    flowType: 'pluginOutput',
+    showStatus: false,
+    position: {
+      x: 994.1266684738011,
+      y: -45.87689365278443
+    },
+    inputs: [],
+    outputs: []
+  },
+  {
+    moduleId: 'w09v30',
+    name: '自定义输入',
+    flowType: 'pluginInput',
+    showStatus: false,
+    position: {
+      x: 457.57860319995154,
+      y: -44.25099042468186
+    },
+    inputs: [],
+    outputs: []
+  }
+];

packages/global/core/plugin/controller.d.ts

@@ -0,0 +1,21 @@
+import type { ModuleItemType } from '../module/type.d';
+
+export type CreateOnePluginParams = {
+  name: string;
+  avatar: string;
+  intro: string;
+  modules?: ModuleItemType[];
+};
+export type UpdatePluginParams = {
+  id: string;
+  name?: string;
+  avatar?: string;
+  intro?: string;
+  modules?: ModuleItemType[];
+};
+export type PluginListItemType = {
+  _id: string;
+  name: string;
+  avatar: string;
+  intro: string;
+};

packages/global/core/plugin/type.d.ts

@@ -0,0 +1,11 @@
+import type { ModuleItemType } from '../module/type.d';
+
+export type PluginItemSchema = {
+  _id: string;
+  userId: string;
+  name: string;
+  avatar: string;
+  intro: string;
+  updateTime: Date;
+  modules: ModuleItemType[];
+};