diff --git a/web/app/components/app/chat/log/index.tsx b/web/app/components/app/chat/log/index.tsx index d4c1cff2b2..c4111eb2e1 100644 --- a/web/app/components/app/chat/log/index.tsx +++ b/web/app/components/app/chat/log/index.tsx @@ -11,7 +11,10 @@ const Log: FC = ({ logItem, }) => { const { t } = useTranslation() - const { setCurrentLogItem, setShowPromptLogModal, setShowAgentLogModal, setShowMessageLogModal } = useAppStore() + const setCurrentLogItem = useAppStore(s => s.setCurrentLogItem) + const setShowPromptLogModal = useAppStore(s => s.setShowPromptLogModal) + const setShowAgentLogModal = useAppStore(s => s.setShowAgentLogModal) + const setShowMessageLogModal = useAppStore(s => s.setShowMessageLogModal) const { workflow_run_id: runID, agent_thoughts } = logItem const isAgent = agent_thoughts && agent_thoughts.length > 0 diff --git a/web/app/components/app/log/list.tsx b/web/app/components/app/log/list.tsx index 35bf7e7b60..776daf3f72 100644 --- a/web/app/components/app/log/list.tsx +++ b/web/app/components/app/log/list.tsx @@ -12,6 +12,7 @@ import { get } from 'lodash-es' import InfiniteScroll from 'react-infinite-scroll-component' import dayjs from 'dayjs' import { createContext, useContext } from 'use-context-selector' +import { useShallow } from 'zustand/react/shallow' import { useTranslation } from 'react-i18next' import cn from 'classnames' import s from './style.module.css' @@ -154,7 +155,16 @@ type IDetailPanel = { function DetailPanel({ detail, onFeedback }: IDetailPanel) { const { onClose, appDetail } = useContext(DrawerContext) - const { currentLogItem, setCurrentLogItem, showPromptLogModal, setShowPromptLogModal, showAgentLogModal, setShowAgentLogModal, showMessageLogModal, setShowMessageLogModal } = useAppStore() + const { currentLogItem, setCurrentLogItem, showPromptLogModal, setShowPromptLogModal, showAgentLogModal, setShowAgentLogModal, showMessageLogModal, setShowMessageLogModal } = useAppStore(useShallow(state => ({ + currentLogItem: state.currentLogItem, + setCurrentLogItem: state.setCurrentLogItem, + showPromptLogModal: state.showPromptLogModal, + setShowPromptLogModal: state.setShowPromptLogModal, + showAgentLogModal: state.showAgentLogModal, + setShowAgentLogModal: state.setShowAgentLogModal, + showMessageLogModal: state.showMessageLogModal, + setShowMessageLogModal: state.setShowMessageLogModal, + }))) const { t } = useTranslation() const [items, setItems] = React.useState([]) const [hasMore, setHasMore] = useState(true) diff --git a/web/app/components/app/switch-app-modal/index.tsx b/web/app/components/app/switch-app-modal/index.tsx index c0b744da31..3644d3d4e8 100644 --- a/web/app/components/app/switch-app-modal/index.tsx +++ b/web/app/components/app/switch-app-modal/index.tsx @@ -35,7 +35,7 @@ const SwitchAppModal = ({ show, appDetail, inAppDetail = false, onSuccess, onClo const { push, replace } = useRouter() const { t } = useTranslation() const { notify } = useContext(ToastContext) - const { setAppDetail } = useAppStore() + const setAppDetail = useAppStore(s => s.setAppDetail) const { isCurrentWorkspaceManager } = useAppContext() const { plan, enableBilling } = useProviderContext() diff --git a/web/app/components/app/text-generate/item/index.tsx b/web/app/components/app/text-generate/item/index.tsx index 0db3c301c7..6c0b5da659 100644 --- a/web/app/components/app/text-generate/item/index.tsx +++ b/web/app/components/app/text-generate/item/index.tsx @@ -121,7 +121,8 @@ const GenerationItem: FC = ({ const [childFeedback, setChildFeedback] = useState({ rating: null, }) - const { setCurrentLogItem, setShowPromptLogModal } = useAppStore() + const setCurrentLogItem = useAppStore(s => s.setCurrentLogItem) + const setShowPromptLogModal = useAppStore(s => s.setShowPromptLogModal) const handleFeedback = async (childFeedback: Feedbacktype) => { await updateFeedback({ url: `/messages/${childMessageId}/feedbacks`, body: { rating: childFeedback.rating } }, isInstalledApp, installedAppId) diff --git a/web/app/components/base/agent-log-modal/detail.tsx b/web/app/components/base/agent-log-modal/detail.tsx index d83901d0a2..b9258a4092 100644 --- a/web/app/components/base/agent-log-modal/detail.tsx +++ b/web/app/components/base/agent-log-modal/detail.tsx @@ -30,7 +30,7 @@ const AgentLogDetail: FC = ({ const { t } = useTranslation() const { notify } = useContext(ToastContext) const [currentTab, setCurrentTab] = useState(activeTab) - const { appDetail } = useAppStore() + const appDetail = useAppStore(s => s.appDetail) const [loading, setLoading] = useState(true) const [runDetail, setRunDetail] = useState() const [list, setList] = useState([])