From 2c44cde2e0b73bb3168bdb7a4b5f093155e9892f Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Thu, 22 Jun 2023 19:03:05 -0300
Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=80=20feat(memories.py):=20add=20input?=
 =?UTF-8?q?=5Fkey=20and=20output=5Fkey=20fields=20to=20MemoryFrontendNode?=
 =?UTF-8?q?=20template=20The=20MemoryFrontendNode=20template=20now=20inclu?=
 =?UTF-8?q?des=20two=20new=20fields,=20input=5Fkey=20and=20output=5Fkey,?=
 =?UTF-8?q?=20which=20are=20string=20fields=20that=20are=20not=20required?=
 =?UTF-8?q?=20and=20are=20shown=20in=20the=20advanced=20section=20of=20the?=
 =?UTF-8?q?=20template.=20These=20fields=20can=20be=20used=20to=20specify?=
 =?UTF-8?q?=20the=20input=20and=20output=20keys=20for=20the=20memory=20nod?=
 =?UTF-8?q?e.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../template/frontend_node/memories.py        | 26 +++++++++++++++++++
 1 file changed, 26 insertions(+)

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 = ""