🐛 fix(base.py): handle import errors when importing utilities to prevent application crashes (#1081)

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

View file

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

View file

@ -28,10 +28,14 @@ class UtilityCreator(LangChainTypeCreator):
"""
if self.type_dict is None:
settings_service = get_settings_service()
self.type_dict = {
utility_name: import_class(f"langchain.utilities.{utility_name}")
for utility_name in utilities.__all__
}
self.type_dict = {}
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
# Filter according to settings.utilities
self.type_dict = {