From da1cab825a3d0ba32cb36e1397a7d1e89a47a901 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 29 Nov 2023 07:19:03 -0300 Subject: [PATCH] Refactor update_frontend_node_with_template_values function --- src/backend/langflow/api/utils.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/api/utils.py b/src/backend/langflow/api/utils.py index 9cbe747cc..de11640e3 100644 --- a/src/backend/langflow/api/utils.py +++ b/src/backend/langflow/api/utils.py @@ -61,19 +61,21 @@ def update_frontend_node_with_template_values(frontend_node, raw_template_data): not frontend_node or "template" not in frontend_node or not raw_template_data - or "template" not in raw_template_data + or not raw_template_data.template ): return frontend_node frontend_template = frontend_node.get("template", {}) - raw_template = raw_template_data.get("template", {}) + raw_template = raw_template_data.template or {} for key, value_dict in raw_template.items(): if key == "code" or not isinstance(value_dict, dict): continue value = value_dict.get("value") - if value is not None and key in frontend_template: + template_field_type = value_dict.get("type") + frontend_node_field_type = frontend_template[key].get("type") + if value is not None and key in frontend_template and template_field_type == frontend_node_field_type: frontend_template[key]["value"] = value return frontend_node