fix: ts and default dataset (#329)

This commit is contained in:
Archer
2023-09-20 11:43:49 +08:00
committed by GitHub
parent 177482b33a
commit 0d94db4331
2 changed files with 21 additions and 15 deletions

View File

@@ -21,7 +21,7 @@ const KBSelect = ({
onChange: (e: SelectedKbType) => void; onChange: (e: SelectedKbType) => void;
}) => { }) => {
const theme = useTheme(); const theme = useTheme();
const { datasets, loadAllDatasets } = useDatasetStore(); const { allDatasets, loadAllDatasets } = useDatasetStore();
const { const {
isOpen: isOpenKbSelect, isOpen: isOpenKbSelect,
onOpen: onOpenKbSelect, onOpen: onOpenKbSelect,
@@ -29,8 +29,8 @@ const KBSelect = ({
} = useDisclosure(); } = useDisclosure();
const showKbList = useMemo( const showKbList = useMemo(
() => datasets.filter((item) => activeKbs.find((kb) => kb.kbId === item._id)), () => allDatasets.filter((item) => activeKbs.find((kb) => kb.kbId === item._id)),
[datasets, activeKbs] [allDatasets, activeKbs]
); );
useQuery(['loadAllDatasets'], loadAllDatasets); useQuery(['loadAllDatasets'], loadAllDatasets);
@@ -57,9 +57,12 @@ const KBSelect = ({
</Flex> </Flex>
))} ))}
</Grid> </Grid>
{isOpenKbSelect && ( <KBSelectModal
<KBSelectModal activeKbs={activeKbs} onChange={onChange} onClose={onCloseKbSelect} /> isOpen={isOpenKbSelect}
)} activeKbs={activeKbs}
onChange={onChange}
onClose={onCloseKbSelect}
/>
</> </>
); );
}; };

View File

@@ -564,15 +564,18 @@ const Settings = ({ appId }: { appId: string }) => {
defaultData={getValues('chatModel')} defaultData={getValues('chatModel')}
/> />
)} )}
<KBSelectModal {isOpenKbSelect && (
isOpen={isOpenKbSelect} <KBSelectModal
activeKbs={selectedKbList.map((item) => ({ isOpen={isOpenKbSelect}
kbId: item._id, activeKbs={selectedKbList.map((item) => ({
vectorModel: item.vectorModel kbId: item._id,
}))} vectorModel: item.vectorModel
onClose={onCloseKbSelect} }))}
onChange={replaceKbList} onClose={onCloseKbSelect}
/> onChange={replaceKbList}
/>
)}
{isOpenKbParams && ( {isOpenKbParams && (
<KbParamsModal <KbParamsModal
searchEmptyText={getValues('kb.searchEmptyText')} searchEmptyText={getValues('kb.searchEmptyText')}