From 7e6c77ee4fabd6635efbe24a64653ba5c8ef2ee7 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 22 Feb 2024 15:20:27 -0300 Subject: [PATCH] Refactor process_type function to handle lowercase field types --- src/backend/langflow/interface/custom/utils.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/interface/custom/utils.py b/src/backend/langflow/interface/custom/utils.py index f0ef4444f..681b5f214 100644 --- a/src/backend/langflow/interface/custom/utils.py +++ b/src/backend/langflow/interface/custom/utils.py @@ -121,7 +121,13 @@ def get_field_properties(extra_field): def process_type(field_type: str): if field_type.startswith("list") or field_type.startswith("List"): return extract_inner_type(field_type) - return "prompt" if field_type == "Prompt" else field_type + + # field_type is a string can be Prompt or Code too + # so we just need to lower if it is the case + lowercase_type = field_type.lower() + if lowercase_type in ["prompt", "code"]: + return lowercase_type + return field_type def add_new_custom_field(