ChatGPT/FastGPT
1
0
Fork 0
mirror of https://github.com/labring/FastGPT.git synced 2025-07-21 11:43:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix DatasetSelectModal.tsx (#4241)

Browse Source
This commit is contained in:
gggaaallleee
2025-03-19 23:08:22 +08:00
committed by GitHub
parent 4ca99a6361
commit 70563d2bcb
1 changed files with 19 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
projects/app/src/components/core/app/DatasetSelectModal.tsx
View File

@@ -41,17 +41,28 @@ export const DatasetSelectModal = ({
const { Loading } = useLoading();
const filterDatasets = useMemo(() => {
const filtered = {
selected: datasets.filter((item) =>
selectedDatasets.find((dataset) => dataset.datasetId === item._id)
),
const selectedInDatasets = datasets.filter((item) =>
selectedDatasets.some((dataset) => dataset.datasetId === item._id)
);
const selectedNotInDatasets = selectedDatasets
.filter((selected) => !datasets.some((dataset) => dataset._id === selected.datasetId))
.map((selected) => ({
_id: selected.datasetId,
avatar: selected.avatar,
name: selected.name,
vectorModel: selected.vectorModel
}));
return {
selected: [...selectedInDatasets, ...selectedNotInDatasets],
unSelected: datasets.filter(
(item) => !selectedDatasets.find((dataset) => dataset.datasetId === item._id)
(item) => !selectedDatasets.some((dataset) => dataset.datasetId === item._id)
)
};
return filtered;
}, [datasets, selectedDatasets]);
const activeVectorModel = defaultSelectedDatasets[0]?.vectorModel?.model;
const activeVectorModel = selectedDatasets[0]?.vectorModel?.model;
return (
<DatasetSelectContainer
@@ -200,13 +211,8 @@ export const DatasetSelectModal = ({
<ModalFooter>
<Button
onClick={() => {
// filter out the dataset that is not in the kList
const filterDatasets = selectedDatasets.filter((dataset) => {
return datasets.find((item) => item._id === dataset.datasetId);
});
onClose();
onChange(filterDatasets);
onChange(selectedDatasets);
}}
>
{t('common:common.Done')}