From 1dcdc02e4f43d24af19fc4788e96b43d6baaf816 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Wed, 9 Aug 2023 21:49:20 -0300
Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=80=20feat(constants.py):=20import=20B?=
 =?UTF-8?q?aseChatMemory=20from=20langchain.memory.chat=5Fmemory=20module?=
 =?UTF-8?q?=20to=20add=20support=20for=20chat=20memory=20in=20custom=20int?=
 =?UTF-8?q?erfaces?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/interface/custom/constants.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/backend/langflow/interface/custom/constants.py b/src/backend/langflow/interface/custom/constants.py
index 891fc75f9..89d187bb8 100644
--- a/src/backend/langflow/interface/custom/constants.py
+++ b/src/backend/langflow/interface/custom/constants.py
@@ -9,7 +9,7 @@ from langchain.tools import Tool
 from langchain.vectorstores.base import VectorStore
 from langchain.schema import BaseOutputParser
 from langchain.schema.memory import BaseMemory
-
+from langchain.memory.chat_memory import BaseChatMemory
 
 LANGCHAIN_BASE_TYPES = {
     "Chain": Chain,
@@ -24,6 +24,7 @@ LANGCHAIN_BASE_TYPES = {
     "BaseRetriever": BaseRetriever,
     "BaseOutputParser": BaseOutputParser,
     "BaseMemory": BaseMemory,
+    "BaseChatMemory": BaseChatMemory,
 }
 
 # Langchain base types plus Python base types