From 026482bef4720b0fb057dc7fe20bbed1bc408910 Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Mon, 10 Apr 2023 15:12:54 -0300 Subject: [PATCH] feat(nodes.py): add required and show properties to input_key and output_key fields in ChainFrontendNode feat(nodes.py): add show property to prompt field in LLMFrontendNode fix(nodes.py): set password property to False for MemoryFrontendNode's Memory Size field --- src/backend/langflow/template/nodes.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/src/backend/langflow/template/nodes.py b/src/backend/langflow/template/nodes.py index ad9c798e5..a945d57c1 100644 --- a/src/backend/langflow/template/nodes.py +++ b/src/backend/langflow/template/nodes.py @@ -292,6 +292,7 @@ class MemoryFrontendNode(FrontendNode): field.field_type = "int" field.value = 10 field.display_name = "Memory Size" + field.password = False class ChainFrontendNode(FrontendNode): @@ -302,6 +303,14 @@ class ChainFrontendNode(FrontendNode): if "key" in field.name: field.password = False field.show = False + if field.name in ["input_key", "output_key"]: + field.required = True + field.show = True + # Separated for possible future changes + if field.name == "prompt": + # if no prompt is provided, use the default prompt + field.required = False + field.show = True class LLMFrontendNode(FrontendNode):