diff --git a/src/backend/langflow/database/base.py b/src/backend/langflow/database/base.py index 12f757a04..13b887d8d 100644 --- a/src/backend/langflow/database/base.py +++ b/src/backend/langflow/database/base.py @@ -39,6 +39,13 @@ class Engine: cls._instance = None cls.create() + @classmethod + def teardown(cls): + logger.debug("Tearing down database engine") + if cls._instance is not None: + cls._instance.dispose() + cls._instance = None + def create_db_and_tables(): logger.debug("Creating database and tables")