From 90621f509bd4f06c09f4c566da291b02480d9703 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 30 Nov 2023 18:06:07 -0300 Subject: [PATCH] Refactor necessary_imports in validate.py --- src/backend/langflow/utils/validate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/langflow/utils/validate.py b/src/backend/langflow/utils/validate.py index d18e64c7f..b9236fe2d 100644 --- a/src/backend/langflow/utils/validate.py +++ b/src/backend/langflow/utils/validate.py @@ -289,7 +289,7 @@ def get_default_imports(code_string): "Tool", "VectorStore", ] - necessary_imports = find_names_in_code(code_string, langflow_imports + ["Optional", "List", "Dict", "Union"]) + necessary_imports = find_names_in_code(code_string, langflow_imports) langflow_module = importlib.import_module("langflow.field_typing") default_imports.update({name: getattr(langflow_module, name) for name in necessary_imports})