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

Refactor necessary_imports in validate.py

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-30 18:06:07 -03:00
commit 90621f509b
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/langflow/utils/validate.py
View file

@ -289,7 +289,7 @@ def get_default_imports(code_string):
"Tool", "Tool",
"VectorStore", "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") langflow_module = importlib.import_module("langflow.field_typing")
default_imports.update({name: getattr(langflow_module, name) for name in necessary_imports}) default_imports.update({name: getattr(langflow_module, name) for name in necessary_imports})