fix: 修复聊天时的性能问题

lobehub · Oct 5, 2024 · f2e1272 · f2e1272
1 parent 5a4c3b5
commit f2e1272
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/src/app/chat/ChatHeader/index.tsx b/src/app/chat/ChatHeader/index.tsx
@@ -2,6 +2,7 @@
 
 import { Skeleton, Space } from 'antd';
 import classNames from 'classnames';
+import { isEqual } from 'lodash-es';
 import React, { Suspense } from 'react';
 import { Flexbox } from 'react-layout-kit';
 
@@ -22,7 +23,7 @@ interface Props {
 export default (props: Props) => {
   const { className, style } = props;
   const { styles } = useStyles();
-  const [currentAgent] = useSessionStore((s) => [sessionSelectors.currentAgent(s)]);
+  const [currentAgent] = useSessionStore((s) => [sessionSelectors.currentAgent(s)], isEqual);
 
   return (
     <Flexbox

diff --git a/src/app/chat/ViewerMode/index.tsx b/src/app/chat/ViewerMode/index.tsx
@@ -1,6 +1,7 @@
 'use client';
 
 import classNames from 'classnames';
+import { isEqual } from 'lodash-es';
 import React, { memo } from 'react';
 import { Flexbox } from 'react-layout-kit';
 
@@ -14,10 +15,10 @@ import { useStyles } from './style';
 
 export default memo(() => {
   const { styles } = useStyles();
-  const [currentAgent, interactive] = useSessionStore((s) => [
-    sessionSelectors.currentAgent(s),
-    s.interactive,
-  ]);
+  const [currentAgent, interactive] = useSessionStore(
+    (s) => [sessionSelectors.currentAgent(s), s.interactive],
+    isEqual,
+  );
 
   return (
     <Flexbox flex={1} style={{ position: 'relative' }}>