From 3dfcaee7557206335db3e966cbec4c77f6603941 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 6 Dec 2023 11:05:00 -0300 Subject: [PATCH] Add get_callbacks method to PluginService --- src/backend/langflow/services/plugins/service.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/src/backend/langflow/services/plugins/service.py b/src/backend/langflow/services/plugins/service.py index ca689b810..0884d6349 100644 --- a/src/backend/langflow/services/plugins/service.py +++ b/src/backend/langflow/services/plugins/service.py @@ -45,3 +45,14 @@ class PluginService(Service): def teardown(self): for plugin in self.plugins.values(): plugin.teardown() + + def get_callbacks(self, _id=None): + callbacks = [] + for plugin in self.plugins.values(): + if hasattr(plugin, "get_callback"): + callback = plugin.get_callback(_id=_id) + if callback: + callbacks.append(callback) + return callbacks + +