fix: update _dataset handling in knowledge retrieval config (#16218)

This commit is contained in:
Wu Tianwei 2025-03-19 17:48:49 +08:00 committed by GitHub
parent 425ea4e5b6
commit 965bfc49dd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -218,13 +218,15 @@ const useConfig = (id: string, payload: KnowledgeRetrievalNodeType) => {
(async () => {
const inputs = inputRef.current
const datasetIds = inputs.dataset_ids
let _datasets = selectedDatasets
if (datasetIds?.length > 0) {
const { data: dataSetsWithDetail } = await fetchDatasets({ url: '/datasets', params: { page: 1, ids: datasetIds } as any })
_datasets = dataSetsWithDetail
setSelectedDatasets(dataSetsWithDetail)
}
const newInputs = produce(inputs, (draft) => {
draft.dataset_ids = datasetIds
draft._datasets = selectedDatasets
draft._datasets = _datasets
})
setInputs(newInputs)
setSelectedDatasetsLoaded(true)