From 71bb633da959fbdd925d504136350a9d2b4194bc Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 15 Feb 2024 16:35:23 -0300 Subject: [PATCH] Refactor build method in instantiate_custom_component function --- src/backend/langflow/interface/initialize/loading.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/interface/initialize/loading.py b/src/backend/langflow/interface/initialize/loading.py index 37214369c..b18f8c658 100644 --- a/src/backend/langflow/interface/initialize/loading.py +++ b/src/backend/langflow/interface/initialize/loading.py @@ -132,12 +132,12 @@ async def instantiate_custom_component(node_type, class_object, params, user_id) if is_async: # Await the build method directly if it's async - built_object = await custom_component.build(**params_copy) + build_result = await custom_component.build(**params_copy) else: # Call the build method directly if it's sync - built_object = custom_component.build(**params_copy) + build_result = custom_component.build(**params_copy) - return built_object, {"repr": custom_component.custom_repr()} + return build_result, {"repr": custom_component.custom_repr()} def instantiate_wrapper(node_type, class_object, params):