From 2ffc8b266f03ab56c333b3ed33f993441cf6d7a9 Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabriel@logspace.ai>
Date: Wed, 10 May 2023 22:31:16 -0300
Subject: [PATCH] feat(nodes.py): add optional memory field to
 MidJourneyPromptChainNode class constructor

---
 src/backend/langflow/template/nodes.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/backend/langflow/template/nodes.py b/src/backend/langflow/template/nodes.py
index 61a41589b..6ce08e57b 100644
--- a/src/backend/langflow/template/nodes.py
+++ b/src/backend/langflow/template/nodes.py
@@ -123,6 +123,13 @@ class MidJourneyPromptChainNode(FrontendNode):
                 multiline=False,
                 name="llm",
             ),
+            TemplateField(
+                field_type="BaseChatMemory",
+                required=False,
+                show=True,
+                name="memory",
+                advanced=False,
+            ),
         ],
     )
     description: str = "MidJourneyPromptChain is a chain you can use to generate new MidJourney prompts."