fix: built in status not sync

This commit is contained in:
Joel 2025-03-06 14:54:52 +08:00
parent f9b2a1c394
commit 9034752d30
3 changed files with 10 additions and 6 deletions

View File

@ -63,7 +63,7 @@ const useEditDatasetMetadata = ({
}, [doDeleteMetaData])
const [builtInEnabled, setBuiltInEnabled] = useState(datasetMetaData?.built_in_field_enabled)
const { mutate } = useUpdateBuiltInStatus(datasetId)
const { mutateAsync: toggleBuiltInStatus } = useUpdateBuiltInStatus(datasetId)
const { data: builtInMetaData } = useBuiltInMetaDataFields()
return {
isShowEditModal,
@ -76,7 +76,7 @@ const useEditDatasetMetadata = ({
builtInMetaData: builtInMetaData?.fields,
builtInEnabled,
setBuiltInEnabled: async (enable: boolean) => {
await mutate(enable)
await toggleBuiltInStatus(enable)
setBuiltInEnabled(enable)
Toast.notify({
message: t('common.actionMsg.modifiedSuccessfully'),

View File

@ -1,4 +1,4 @@
import { useBatchUpdateDocMetadata, useDocumentMetaData } from '@/service/knowledge/use-metadata'
import { useBatchUpdateDocMetadata, useDatasetMetaData, useDocumentMetaData } from '@/service/knowledge/use-metadata'
import type { BuiltInMetadataItem } from '../types'
import { DataType, type MetadataItemWithValue } from '../types'
import { useCallback, useState } from 'react'
@ -88,7 +88,9 @@ const useMetadataDocument = ({
setIsEdit(true)
}
const builtInEnabled = true
// built in enabled is set in dataset
const { data: datasetMetaData } = useDatasetMetaData(datasetId)
const builtInEnabled = datasetMetaData?.built_in_field_enabled
// old metadata and technical params
const metadataMap = useMetadataMap()

View File

@ -102,9 +102,11 @@ export const useBatchUpdateDocMetadata = () => {
}
export const useUpdateBuiltInStatus = (datasetId: string) => {
const invalidDatasetMetaData = useInvalidDatasetMetaData(datasetId)
return useMutation({
mutationFn: (enabled: boolean) => {
return post(`/datasets/${datasetId}/metadata/built-in/${enabled ? 'enable' : 'disable'}`)
mutationFn: async (enabled: boolean) => {
await post(`/datasets/${datasetId}/metadata/built-in/${enabled ? 'enable' : 'disable'}`)
invalidDatasetMetaData()
},
})
}