From 8c044513d29575be1b1aedaeea0993fe0145f0ec Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Sat, 8 Apr 2023 18:07:20 -0300 Subject: [PATCH] feat(nodes.py): add ChainFrontendNode class and format_field method to handle key field display --- src/backend/langflow/template/nodes.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/src/backend/langflow/template/nodes.py b/src/backend/langflow/template/nodes.py index 99c180c9e..dcb394f39 100644 --- a/src/backend/langflow/template/nodes.py +++ b/src/backend/langflow/template/nodes.py @@ -327,3 +327,13 @@ class LLMFrontendNode(FrontendNode): field.display_name = display_name if field.name == "model_kwargs": field.field_type = "code" + + +class ChainFrontendNode(FrontendNode): + @staticmethod + def format_field(field: TemplateField, name: Optional[str] = None) -> None: + FrontendNode.format_field(field, name) + + if "key" in field.name: + field.password = False + field.show = False