From f9d9adf60b033fe35bb234f5d380afee1174745f Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 4 Oct 2023 12:49:10 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(manager.py):=20catch=20and?= =?UTF-8?q?=20log=20exceptions=20when=20tearing=20down=20services=20to=20p?= =?UTF-8?q?revent=20silent=20failures=20=F0=9F=90=9B=20fix(utils.py):=20ca?= =?UTF-8?q?tch=20and=20log=20exceptions=20when=20tearing=20down=20services?= =?UTF-8?q?=20and=20superuser=20to=20prevent=20silent=20failures?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/services/manager.py | 11 ++++------- src/backend/langflow/services/utils.py | 3 +++ 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/src/backend/langflow/services/manager.py b/src/backend/langflow/services/manager.py index 496544829..078d694d5 100644 --- a/src/backend/langflow/services/manager.py +++ b/src/backend/langflow/services/manager.py @@ -90,7 +90,10 @@ class ServiceManager: if service is None: continue logger.debug(f"Teardown service {service.name}") - service.teardown() + try: + service.teardown() + except Exception as exc: + logger.exception(exc) self.services = {} self.factories = {} self.dependencies = {} @@ -150,9 +153,3 @@ def initialize_session_service(): dependencies=[ServiceType.CACHE_SERVICE], ) - -def teardown_services(): - """ - Teardown all the services. - """ - service_manager.teardown() diff --git a/src/backend/langflow/services/utils.py b/src/backend/langflow/services/utils.py index c17963bc3..1909088bd 100644 --- a/src/backend/langflow/services/utils.py +++ b/src/backend/langflow/services/utils.py @@ -153,6 +153,9 @@ def teardown_services(): """ try: teardown_superuser(get_settings_service(), next(get_session())) + except Exception as exc: + logger.exception(exc) + try: service_manager.teardown() except Exception as exc: logger.exception(exc)