fix: #136.

skit-ai · Mar 29, 2022 · d7bdfd9 · d7bdfd9
1 parent 8943fb2
commit d7bdfd9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/dialogy/workflow/workflow.py b/dialogy/workflow/workflow.py
@@ -277,7 +277,12 @@ def run(self, input_: Input) -> Tuple[Dict[str, Any], Dict[str, Any]]:
         """
         with self.lock:
             self.input = input_
-            return self.execute().flush()
+            try:
+                return self.execute().flush()
+            except Exception as e:
+                raise e
+            finally:
+                self.__reset()
 
     def flush(self) -> Tuple[Dict[str, Any], Dict[str, Any]]:
         """