AssistantBuilderDataSourceModal: return to list of datasource when cl… (

#3411) * AssistantBuilderDataSourceModal: return to list of datasource when closing selector * nit
dust-tt · Jan 24, 2024 · 68dde0b · 68dde0b
1 parent a6d84bb
commit 68dde0b
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/front/components/assistant_builder/AssistantBuilderDataSourceModal.tsx b/front/components/assistant_builder/AssistantBuilderDataSourceModal.tsx
@@ -50,12 +50,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 +81,7 @@ export default function AssistantBuilderDataSourceModal({
   return (
     <Modal
       isOpen={isOpen}
-      onClose={onClose}
+      onClose={selectedDataSource && !dataSourceToManage ? onReset : onClose}
       onSave={() => onSaveLocal({ isSelectAll })}
       hasChanged={
         !!selectedDataSource &&