4.6.5- CoreferenceResolution Module (#631)

projects/app/src/components/core/module/Flow/components/modules/ButtonEdge.tsx

@@ -1,5 +1,11 @@
 import React from 'react';
-import { SmoothStepEdge, EdgeLabelRenderer, EdgeProps, getSmoothStepPath } from 'reactflow';
+import {
+  BezierEdge,
+  getBezierPath,
+  EdgeLabelRenderer,
+  EdgeProps,
+  getSmoothStepPath
+} from 'reactflow';
 import { Flex } from '@chakra-ui/react';
 import MyIcon from '@/components/Icon';
 
@@ -21,7 +27,7 @@ const ButtonEdge = (
     style = {}
   } = props;
 
-  const [edgePath, labelX, labelY] = getSmoothStepPath({
+  const [edgePath, labelX, labelY] = getBezierPath({
     sourceX,
     sourceY,
     sourcePosition,
@@ -42,7 +48,7 @@ const ButtonEdge = (
 
   return (
     <>
-      <SmoothStepEdge {...props} style={edgeStyle} />
+      <BezierEdge {...props} style={edgeStyle} />
       <EdgeLabelRenderer>
         <Flex
           alignItems={'center'}

projects/app/src/components/core/module/Flow/index.tsx

@@ -14,7 +14,7 @@ import 'reactflow/dist/style.css';
 import type { ModuleItemType } from '@fastgpt/global/core/module/type.d';
 
 const NodeSimple = dynamic(() => import('./components/nodes/NodeSimple'));
-const nodeTypes = {
+const nodeTypes: Record<`${FlowNodeTypeEnum}`, any> = {
   [FlowNodeTypeEnum.userGuide]: dynamic(() => import('./components/nodes/NodeUserGuide')),
   [FlowNodeTypeEnum.variable]: dynamic(() => import('./components/nodes/abandon/NodeVariable')),
   [FlowNodeTypeEnum.questionInput]: dynamic(() => import('./components/nodes/NodeQuestionInput')),
@@ -28,7 +28,8 @@ const nodeTypes = {
   [FlowNodeTypeEnum.runApp]: NodeSimple,
   [FlowNodeTypeEnum.pluginInput]: dynamic(() => import('./components/nodes/NodePluginInput')),
   [FlowNodeTypeEnum.pluginOutput]: dynamic(() => import('./components/nodes/NodePluginOutput')),
-  [FlowNodeTypeEnum.pluginModule]: NodeSimple
+  [FlowNodeTypeEnum.pluginModule]: NodeSimple,
+  [FlowNodeTypeEnum.cfr]: NodeSimple
 };
 const edgeTypes = {
   [EDGE_TYPE]: ButtonEdge

projects/app/src/components/core/module/utils.ts

@@ -1,14 +1,16 @@
+import { ModuleInputKeyEnum } from '@fastgpt/global/core/module/constants';
+import { FlowNodeTypeEnum } from '@fastgpt/global/core/module/node/constant';
 import { FlowNodeOutputTargetItemType } from '@fastgpt/global/core/module/node/type';
 import { FlowModuleItemType, ModuleItemType } from '@fastgpt/global/core/module/type';
 import { type Node, type Edge } from 'reactflow';
 
-export function flowNode2Modules({
+export const flowNode2Modules = ({
   nodes,
   edges
 }: {
   nodes: Node<FlowModuleItemType, string | undefined>[];
   edges: Edge<any>[];
-}) {
+}) => {
   const modules: ModuleItemType[] = nodes.map((item) => ({
     moduleId: item.data.moduleId,
     name: item.data.name,
@@ -48,4 +50,19 @@ export function flowNode2Modules({
   });
 
   return modules;
-}
+};
+
+export const filterExportModules = (modules: ModuleItemType[]) => {
+  modules.forEach((module) => {
+    // dataset - remove select dataset value
+    if (module.flowType === FlowNodeTypeEnum.datasetSearchNode) {
+      module.inputs.forEach((item) => {
+        if (item.key === ModuleInputKeyEnum.datasetSelectList) {
+          item.value = [];
+        }
+      });
+    }
+  });
+
+  return JSON.stringify(modules, null, 2);
+};