From ef357198b9334303a0b04d1c28bbb508b5d40951 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Thu, 22 Feb 2024 12:36:49 -0300
Subject: [PATCH] Refactor import statements and update field info in
 MemoryFrontendNode class

---
 .../langflow/template/frontend_node/memories.py      | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/backend/langflow/template/frontend_node/memories.py b/src/backend/langflow/template/frontend_node/memories.py
index fc9df6291..1b2b5c9dc 100644
--- a/src/backend/langflow/template/frontend_node/memories.py
+++ b/src/backend/langflow/template/frontend_node/memories.py
@@ -4,7 +4,9 @@ from langchain_community.chat_message_histories.mongodb import (
     DEFAULT_COLLECTION_NAME,
     DEFAULT_DBNAME,
 )
-from langchain_community.chat_message_histories.postgres import DEFAULT_CONNECTION_STRING
+from langchain_community.chat_message_histories.postgres import (
+    DEFAULT_CONNECTION_STRING,
+)
 
 from langflow.template.field.base import TemplateField
 from langflow.template.frontend_node.base import FrontendNode
@@ -13,7 +15,9 @@ from langflow.template.template.base import Template
 
 
 class MemoryFrontendNode(FrontendNode):
-    #! Needs testing
+
+    pinned: bool = True
+
     def add_extra_fields(self) -> None:
         # chat history should have another way to add common field?
         # prevent adding incorect field in ChatMessageHistory
@@ -77,7 +81,9 @@ class MemoryFrontendNode(FrontendNode):
             field.show = True
             field.advanced = False
             field.value = ""
-            field.info = INPUT_KEY_INFO if field.name == "input_key" else OUTPUT_KEY_INFO
+            field.info = (
+                INPUT_KEY_INFO if field.name == "input_key" else OUTPUT_KEY_INFO
+            )
 
         if field.name == "memory_key":
             field.value = "chat_history"