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

🐛 fix(base.py): handle import errors when importing utilities to prevent crashes during initialization

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-25 15:55:37 -03:00
commit 91d21a1214
1 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
src/backend/langflow/interface/utilities/base.py
View file

@ -28,10 +28,14 @@ class UtilityCreator(LangChainTypeCreator):
""" """
if self.type_dict is None: if self.type_dict is None:
settings_service = get_settings_service() settings_service = get_settings_service()
self.type_dict = { self.type_dict = {}
utility_name: import_class(f"langchain.utilities.{utility_name}") for utility_name in utilities.__all__:
for utility_name in utilities.__all__ try:
} imported = import_class(f"langchain.utilities.{utility_name}")
self.type_dict[utility_name] = imported
except Exception:
pass
self.type_dict["SQLDatabase"] = utilities.SQLDatabase self.type_dict["SQLDatabase"] = utilities.SQLDatabase
# Filter according to settings.utilities # Filter according to settings.utilities
self.type_dict = { self.type_dict = {