From 8b38e3f79d3cd4964a489b38fca3fdc6c9833a08 Mon Sep 17 00:00:00 2001
From: wangxiaolei <fatelei@gmail.com>
Date: Mon, 29 Dec 2025 10:03:15 +0800
Subject: [PATCH] feat: document batch operation tool add re-index operation
 (#30275)

---
 .../detail/completed/common/batch-action.tsx  | 14 ++++++++++++-
 .../components/datasets/documents/list.tsx    | 20 ++++++++++++++++++-
 web/i18n/en-US/dataset.ts                     |  1 +
 web/i18n/zh-Hans/dataset.ts                   |  1 +
 web/service/knowledge/use-document.ts         | 14 ++++++++++++-
 5 files changed, 47 insertions(+), 3 deletions(-)

diff --git a/web/app/components/datasets/documents/detail/completed/common/batch-action.tsx b/web/app/components/datasets/documents/detail/completed/common/batch-action.tsx
index 6cc60453dd..bf43e41dee 100644
--- a/web/app/components/datasets/documents/detail/completed/common/batch-action.tsx
+++ b/web/app/components/datasets/documents/detail/completed/common/batch-action.tsx
@@ -1,5 +1,5 @@
 import type { FC } from 'react'
-import { RiArchive2Line, RiCheckboxCircleLine, RiCloseCircleLine, RiDeleteBinLine, RiDraftLine } from '@remixicon/react'
+import { RiArchive2Line, RiCheckboxCircleLine, RiCloseCircleLine, RiDeleteBinLine, RiDraftLine, RiRefreshLine } from '@remixicon/react'
 import { useBoolean } from 'ahooks'
 import * as React from 'react'
 import { useTranslation } from 'react-i18next'
@@ -17,6 +17,7 @@ type IBatchActionProps = {
   onBatchDelete: () => Promise<void>
   onArchive?: () => void
   onEditMetadata?: () => void
+  onBatchReIndex?: () => void
   onCancel: () => void
 }
 
@@ -28,6 +29,7 @@ const BatchAction: FC<IBatchActionProps> = ({
   onArchive,
   onBatchDelete,
   onEditMetadata,
+  onBatchReIndex,
   onCancel,
 }) => {
   const { t } = useTranslation()
@@ -91,6 +93,16 @@ const BatchAction: FC<IBatchActionProps> = ({
             <span className="px-0.5">{t(`${i18nPrefix}.archive`)}</span>
           </Button>
         )}
+        {onBatchReIndex && (
+          <Button
+            variant="ghost"
+            className="gap-x-0.5 px-3"
+            onClick={onBatchReIndex}
+          >
+            <RiRefreshLine className="size-4" />
+            <span className="px-0.5">{t(`${i18nPrefix}.reIndex`)}</span>
+          </Button>
+        )}
         <Button
           variant="ghost"
           destructive
diff --git a/web/app/components/datasets/documents/list.tsx b/web/app/components/datasets/documents/list.tsx
index 670cdd659c..348ecb271e 100644
--- a/web/app/components/datasets/documents/list.tsx
+++ b/web/app/components/datasets/documents/list.tsx
@@ -26,7 +26,7 @@ import { useDatasetDetailContextWithSelector as useDatasetDetailContext } from '
 import useTimestamp from '@/hooks/use-timestamp'
 import { ChunkingMode, DataSourceType, DocumentActionType } from '@/models/datasets'
 import { DatasourceType } from '@/models/pipeline'
-import { useDocumentArchive, useDocumentDelete, useDocumentDisable, useDocumentEnable } from '@/service/knowledge/use-document'
+import { useDocumentArchive, useDocumentBatchRetryIndex, useDocumentDelete, useDocumentDisable, useDocumentEnable } from '@/service/knowledge/use-document'
 import { asyncRunSafe } from '@/utils'
 import { cn } from '@/utils/classnames'
 import { formatNumber } from '@/utils/format'
@@ -220,6 +220,7 @@ const DocumentList: FC<IDocumentListProps> = ({
   const { mutateAsync: enableDocument } = useDocumentEnable()
   const { mutateAsync: disableDocument } = useDocumentDisable()
   const { mutateAsync: deleteDocument } = useDocumentDelete()
+  const { mutateAsync: retryIndexDocument } = useDocumentBatchRetryIndex()
 
   const handleAction = (actionName: DocumentActionType) => {
     return async () => {
@@ -250,6 +251,22 @@ const DocumentList: FC<IDocumentListProps> = ({
     }
   }
 
+  const handleBatchReIndex = async () => {
+    const [e] = await asyncRunSafe<CommonResponse>(retryIndexDocument({ datasetId, documentIds: selectedIds }))
+    if (!e) {
+      onSelectedIdChange([])
+      Toast.notify({ type: 'success', message: t('common.actionMsg.modifiedSuccessfully') })
+      onUpdate()
+    }
+    else {
+      Toast.notify({ type: 'error', message: t('common.actionMsg.modifiedUnsuccessfully') })
+    }
+  }
+
+  const hasErrorDocumentsSelected = useMemo(() => {
+    return localDocs.some(doc => selectedIds.includes(doc.id) && doc.display_status === 'error')
+  }, [localDocs, selectedIds])
+
   const getFileExtension = useCallback((fileName: string): string => {
     if (!fileName)
       return ''
@@ -447,6 +464,7 @@ const DocumentList: FC<IDocumentListProps> = ({
           onBatchDisable={handleAction(DocumentActionType.disable)}
           onBatchDelete={handleAction(DocumentActionType.delete)}
           onEditMetadata={showEditModal}
+          onBatchReIndex={hasErrorDocumentsSelected ? handleBatchReIndex : undefined}
           onCancel={() => {
             onSelectedIdChange([])
           }}
diff --git a/web/i18n/en-US/dataset.ts b/web/i18n/en-US/dataset.ts
index ee1997f699..af14426690 100644
--- a/web/i18n/en-US/dataset.ts
+++ b/web/i18n/en-US/dataset.ts
@@ -170,6 +170,7 @@ const translation = {
     enable: 'Enable',
     disable: 'Disable',
     archive: 'Archive',
+    reIndex: 'Re-index',
     delete: 'Delete',
     cancel: 'Cancel',
   },
diff --git a/web/i18n/zh-Hans/dataset.ts b/web/i18n/zh-Hans/dataset.ts
index 781fb5aa94..7abd208edf 100644
--- a/web/i18n/zh-Hans/dataset.ts
+++ b/web/i18n/zh-Hans/dataset.ts
@@ -170,6 +170,7 @@ const translation = {
     enable: '启用',
     disable: '禁用',
     archive: '归档',
+    reIndex: '重新索引',
     delete: '删除',
     cancel: '取消',
   },
diff --git a/web/service/knowledge/use-document.ts b/web/service/knowledge/use-document.ts
index 476f3534bc..63713130d9 100644
--- a/web/service/knowledge/use-document.ts
+++ b/web/service/knowledge/use-document.ts
@@ -7,7 +7,7 @@ import {
 } from '@tanstack/react-query'
 import { normalizeStatusForQuery } from '@/app/components/datasets/documents/status-filter'
 import { DocumentActionType } from '@/models/datasets'
-import { del, get, patch } from '../base'
+import { del, get, patch, post } from '../base'
 import { pauseDocIndexing, resumeDocIndexing } from '../datasets'
 import { useInvalid } from '../use-base'
 
@@ -163,3 +163,15 @@ export const useDocumentResume = () => {
     },
   })
 }
+
+export const useDocumentBatchRetryIndex = () => {
+  return useMutation({
+    mutationFn: ({ datasetId, documentIds }: { datasetId: string, documentIds: string[] }) => {
+      return post<CommonResponse>(`/datasets/${datasetId}/retry`, {
+        body: {
+          document_ids: documentIds,
+        },
+      })
+    },
+  })
+}