From 868d5fbb5246fa006d205103f66c5caf89da5a55 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 27 Mar 2024 16:09:33 -0300 Subject: [PATCH] Replace import statement with langflow.custom --- src/backend/base/langflow/utils/validate.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/backend/base/langflow/utils/validate.py b/src/backend/base/langflow/utils/validate.py index e7bd4ae05..b5363dc8e 100644 --- a/src/backend/base/langflow/utils/validate.py +++ b/src/backend/base/langflow/utils/validate.py @@ -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)