refactor: replace useResetDatasetList with useInvalidDatasetList (#25413)

web/app/components/app-sidebar/dataset-info/dropdown.tsx

@@ -7,7 +7,7 @@ import Menu from './menu'
 import { useSelector as useAppContextWithSelector } from '@/context/app-context'
 import { useDatasetDetailContextWithSelector } from '@/context/dataset-detail'
 import type { DataSet } from '@/models/datasets'
-import { datasetDetailQueryKeyPrefix, useResetDatasetList } from '@/service/knowledge/use-dataset'
+import { datasetDetailQueryKeyPrefix, useInvalidDatasetList } from '@/service/knowledge/use-dataset'
 import { useInvalid } from '@/service/use-base'
 import { useExportPipelineDSL } from '@/service/use-pipeline'
 import Toast from '../../base/toast'
@@ -38,13 +38,13 @@ const DropDown = ({
     setOpen(prev => !prev)
   }, [])
 
-  const resetDatasetList = useResetDatasetList()
+  const invalidDatasetList = useInvalidDatasetList()
   const invalidDatasetDetail = useInvalid([...datasetDetailQueryKeyPrefix, dataset.id])
 
   const refreshDataset = useCallback(() => {
-    resetDatasetList()
+    invalidDatasetList()
     invalidDatasetDetail()
-  }, [invalidDatasetDetail, resetDatasetList])
+  }, [invalidDatasetDetail, invalidDatasetList])
 
   const openRenameModal = useCallback(() => {
     setShowRenameModal(true)
@@ -93,13 +93,13 @@ const DropDown = ({
     try {
       await deleteDataset(dataset.id)
       Toast.notify({ type: 'success', message: t('dataset.datasetDeleted') })
-      resetDatasetList()
+      invalidDatasetList()
       replace('/datasets')
     }
     finally {
       setShowConfirmDelete(false)
     }
-  }, [dataset.id, replace, resetDatasetList, t])
+  }, [dataset.id, replace, invalidDatasetList, t])
 
   return (
     <PortalToFollowElem

web/app/components/datasets/create-from-pipeline/footer.tsx

@@ -4,7 +4,7 @@ import Divider from '../../base/divider'
 import { useTranslation } from 'react-i18next'
 import CreateFromDSLModal, { CreateFromDSLModalTab } from './create-options/create-from-dsl-modal'
 import { useRouter, useSearchParams } from 'next/navigation'
-import { useResetDatasetList } from '@/service/knowledge/use-dataset'
+import { useInvalidDatasetList } from '@/service/knowledge/use-dataset'
 
 const Footer = () => {
   const { t } = useTranslation()
@@ -14,7 +14,7 @@ const Footer = () => {
   const searchParams = useSearchParams()
   const { replace } = useRouter()
   const dslUrl = searchParams.get('remoteInstallUrl') || undefined
-  const resetDatasetList = useResetDatasetList()
+  const invalidDatasetList = useInvalidDatasetList()
 
   const activeTab = useMemo(() => {
     if (dslUrl)
@@ -34,8 +34,8 @@ const Footer = () => {
   }, [dslUrl, replace])
 
   const onImportFromDSLSuccess = useCallback(() => {
-    resetDatasetList()
+    invalidDatasetList()
-  }, [resetDatasetList])
+  }, [invalidDatasetList])
 
   return (
     <div className='absolute bottom-0 left-0 right-0 z-10 flex flex-col gap-y-4 bg-knowledge-pipeline-creation-footer-bg px-16 pb-6 backdrop-blur-[6px]'>

web/app/components/datasets/create-from-pipeline/list/create-card.tsx

@@ -2,7 +2,7 @@ import React, { useCallback } from 'react'
 import { useTranslation } from 'react-i18next'
 import { RiAddCircleLine } from '@remixicon/react'
 import { useCreatePipelineDataset } from '@/service/knowledge/use-create-dataset'
-import { useResetDatasetList } from '@/service/knowledge/use-dataset'
+import { useInvalidDatasetList } from '@/service/knowledge/use-dataset'
 import Toast from '@/app/components/base/toast'
 import { useRouter } from 'next/navigation'
 
@@ -11,7 +11,7 @@ const CreateCard = () => {
   const { push } = useRouter()
 
   const { mutateAsync: createEmptyDataset } = useCreatePipelineDataset()
-  const resetDatasetList = useResetDatasetList()
+  const invalidDatasetList = useInvalidDatasetList()
 
   const handleCreate = useCallback(async () => {
     await createEmptyDataset(undefined, {
@@ -22,7 +22,7 @@ const CreateCard = () => {
             type: 'success',
             message: t('datasetPipeline.creation.successTip'),
           })
-          resetDatasetList()
+          invalidDatasetList()
           push(`/datasets/${id}/pipeline`)
         }
       },
@@ -33,7 +33,7 @@ const CreateCard = () => {
         })
       },
     })
-  }, [createEmptyDataset, push, resetDatasetList, t])
+  }, [createEmptyDataset, push, invalidDatasetList, t])
 
   return (
     <div

web/app/components/datasets/create-from-pipeline/list/template-card/index.tsx

@@ -19,7 +19,7 @@ import Content from './content'
 import Actions from './actions'
 import { useCreatePipelineDatasetFromCustomized } from '@/service/knowledge/use-create-dataset'
 import { useInvalid } from '@/service/use-base'
-import { useResetDatasetList } from '@/service/knowledge/use-dataset'
+import { useInvalidDatasetList } from '@/service/knowledge/use-dataset'
 
 type TemplateCardProps = {
   pipeline: PipelineTemplate
@@ -44,7 +44,7 @@ const TemplateCard = ({
   }, false)
   const { mutateAsync: createDataset } = useCreatePipelineDatasetFromCustomized()
   const { handleCheckPluginDependencies } = usePluginDependencies()
-  const resetDatasetList = useResetDatasetList()
+  const invalidDatasetList = useInvalidDatasetList()
 
   const handleUseTemplate = useCallback(async () => {
     const { data: pipelineTemplateInfo } = await getPipelineTemplateInfo()
@@ -64,7 +64,7 @@ const TemplateCard = ({
           type: 'success',
           message: t('datasetPipeline.creation.successTip'),
         })
-        resetDatasetList()
+        invalidDatasetList()
         if (newDataset.pipeline_id)
           await handleCheckPluginDependencies(newDataset.pipeline_id, true)
         push(`/datasets/${newDataset.dataset_id}/pipeline`)
@@ -76,7 +76,7 @@ const TemplateCard = ({
         })
       },
     })
-  }, [getPipelineTemplateInfo, createDataset, t, handleCheckPluginDependencies, push, resetDatasetList])
+  }, [getPipelineTemplateInfo, createDataset, t, handleCheckPluginDependencies, push, invalidDatasetList])
 
   const handleShowTemplateDetails = useCallback(() => {
     setShowDetailModal(true)

web/app/components/datasets/create/empty-dataset-creation-modal/index.tsx

@@ -11,7 +11,7 @@ import Button from '@/app/components/base/button'
 
 import { ToastContext } from '@/app/components/base/toast'
 import { createEmptyDataset } from '@/service/datasets'
-import { useResetDatasetList } from '@/service/knowledge/use-dataset'
+import { useInvalidDatasetList } from '@/service/knowledge/use-dataset'
 
 type IProps = {
   show: boolean
@@ -26,7 +26,7 @@ const EmptyDatasetCreationModal = ({
   const { t } = useTranslation()
   const { notify } = useContext(ToastContext)
   const router = useRouter()
-  const resetDatasetList = useResetDatasetList()
+  const invalidDatasetList = useInvalidDatasetList()
 
   const submit = async () => {
     if (!inputValue) {
@@ -39,7 +39,7 @@ const EmptyDatasetCreationModal = ({
     }
     try {
       const dataset = await createEmptyDataset({ name: inputValue })
-      resetDatasetList()
+      invalidDatasetList()
       onHide()
       router.push(`/datasets/${dataset.id}/documents`)
     }

web/app/components/datasets/create/step-two/index.tsx

@@ -62,7 +62,7 @@ import CustomDialog from '@/app/components/base/dialog'
 import { AlertTriangle } from '@/app/components/base/icons/src/vender/solid/alertsAndFeedback'
 import { noop } from 'lodash-es'
 import { useDocLink } from '@/context/i18n'
-import { useResetDatasetList } from '@/service/knowledge/use-dataset'
+import { useInvalidDatasetList } from '@/service/knowledge/use-dataset'
 
 const TextLabel: FC<PropsWithChildren> = (props) => {
   return <label className='system-sm-semibold text-text-secondary'>{props.children}</label>
@@ -556,7 +556,7 @@ const StepTwo = ({
   })
 
   const isCreating = createFirstDocumentMutation.isPending || createDocumentMutation.isPending
-  const resetDatasetList = useResetDatasetList()
+  const invalidDatasetList = useInvalidDatasetList()
 
   const createHandle = async () => {
     const params = getCreationParams()
@@ -586,7 +586,7 @@ const StepTwo = ({
     }
     if (mutateDatasetRes)
       mutateDatasetRes()
-    resetDatasetList()
+    invalidDatasetList()
     onStepChange && onStepChange(+1)
     isSetting && onSave && onSave()
   }

web/app/components/datasets/list/datasets.tsx

@@ -5,7 +5,7 @@ import { useTranslation } from 'react-i18next'
 import NewDatasetCard from './new-dataset-card'
 import DatasetCard from './dataset-card'
 import { useSelector as useAppContextWithSelector } from '@/context/app-context'
-import { useDatasetList, useResetDatasetList } from '@/service/knowledge/use-dataset'
+import { useDatasetList, useInvalidDatasetList } from '@/service/knowledge/use-dataset'
 
 type Props = {
   tags: string[]
@@ -32,7 +32,7 @@ const Datasets = ({
     include_all: includeAll,
     keyword: keywords,
   })
-  const resetDatasetList = useResetDatasetList()
+  const invalidDatasetList = useInvalidDatasetList()
   const anchorRef = useRef<HTMLDivElement>(null)
   const observerRef = useRef<IntersectionObserver>(null)
 
@@ -58,7 +58,7 @@ const Datasets = ({
       <nav className='grid grow grid-cols-1 content-start gap-3 px-12 pt-2 sm:grid-cols-2 md:grid-cols-3 lg:grid-cols-4'>
         {isCurrentWorkspaceEditor && <NewDatasetCard />}
         {datasetList?.pages.map(({ data: datasets }) => datasets.map(dataset => (
-          <DatasetCard key={dataset.id} dataset={dataset} onSuccess={resetDatasetList} />),
+          <DatasetCard key={dataset.id} dataset={dataset} onSuccess={invalidDatasetList} />),
         ))}
         <div ref={anchorRef} className='h-0' />
       </nav>

web/app/components/datasets/settings/form/index.tsx

@@ -36,7 +36,7 @@ import ChunkStructure from '../chunk-structure'
 import Toast from '@/app/components/base/toast'
 import { RiAlertFill } from '@remixicon/react'
 import { useDocLink } from '@/context/i18n'
-import { useResetDatasetList } from '@/service/knowledge/use-dataset'
+import { useInvalidDatasetList } from '@/service/knowledge/use-dataset'
 
 const rowClass = 'flex gap-x-1'
 const labelClass = 'flex items-center shrink-0 w-[180px] h-7 pt-1'
@@ -127,7 +127,7 @@ const Form = () => {
     getMembers()
   })
 
-  const resetDatasetList = useResetDatasetList()
+  const invalidDatasetList = useInvalidDatasetList()
   const handleSave = async () => {
     if (loading)
       return
@@ -190,7 +190,7 @@ const Form = () => {
       Toast.notify({ type: 'success', message: t('common.actionMsg.modifiedSuccessfully') })
       if (mutateDatasets) {
         await mutateDatasets()
-        resetDatasetList()
+        invalidDatasetList()
       }
     }
     catch {

web/app/components/rag-pipeline/components/rag-pipeline-header/publisher/popup.tsx

@@ -38,7 +38,7 @@ import {
 import Confirm from '@/app/components/base/confirm'
 import PublishAsKnowledgePipelineModal from '../../publish-as-knowledge-pipeline-modal'
 import type { IconInfo } from '@/models/datasets'
-import { useResetDatasetList } from '@/service/knowledge/use-dataset'
+import { useInvalidDatasetList } from '@/service/knowledge/use-dataset'
 import { useProviderContext } from '@/context/provider-context'
 import classNames from '@/utils/classnames'
 import PremiumBadge from '@/app/components/base/premium-badge'
@@ -86,7 +86,7 @@ const Popup = () => {
   }] = useBoolean(false)
 
   const invalidPublishedPipelineInfo = useInvalid([...publishedPipelineInfoQueryKeyPrefix, pipelineId])
-  const resetDatasetList = useResetDatasetList()
+  const invalidDatasetList = useInvalidDatasetList()
 
   const handlePublish = useCallback(async (params?: PublishWorkflowParams) => {
     if (publishing)
@@ -130,7 +130,7 @@ const Popup = () => {
           workflowStore.getState().setPublishedAt(res.created_at)
           mutateDatasetRes?.()
           invalidPublishedPipelineInfo()
-          resetDatasetList()
+          invalidDatasetList()
         }
       }
     }
@@ -150,9 +150,7 @@ const Popup = () => {
     if (published)
       return
     handlePublish()
-  },
+  }, { exactMatch: true, useCapture: true })
-    { exactMatch: true, useCapture: true },
-  )
 
   const goToAddDocuments = useCallback(() => {
     push(`/datasets/${datasetId}/documents/create-from-pipeline`)

web/service/knowledge/use-dataset.ts

@@ -10,7 +10,7 @@ import type {
   RelatedAppResponse,
 } from '@/models/datasets'
 import { get } from '../base'
-import { useReset } from '../use-base'
+import { useInvalid } from '../use-base'
 import qs from 'qs'
 
 const NAME_SPACE = 'dataset'
@@ -36,8 +36,8 @@ export const useDatasetList = (params: DatasetListRequest) => {
   })
 }
 
-export const useResetDatasetList = () => {
+export const useInvalidDatasetList = () => {
-  return useReset([...DatasetListKey])
+  return useInvalid([...DatasetListKey])
 }
 
 export const datasetDetailQueryKeyPrefix = [NAME_SPACE, 'detail']