diff --git a/src/backend/langflow/template/frontend_node/memories.py b/src/backend/langflow/template/frontend_node/memories.py index 20c3c9272..7cf4096c9 100644 --- a/src/backend/langflow/template/frontend_node/memories.py +++ b/src/backend/langflow/template/frontend_node/memories.py @@ -18,6 +18,27 @@ class MemoryFrontendNode(FrontendNode): value=False, ) ) + # add input_key and output_key str fields + self.template.add_field( + TemplateField( + field_type="str", + required=False, + show=True, + name="input_key", + advanced=True, + value="", + ) + ) + self.template.add_field( + TemplateField( + field_type="str", + required=False, + show=True, + name="output_key", + advanced=True, + value="", + ) + ) @staticmethod def format_field(field: TemplateField, name: Optional[str] = None) -> None: @@ -36,3 +57,8 @@ class MemoryFrontendNode(FrontendNode): field.required = False field.show = True field.advanced = False + if field.name in ["input_key", "output_key"]: + field.required = False + field.show = True + field.advanced = True + field.value = ""