From 022fa663c5e96ad1d715173a94c33b93ba1c5f92 Mon Sep 17 00:00:00 2001
From: twwu <twwu@dify.ai>
Date: Mon, 16 Dec 2024 10:55:13 +0800
Subject: [PATCH] fix: improve memoization for document detail processing modes

---
 web/app/components/datasets/documents/detail/index.tsx | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/web/app/components/datasets/documents/detail/index.tsx b/web/app/components/datasets/documents/detail/index.tsx
index 1f4b98ee9f..f4e03de4ca 100644
--- a/web/app/components/datasets/documents/detail/index.tsx
+++ b/web/app/components/datasets/documents/detail/index.tsx
@@ -157,11 +157,11 @@ const DocumentDetail: FC<Props> = ({ datasetId, documentId }) => {
 
   const mode = useMemo(() => {
     return documentDetail?.dataset_process_rule?.mode
-  }, [documentDetail])
+  }, [documentDetail?.dataset_process_rule])
 
   const parentMode = useMemo(() => {
-    return documentDetail?.dataset_process_rule.rules.parent_mode
-  }, [documentDetail])
+    return documentDetail?.dataset_process_rule?.rules?.parent_mode
+  }, [documentDetail?.dataset_process_rule])
 
   const isFullDocMode = useMemo(() => {
     return mode === 'hierarchical' && parentMode === 'full-doc'
@@ -185,7 +185,8 @@ const DocumentDetail: FC<Props> = ({ datasetId, documentId }) => {
             extension={documentDetail?.data_source_info?.upload_file?.extension}
             name={documentDetail?.name}
             wrapperCls='mr-2'
-            processMode={documentDetail?.dataset_process_rule?.mode}
+            parent_mode={parentMode}
+            processMode={mode}
           />
           <div className='flex items-center flex-wrap'>
             {embeddingAvailable && documentDetail && !documentDetail.archived && !isFullDocMode && (