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

Refactor plugin loading logic

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

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

@ -15,20 +15,24 @@ class PluginService(Service):
def __init__(self, settings_service: "SettingsService"): def __init__(self, settings_service: "SettingsService"):
self.plugins: dict[str, BasePlugin] = {} self.plugins: dict[str, BasePlugin] = {}
plugin_dir = settings_service.settings.PLUGIN_DIR # plugin_dir = settings_service.settings.PLUGIN_DIR
self.plugin_dir = plugin_dir or os.path.dirname(__file__) self.plugin_dir = os.path.dirname(__file__)
self.plugins_base_module = "langflow.services.plugins"
def load_plugins(self): def load_plugins(self):
base_files = ["base.py", "service.py", "factory.py", "__init__.py"] base_files = ["base.py", "service.py", "factory.py", "__init__.py"]
for module in os.listdir(self.plugin_dir): for module in os.listdir(self.plugin_dir):
if module.endswith(".py") and module not in base_files: if module.endswith(".py") and module not in base_files:
plugin_name = module[:-3] plugin_name = module[:-3]
module_path = f"{self.plugin_dir}.{plugin_name}" module_path = f"{self.plugins_base_module}.{plugin_name}"
mod = importlib.import_module(module_path) try:
for attr_name in dir(mod): mod = importlib.import_module(module_path)
attr = getattr(mod, attr_name) for attr_name in dir(mod):
if inspect.isclass(attr) and issubclass(attr, BasePlugin) and attr is not BasePlugin: attr = getattr(mod, attr_name)
self.register_plugin(plugin_name, attr()) if inspect.isclass(attr) and issubclass(attr, BasePlugin) and attr not in [CallbackPlugin, BasePlugin]:
self.register_plugin(plugin_name, attr())
except Exception as exc:
print(f"Error loading plugin {plugin_name}: {exc}")
def register_plugin(self, plugin_name, plugin_instance): def register_plugin(self, plugin_name, plugin_instance):
self.plugins[plugin_name] = plugin_instance self.plugins[plugin_name] = plugin_instance