From 33f8b634312b09c969baf4a0ec4ca014a59455a4 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 21 Jul 2023 09:27:30 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(loading.py):=20refactor=20in?= =?UTF-8?q?stantiate=5Fcustom=5Fcomponent=20function=20to=20improve=20read?= =?UTF-8?q?ability=20and=20maintainability?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/interface/initialize/loading.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/interface/initialize/loading.py b/src/backend/langflow/interface/initialize/loading.py index 9ec149ec9..c9d12f4b9 100644 --- a/src/backend/langflow/interface/initialize/loading.py +++ b/src/backend/langflow/interface/initialize/loading.py @@ -109,7 +109,9 @@ def instantiate_based_on_type(class_object, base_type, node_type, params): def instantiate_custom_component(node_type, class_object, params): class_object = get_function_custom(params.pop("code")) - return class_object().build(**params) + custom_component = class_object() + built_object = custom_component.build(**params) + return built_object, {"repr": custom_component.custom_repr()} def instantiate_wrapper(node_type, class_object, params):