From 74d1eeb4d3793d8c70d184bfc42bc9c7730dcf97 Mon Sep 17 00:00:00 2001
From: balibabu <cike8899@users.noreply.github.com>
Date: Fri, 8 Nov 2024 17:28:11 +0800
Subject: [PATCH] feat: Automatically save agent page data #3301 (#3302)

### What problem does this PR solve?

feat: Automatically save agent page data #3301

### Type of change


- [x] New Feature (non-breaking change which adds functionality)
---
 web/src/locales/en.ts                         |  1 +
 web/src/locales/zh-traditional.ts             |  1 +
 web/src/locales/zh.ts                         |  1 +
 web/src/pages/flow/canvas/index.tsx           | 15 ++----
 .../pages/flow/canvas/node/relevant-node.tsx  |  7 ++-
 web/src/pages/flow/header/index.less          |  2 +-
 web/src/pages/flow/header/index.tsx           | 14 +++--
 web/src/pages/flow/hooks.ts                   | 53 +++++++++++++++++--
 web/src/pages/flow/store.ts                   |  6 ++-
 web/src/pages/flow/utils.ts                   | 28 ++++++++++
 10 files changed, 105 insertions(+), 23 deletions(-)

diff --git a/web/src/locales/en.ts b/web/src/locales/en.ts
index c951120bbd0..aa5ef8df874 100644
--- a/web/src/locales/en.ts
+++ b/web/src/locales/en.ts
@@ -1036,6 +1036,7 @@ The above is the content you need to summarize.`,
       howUseId: 'How to use agent ID?',
       content: 'Content',
       operationResults: 'Operation Results',
+      autosave: 'Automatically saved',
     },
     footer: {
       profile: 'All rights reserved @ React',
diff --git a/web/src/locales/zh-traditional.ts b/web/src/locales/zh-traditional.ts
index db9b953d0b9..df390fba63a 100644
--- a/web/src/locales/zh-traditional.ts
+++ b/web/src/locales/zh-traditional.ts
@@ -984,6 +984,7 @@ export default {
       howUseId: '如何使用Agent ID？',
       content: '內容',
       operationResults: '運行結果',
+      autosave: '已自動儲存',
     },
     footer: {
       profile: '“保留所有權利 @ react”',
diff --git a/web/src/locales/zh.ts b/web/src/locales/zh.ts
index 780a3499a4c..41098f4c493 100644
--- a/web/src/locales/zh.ts
+++ b/web/src/locales/zh.ts
@@ -1004,6 +1004,7 @@ export default {
       howUseId: '如何使用Agent ID？',
       content: '内容',
       operationResults: '运行结果',
+      autosave: '已自动保存',
     },
     footer: {
       profile: 'All rights reserved @ React',
diff --git a/web/src/pages/flow/canvas/index.tsx b/web/src/pages/flow/canvas/index.tsx
index d92420e580d..c38f6e8efd3 100644
--- a/web/src/pages/flow/canvas/index.tsx
+++ b/web/src/pages/flow/canvas/index.tsx
@@ -128,6 +128,9 @@ function FlowCanvas({ chatDrawerVisible, hideChatDrawer }: IProps) {
         onSelectionChange={onSelectionChange}
         nodeOrigin={[0.5, 0]}
         isValidConnection={isValidConnection}
+        onChangeCapture={(...params) => {
+          console.info('onChangeCapture:', ...params);
+        }}
         onChange={(...params) => {
           console.info('params:', ...params);
         }}
@@ -140,18 +143,6 @@ function FlowCanvas({ chatDrawerVisible, hideChatDrawer }: IProps) {
           },
         }}
         deleteKeyCode={['Delete', 'Backspace']}
-        onPaste={(...params) => {
-          console.info('onPaste:', ...params);
-        }}
-        onPasteCapture={(...params) => {
-          console.info('onPasteCapture:', ...params);
-        }}
-        onCopy={(...params) => {
-          console.info('onCopy:', ...params);
-        }}
-        onCopyCapture={(...params) => {
-          console.info('onCopyCapture:', ...params);
-        }}
       >
         <Background />
         <Controls />
diff --git a/web/src/pages/flow/canvas/node/relevant-node.tsx b/web/src/pages/flow/canvas/node/relevant-node.tsx
index 02407fc4dfe..5e22517591e 100644
--- a/web/src/pages/flow/canvas/node/relevant-node.tsx
+++ b/web/src/pages/flow/canvas/node/relevant-node.tsx
@@ -5,12 +5,15 @@ import { NodeData } from '../../interface';
 import { RightHandleStyle } from './handle-icon';
 
 import { get } from 'lodash';
+import { useReplaceIdWithName } from '../../hooks';
 import styles from './index.less';
 import NodeHeader from './node-header';
 
 export function RelevantNode({ id, data, selected }: NodeProps<NodeData>) {
   const yes = get(data, 'form.yes');
   const no = get(data, 'form.no');
+  const replaceIdWithName = useReplaceIdWithName();
+
   return (
     <section
       className={classNames(styles.logicNode, {
@@ -50,11 +53,11 @@ export function RelevantNode({ id, data, selected }: NodeProps<NodeData>) {
       <Flex vertical gap={10}>
         <Flex vertical>
           <div className={styles.relevantLabel}>Yes</div>
-          <div className={styles.nodeText}>{yes}</div>
+          <div className={styles.nodeText}>{replaceIdWithName(yes)}</div>
         </Flex>
         <Flex vertical>
           <div className={styles.relevantLabel}>No</div>
-          <div className={styles.nodeText}>{no}</div>
+          <div className={styles.nodeText}>{replaceIdWithName(no)}</div>
         </Flex>
       </Flex>
     </section>
diff --git a/web/src/pages/flow/header/index.less b/web/src/pages/flow/header/index.less
index 5d637286e76..1d78c87d29f 100644
--- a/web/src/pages/flow/header/index.less
+++ b/web/src/pages/flow/header/index.less
@@ -1,3 +1,3 @@
 .flowHeader {
-  padding: 0 20px;
+  padding: 10px 20px;
 }
diff --git a/web/src/pages/flow/header/index.tsx b/web/src/pages/flow/header/index.tsx
index e03b03a7722..70ef319ea80 100644
--- a/web/src/pages/flow/header/index.tsx
+++ b/web/src/pages/flow/header/index.tsx
@@ -5,7 +5,11 @@ import { ArrowLeftOutlined } from '@ant-design/icons';
 import { Button, Flex, Space } from 'antd';
 import { Link, useParams } from 'umi';
 import FlowIdModal from '../flow-id-modal';
-import { useSaveGraph, useSaveGraphBeforeOpeningDebugDrawer } from '../hooks';
+import {
+  useSaveGraph,
+  useSaveGraphBeforeOpeningDebugDrawer,
+  useWatchAgentChange,
+} from '../hooks';
 import styles from './index.less';
 
 interface IProps {
@@ -20,10 +24,11 @@ const FlowHeader = ({ showChatDrawer }: IProps) => {
   const {
     visible: overviewVisible,
     hideModal: hideOverviewModal,
-    showModal: showOverviewModal,
+    // showModal: showOverviewModal,
   } = useSetModalState();
   const { visible, hideModal, showModal } = useSetModalState();
   const { id } = useParams();
+  const time = useWatchAgentChange();
 
   return (
     <>
@@ -37,7 +42,10 @@ const FlowHeader = ({ showChatDrawer }: IProps) => {
           <Link to={`/flow`}>
             <ArrowLeftOutlined />
           </Link>
-          <h3>{data.title}</h3>
+          <div className="flex flex-col">
+            <span className="font-semibold text-[18px]">{data.title}</span>
+            <span className="font-normal text-sm">已自动保存 {time}</span>
+          </div>
         </Space>
         <Space size={'large'}>
           <Button onClick={handleRun}>
diff --git a/web/src/pages/flow/hooks.ts b/web/src/pages/flow/hooks.ts
index 0fb390db1a7..8a004e41314 100644
--- a/web/src/pages/flow/hooks.ts
+++ b/web/src/pages/flow/hooks.ts
@@ -19,7 +19,9 @@ import {
 import { useFetchModelId, useSendMessageWithSse } from '@/hooks/logic-hooks';
 import { Variable } from '@/interfaces/database/chat';
 import api from '@/utils/api';
+import { useDebounceEffect } from 'ahooks';
 import { FormInstance, message } from 'antd';
+import dayjs from 'dayjs';
 import { humanId } from 'human-id';
 import { lowerFirst } from 'lodash';
 import trim from 'lodash/trim';
@@ -446,12 +448,21 @@ export const useSaveGraphBeforeOpeningDebugDrawer = (show: () => void) => {
   return handleRun;
 };
 
-export const useReplaceIdWithText = (output: unknown) => {
+export const useReplaceIdWithName = () => {
   const getNode = useGraphStore((state) => state.getNode);
 
-  const getNameById = (id?: string) => {
-    return getNode(id)?.data.name;
-  };
+  const replaceIdWithName = useCallback(
+    (id?: string) => {
+      return getNode(id)?.data.name;
+    },
+    [getNode],
+  );
+
+  return replaceIdWithName;
+};
+
+export const useReplaceIdWithText = (output: unknown) => {
+  const getNameById = useReplaceIdWithName();
 
   return {
     replacedOutput: replaceIdWithText(output, getNameById),
@@ -547,6 +558,7 @@ export const useWatchNodeFormDataChange = () => {
   );
 
   useEffect(() => {
+    console.info('xxx');
     nodes.forEach((node) => {
       const currentNode = getNode(node.id);
       const form = currentNode?.data.form ?? {};
@@ -668,3 +680,36 @@ export const useCopyPaste = () => {
     };
   }, [onPasteCapture]);
 };
+
+export const useWatchAgentChange = () => {
+  const [time, setTime] = useState<string>();
+  const nodes = useGraphStore((state) => state.nodes);
+  const edges = useGraphStore((state) => state.edges);
+  const { saveGraph } = useSaveGraph();
+  const { data: flowDetail } = useFetchFlow();
+
+  const setSaveTime = useCallback((updateTime: number) => {
+    setTime(dayjs(updateTime).format('YYYY-MM-DD HH:mm:ss'));
+  }, []);
+
+  useEffect(() => {
+    setSaveTime(flowDetail?.update_time);
+  }, [flowDetail, setSaveTime]);
+
+  const saveAgent = useCallback(async () => {
+    const ret = await saveGraph();
+    setSaveTime(ret.data.update_time);
+  }, [saveGraph, setSaveTime]);
+
+  useDebounceEffect(
+    () => {
+      saveAgent();
+    },
+    [nodes, edges],
+    {
+      wait: 1000 * 20,
+    },
+  );
+
+  return time;
+};
diff --git a/web/src/pages/flow/store.ts b/web/src/pages/flow/store.ts
index ea302ef9aad..e78246570ed 100644
--- a/web/src/pages/flow/store.ts
+++ b/web/src/pages/flow/store.ts
@@ -24,6 +24,7 @@ import { immer } from 'zustand/middleware/immer';
 import { Operator, SwitchElseTo } from './constant';
 import { NodeData } from './interface';
 import {
+  duplicateNodeForm,
   generateNodeNamesWithIncreasingIndex,
   getNodeDragHandle,
   getOperatorIndex,
@@ -242,7 +243,10 @@ const useGraphStore = create<RFState>()(
 
         addNode({
           ...(node || {}),
-          data: { ...(node?.data ?? {}), name: generateNodeName(name) },
+          data: {
+            ...duplicateNodeForm(node?.data),
+            name: generateNodeName(name),
+          },
           selected: false,
           dragging: false,
           id: `${node?.data?.label}:${humanId()}`,
diff --git a/web/src/pages/flow/utils.ts b/web/src/pages/flow/utils.ts
index e53a67b04dd..8bb1142cf8b 100644
--- a/web/src/pages/flow/utils.ts
+++ b/web/src/pages/flow/utils.ts
@@ -289,3 +289,31 @@ export const generateNodeNamesWithIncreasingIndex = (
 
   return `${name}_${index}`;
 };
+
+export const duplicateNodeForm = (nodeData?: NodeData) => {
+  const form: Record<string, any> = { ...(nodeData?.form ?? {}) };
+
+  // Delete the downstream node corresponding to the to field of the Categorize operator
+  if (nodeData?.label === Operator.Categorize) {
+    form.category_description = Object.keys(form.category_description).reduce<
+      Record<string, Record<string, any>>
+    >((pre, cur) => {
+      pre[cur] = {
+        ...form.category_description[cur],
+        to: undefined,
+      };
+      return pre;
+    }, {});
+  }
+
+  // Delete the downstream nodes corresponding to the yes and no fields of the Relevant operator
+  if (nodeData?.label === Operator.Relevant) {
+    form.yes = undefined;
+    form.no = undefined;
+  }
+
+  return {
+    ...(nodeData ?? {}),
+    form,
+  };
+};