From 2b4b4a761e9d259413d0449e6daa928deb24f5e6 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 18 Nov 2024 09:47:15 -0300 Subject: [PATCH] Add check for 'set_event_manager' method in custom_component before calling it --- src/backend/base/langflow/graph/vertex/base.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/graph/vertex/base.py b/src/backend/base/langflow/graph/vertex/base.py index 7e53e84249a3..fb4110918cb9 100644 --- a/src/backend/base/langflow/graph/vertex/base.py +++ b/src/backend/base/langflow/graph/vertex/base.py @@ -490,7 +490,8 @@ async def _build( ) else: custom_component = self.custom_component - self.custom_component.set_event_manager(event_manager) + if hasattr(self.custom_component, "set_event_manager"): + self.custom_component.set_event_manager(event_manager) custom_params = initialize.loading.get_params(self.params) await self._build_results(