Commit 0d94db43 by Archer Committed by GitHub

fix: ts and default dataset (#329)

parent 177482b3
...@@ -21,7 +21,7 @@ const KBSelect = ({ ...@@ -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 = ({ ...@@ -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 = ({ ...@@ -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}
/>
</> </>
); );
}; };
......
...@@ -564,15 +564,18 @@ const Settings = ({ appId }: { appId: string }) => { ...@@ -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')}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment