jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Add get_callbacks method to PluginService

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-06 11:05:00 -03:00
commit 3dfcaee755
1 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/backend/langflow/services/plugins/service.py
View file

@ -45,3 +45,14 @@ class PluginService(Service):
def teardown(self): def teardown(self):
for plugin in self.plugins.values(): for plugin in self.plugins.values():
plugin.teardown() 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