Feat: Rename a dataset #3221 (#7162)

### What problem does this PR solve?

Feat: Rename a dataset #3221

### Type of change


- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
balibabu
2025-04-22 10:09:41 +08:00
committed by GitHub
parent ad220a0a3c
commit e7f83b13ca
9 changed files with 151 additions and 20 deletions

View File

@ -0,0 +1,47 @@
import { useSetModalState } from '@/hooks/common-hooks';
import { useUpdateKnowledge } from '@/hooks/knowledge-hooks';
import { IKnowledge } from '@/interfaces/database/knowledge';
import { omit } from 'lodash';
import { useCallback, useState } from 'react';
export const useRenameDataset = () => {
const [dataset, setDataset] = useState<IKnowledge>({} as IKnowledge);
const {
visible: datasetRenameVisible,
hideModal: hideDatasetRenameModal,
showModal: showDatasetRenameModal,
} = useSetModalState();
const { saveKnowledgeConfiguration, loading } = useUpdateKnowledge(true);
const onDatasetRenameOk = useCallback(
async (name: string) => {
const ret = await saveKnowledgeConfiguration({
...omit(dataset, ['id', 'update_time', 'nickname', 'tenant_avatar']),
kb_id: dataset.id,
name,
});
if (ret.code === 0) {
hideDatasetRenameModal();
}
},
[saveKnowledgeConfiguration, dataset, hideDatasetRenameModal],
);
const handleShowDatasetRenameModal = useCallback(
async (record: IKnowledge) => {
setDataset(record);
showDatasetRenameModal();
},
[showDatasetRenameModal],
);
return {
datasetRenameLoading: loading,
initialDatasetName: dataset?.name,
onDatasetRenameOk,
datasetRenameVisible,
hideDatasetRenameModal,
showDatasetRenameModal: handleShowDatasetRenameModal,
};
};