From 9ec66cdb1307d09d605e218b46d8b66ae1ee7683 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 18 Nov 2024 11:02:04 -0300 Subject: [PATCH] fix: Add check for 'set_event_manager' method in custom_component before calling it (#4668) 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 7e53e8424..fb4110918 100644 --- a/src/backend/base/langflow/graph/vertex/base.py +++ b/src/backend/base/langflow/graph/vertex/base.py @@ -490,7 +490,8 @@ class Vertex: ) 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(