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 application crashes (#1081)

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-25 16:04:32 -03:00 • committed by GitHub
commit 05b392ece7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

1
Makefile
View file

@ -28,6 +28,7 @@ format:
cd src/frontend && npm run format cd src/frontend && npm run format
lint: lint:
make install_backend
poetry run mypy src/backend/langflow poetry run mypy src/backend/langflow
poetry run black . --check poetry run black . --check
poetry run ruff . --fix poetry run ruff . --fix

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 = {