From 56133bc57ce7eca4f1638ca05d4113bb27f57587 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 9 Feb 2024 09:38:38 -0300 Subject: [PATCH] Update update_params_with_load_from_db_fields signature to include type hint for custom_component --- src/backend/langflow/interface/initialize/loading.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/langflow/interface/initialize/loading.py b/src/backend/langflow/interface/initialize/loading.py index 40f9f4c3b..fb986c9a4 100644 --- a/src/backend/langflow/interface/initialize/loading.py +++ b/src/backend/langflow/interface/initialize/loading.py @@ -122,7 +122,7 @@ async def instantiate_based_on_type(class_object, base_type, node_type, load_fro return class_object(**params) -def update_params_with_load_from_db_fields(custom_component, params, load_from_db_fields): +def update_params_with_load_from_db_fields(custom_component: "CustomComponent", params, load_from_db_fields): # For each field in load_from_db_fields, we will check if it's in the params # and if it is, we will get the value from the custom_component.keys(name) # and update the params with the value