From 4761eda9a046a45f8dbabfb2e430f8e6edfc3448 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 29 Jun 2023 18:38:08 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=80=20chore(base.py):=20add=20'SQLiteE?= =?UTF-8?q?ntityStore'=20to=20'from=5Fmethod=5Fnodes'=20dictionary=20The?= =?UTF-8?q?=20'from=5Fmethod=5Fnodes'=20dictionary=20in=20the=20'MemoryCre?= =?UTF-8?q?ator'=20class=20in=20'base.py'=20has=20been=20updated=20to=20in?= =?UTF-8?q?clude=20the=20'SQLiteEntityStore'=20class=20and=20its=20'=5F=5F?= =?UTF-8?q?init=5F=5F'=20method.=20This=20change=20allows=20the=20'SQLiteE?= =?UTF-8?q?ntityStore'=20class=20to=20be=20used=20as=20a=20memory=20node?= =?UTF-8?q?=20in=20the=20LangFlow=20application.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/interface/memories/base.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/interface/memories/base.py b/src/backend/langflow/interface/memories/base.py index 4d7f8d5aa..0f97a02fe 100644 --- a/src/backend/langflow/interface/memories/base.py +++ b/src/backend/langflow/interface/memories/base.py @@ -13,7 +13,10 @@ from langflow.custom.customs import get_custom_nodes class MemoryCreator(LangChainTypeCreator): type_name: str = "memories" - from_method_nodes = {"ZepChatMessageHistory": "__init__"} + from_method_nodes = { + "ZepChatMessageHistory": "__init__", + "SQLiteEntityStore": "__init__", + } @property def frontend_node_class(self) -> Type[FrontendNode]: