From 0c16cbed0431f63a3d3bf342042945cc960c7fe4 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 30 Jan 2024 08:54:41 -0300 Subject: [PATCH] Refactor instantiate_custom_component function signature This commit refactors the function signature of the `instantiate_custom_component` function in `loading.py`. The `class_object` parameter is now annotated with the `Type["CustomComponent"]` type hint to improve type safety. --- src/backend/langflow/interface/initialize/loading.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/interface/initialize/loading.py b/src/backend/langflow/interface/initialize/loading.py index 8fab13351..d09c239d6 100644 --- a/src/backend/langflow/interface/initialize/loading.py +++ b/src/backend/langflow/interface/initialize/loading.py @@ -119,8 +119,8 @@ async def instantiate_based_on_type(class_object, base_type, node_type, params, async def instantiate_custom_component(node_type, class_object, params, user_id): params_copy = params.copy() - class_object: "CustomComponent" = eval_custom_component_code(params_copy.pop("code")) - custom_component = class_object(user_id=user_id) + class_object: Type["CustomComponent"] = eval_custom_component_code(params_copy.pop("code")) + custom_component: "CustomComponent" = class_object(user_id=user_id) if "retriever" in params_copy and hasattr(params_copy["retriever"], "as_retriever"): params_copy["retriever"] = params_copy["retriever"].as_retriever()