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

Replace import statement with langflow.custom

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-27 16:09:33 -03:00
commit 868d5fbb52
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/backend/base/langflow/utils/validate.py
View file

@ -157,6 +157,9 @@ def create_class(code, class_name):
if not hasattr(ast, "TypeIgnore"):
ast.TypeIgnore = create_type_ignore_class()
# Replace from langflow import CustomComponent with from langflow.custom import CustomComponent
code = code.replace("from langflow import CustomComponent", "from langflow.custom import CustomComponent")
module = ast.parse(code)
exec_globals = prepare_global_scope(code, module)