From 9abad780b265ae5ff003c854779f654bb2443000 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 27 Mar 2024 15:37:22 -0300 Subject: [PATCH] Replace import statement in custom component class --- .../langflow/interface/custom/custom_component/component.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/backend/base/langflow/interface/custom/custom_component/component.py b/src/backend/base/langflow/interface/custom/custom_component/component.py index ce40b0f74..8a9ba1a22 100644 --- a/src/backend/base/langflow/interface/custom/custom_component/component.py +++ b/src/backend/base/langflow/interface/custom/custom_component/component.py @@ -34,6 +34,10 @@ class Component: if key == "user_id": setattr(self, "_user_id", value) else: + if key == "code" and "from langflow import CustomComponent" in value: + value = value.replace( + "from langflow import CustomComponent", "from langflow.custom import CustomComponent" + ) setattr(self, key, value) def __setattr__(self, key, value):