Fix:Automatically enable metadata and optimize parser dialog logic (#12553)

### What problem does this PR solve?

Fix:Automatically enable metadata and optimize parser dialog logic

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

web/src/interfaces/database/knowledge.ts

@@ -67,6 +67,7 @@ export interface ParserConfig {
   tag_kb_ids?: string[];
   topn_tags?: number;
   graphrag?: { use_graphrag?: boolean };
+  enable_metadata?: boolean;
 }
 
 export interface IKnowledgeFileParserConfig {

web/src/pages/dataset/components/metedata/hooks/use-manage-modal.ts

@@ -389,6 +389,7 @@ export const useManageMetaDataModal = (
       const { data: res } = await kbService.kbUpdateMetaData({
         kb_id: id,
         metadata: data,
+        enable_metadata: true,
       });
       if (res.code === 0) {
         message.success(t('message.operated'));

web/src/pages/dataset/contexts/knowledge-base-context.tsx

@@ -0,0 +1,34 @@
+import { IKnowledge } from '@/interfaces/database/knowledge';
+import React, { createContext, ReactNode, useContext } from 'react';
+
+interface KnowledgeBaseContextType {
+  knowledgeBase: IKnowledge | null;
+  loading: boolean;
+}
+
+const KnowledgeBaseContext = createContext<
+  KnowledgeBaseContextType | undefined
+>(undefined);
+
+export const KnowledgeBaseProvider: React.FC<{
+  children: ReactNode;
+  knowledgeBase: IKnowledge | null;
+  loading: boolean;
+}> = ({ children, knowledgeBase, loading }) => {
+  return (
+    <KnowledgeBaseContext.Provider value={{ knowledgeBase, loading }}>
+      {children}
+    </KnowledgeBaseContext.Provider>
+  );
+};
+
+// eslint-disable-next-line react-refresh/only-export-components
+export const useKnowledgeBaseContext = (): KnowledgeBaseContextType => {
+  const context = useContext(KnowledgeBaseContext);
+  if (context === undefined) {
+    throw new Error(
+      'useKnowledgeBaseContext must be used within a KnowledgeBaseProvider',
+    );
+  }
+  return context;
+};

web/src/pages/dataset/dataset-setting/configuration/common-item.tsx

@@ -431,6 +431,7 @@ export function AutoMetadata({
 
   const handleSaveMetadata = (data?: IMetaDataReturnJSONSettings) => {
     form.setValue('parser_config.metadata', data || []);
+    form.setValue('parser_config.enable_metadata', true);
   };
   return (
     <>

web/src/pages/dataset/dataset/index.tsx

@@ -21,6 +21,7 @@ import {
   useManageMetadata,
 } from '../components/metedata/hooks/use-manage-modal';
 import { ManageMetadataModal } from '../components/metedata/manage-modal';
+import { useKnowledgeBaseContext } from '../contexts/knowledge-base-context';
 import { DatasetTable } from './dataset-table';
 import Generate from './generate-button/generate';
 import { ReparseDialog } from './reparse-dialog';
@@ -38,7 +39,7 @@ export default function Dataset() {
     onDocumentUploadOk,
     documentUploadLoading,
   } = useHandleUploadDocument();
-
+  const { knowledgeBase } = useKnowledgeBaseContext();
   const {
     searchString,
     documents,
@@ -221,8 +222,11 @@ export default function Dataset() {
         )}
         {reparseDialogVisible && (
           <ReparseDialog
-            // hidden={isZeroChunk || isRunning}
-            hidden={false}
+            hidden={
+              chunkNum === 0 && !knowledgeBase?.parser_config?.enable_metadata
+            }
+            // hidden={false}
+            enable_metadata={knowledgeBase?.parser_config?.enable_metadata}
             handleOperationIconClick={handleOperationIconClick}
             chunk_num={chunkNum}
             visible={reparseDialogVisible}

web/src/pages/dataset/dataset/parsing-status-cell.tsx

@@ -183,8 +183,12 @@ export function ParsingStatusCell({
       )}
       {reparseDialogVisible && (
         <ReparseDialog
-          hidden={isRunning}
+          hidden={
+            (isZeroChunk && !record?.parser_config?.enable_metadata) ||
+            isRunning
+          }
           // hidden={false}
+          enable_metadata={record?.parser_config?.enable_metadata}
           handleOperationIconClick={handleOperationIconClick}
           chunk_num={chunk_num}
           visible={reparseDialogVisible}

web/src/pages/dataset/dataset/reparse-dialog.tsx

@@ -15,6 +15,7 @@ export const ReparseDialog = memo(
   ({
     handleOperationIconClick,
     chunk_num,
+    enable_metadata = false,
     hidden = false,
     visible = true,
     hideModal,
@@ -24,6 +25,7 @@ export const ReparseDialog = memo(
       delete: boolean;
       apply_kb: boolean;
     }) => void;
+    enable_metadata?: boolean;
     visible: boolean;
     hideModal: () => void;
     hidden?: boolean;
@@ -92,13 +94,16 @@ export const ReparseDialog = memo(
           </div>
         ),
       };
-      if (chunk_num > 0) {
+      if (chunk_num > 0 && enable_metadata) {
         setFields([deleteField, applyKBField]);
-      }
-      if (chunk_num <= 0) {
+      } else if (chunk_num > 0 && !enable_metadata) {
+        setFields([deleteField]);
+      } else if (chunk_num <= 0 && enable_metadata) {
         setFields([applyKBField]);
+      } else {
+        setFields([]);
       }
-    }, [chunk_num, t]);
+    }, [chunk_num, t, enable_metadata]);
 
     const formCallbackRef = useRef<DynamicFormRef>(null);
 
@@ -167,3 +172,5 @@ export const ReparseDialog = memo(
     );
   },
 );
+
+ReparseDialog.displayName = 'ReparseDialog';

web/src/pages/dataset/index.tsx

@@ -9,6 +9,7 @@ import {
 } from '@/components/ui/breadcrumb';
 import { useNavigatePage } from '@/hooks/logic-hooks/navigate-hooks';
 import { useFetchKnowledgeBaseConfiguration } from '@/hooks/use-knowledge-request';
+import { KnowledgeBaseProvider } from '@/pages/dataset/contexts/knowledge-base-context';
 import { useTranslation } from 'react-i18next';
 import { Outlet } from 'react-router';
 import { SideBar } from './sidebar';
@@ -16,33 +17,35 @@ import { SideBar } from './sidebar';
 export default function DatasetWrapper() {
   const { navigateToDatasetList } = useNavigatePage();
   const { t } = useTranslation();
-  const { data } = useFetchKnowledgeBaseConfiguration();
+  const { data, loading } = useFetchKnowledgeBaseConfiguration();
 
   return (
-    <section className="flex h-full flex-col w-full">
-      <PageHeader>
-        <Breadcrumb>
-          <BreadcrumbList>
-            <BreadcrumbItem>
-              <BreadcrumbLink onClick={navigateToDatasetList}>
-                {t('knowledgeDetails.dataset')}
-              </BreadcrumbLink>
-            </BreadcrumbItem>
-            <BreadcrumbSeparator />
-            <BreadcrumbItem>
-              <BreadcrumbPage className="w-28 whitespace-nowrap text-ellipsis overflow-hidden">
-                {data.name}
-              </BreadcrumbPage>
-            </BreadcrumbItem>
-          </BreadcrumbList>
-        </Breadcrumb>
-      </PageHeader>
-      <div className="flex flex-1 min-h-0">
-        <SideBar></SideBar>
-        <div className="flex-1 overflow-auto">
-          <Outlet />
+    <KnowledgeBaseProvider knowledgeBase={data} loading={loading}>
+      <section className="flex h-full flex-col w-full">
+        <PageHeader>
+          <Breadcrumb>
+            <BreadcrumbList>
+              <BreadcrumbItem>
+                <BreadcrumbLink onClick={navigateToDatasetList}>
+                  {t('knowledgeDetails.dataset')}
+                </BreadcrumbLink>
+              </BreadcrumbItem>
+              <BreadcrumbSeparator />
+              <BreadcrumbItem>
+                <BreadcrumbPage className="w-28 whitespace-nowrap text-ellipsis overflow-hidden">
+                  {data.name}
+                </BreadcrumbPage>
+              </BreadcrumbItem>
+            </BreadcrumbList>
+          </Breadcrumb>
+        </PageHeader>
+        <div className="flex flex-1 min-h-0">
+          <SideBar></SideBar>
+          <div className="flex-1 overflow-auto">
+            <Outlet />
+          </div>
         </div>
-      </div>
-    </section>
+      </section>
+    </KnowledgeBaseProvider>
   );
 }