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

chore: Update import statement for CustomComponent in validate.py

Browse source
This commit is contained in:
ogabrielluiz 2024-06-07 16:35:28 -03:00
commit c4cdb186b8
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -159,7 +159,10 @@ def create_class(code, class_name):
# Replace from langflow import CustomComponent with from langflow.custom import CustomComponent
code = code.replace("from langflow import CustomComponent", "from langflow.custom import CustomComponent")
code = code.replace(
"from langflow.interface.custom.custom_component import CustomComponent",
"from langflow.custom import CustomComponent",
)
module = ast.parse(code)
exec_globals = prepare_global_scope(code, module)