AssistantBuilderDataSourceModal: return to list of datasource when cl…

…osing selector
dust-tt · Jan 24, 2024 · 7afa81e · 7afa81e
1 parent 67806b9
commit 7afa81e
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/front/components/assistant_builder/AssistantBuilderDataSourceModal.tsx b/front/components/assistant_builder/AssistantBuilderDataSourceModal.tsx
@@ -1,4 +1,5 @@
 import {
+  ChevronLeftIcon,
   CloudArrowDownIcon,
   CloudArrowLeftRightIcon,
   Item,
@@ -50,12 +51,16 @@ export default function AssistantBuilderDataSourceModal({
     }
   }, [dataSourceToManage]);
 
+  const onReset = () => {
+    setSelectedDataSource(null);
+    setSelectedResources({});
+    setIsSelectAll(false);
+  };
+
   const onClose = () => {
     setOpen(false);
     setTimeout(() => {
-      setSelectedDataSource(null);
-      setSelectedResources({});
-      setIsSelectAll(false);
+      onReset();
     }, 200);
   };
 
@@ -77,7 +82,7 @@ export default function AssistantBuilderDataSourceModal({
   return (
     <Modal
       isOpen={isOpen}
-      onClose={onClose}
+      onClose={selectedDataSource && !dataSourceToManage ? onReset : onClose}
       onSave={() => onSaveLocal({ isSelectAll })}
       hasChanged={
         !!selectedDataSource &&