From 3845a60bcfda2684fac62cab88b2e2e6f7fd442e Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 29 Jun 2023 08:52:45 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=80=20feat(memories.py):=20add=20defau?= =?UTF-8?q?lt=20value=20for=20connection=5Fstring=20field=20in=20PostgresC?= =?UTF-8?q?hatMessageHistoryFrontendNode=20The=20trailing=20whitespace=20h?= =?UTF-8?q?as=20been=20removed=20for=20consistency=20and=20readability.=20?= =?UTF-8?q?The=20unused=20import=20of=20`DEFAULT=5FCONNECTION=5FSTRING`=20?= =?UTF-8?q?from=20`langchain.memory.chat=5Fmessage=5Fhistories.postgres`?= =?UTF-8?q?=20has=20been=20removed.=20The=20`PostgresChatMessageHistoryFro?= =?UTF-8?q?ntendNode`=20now=20has=20a=20default=20value=20for=20the=20`con?= =?UTF-8?q?nection=5Fstring`=20field,=20which=20is=20set=20to=20`DEFAULT?= =?UTF-8?q?=5FCONNECTION=5FSTRING`.=20This=20ensures=20that=20the=20field?= =?UTF-8?q?=20has=20a=20default=20value=20when=20creating=20instances=20of?= =?UTF-8?q?=20the=20class.=20=F0=9F=94=A7=20chore(memories.py):=20remove?= =?UTF-8?q?=20trailing=20whitespace=20and=20unused=20import?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/template/frontend_node/memories.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/src/backend/langflow/template/frontend_node/memories.py b/src/backend/langflow/template/frontend_node/memories.py index 7064eaa05..936bdb8f4 100644 --- a/src/backend/langflow/template/frontend_node/memories.py +++ b/src/backend/langflow/template/frontend_node/memories.py @@ -3,6 +3,7 @@ from typing import Optional from langflow.template.field.base import TemplateField from langflow.template.frontend_node.base import FrontendNode from langflow.template.template.base import Template +from langchain.memory.chat_message_histories.postgres import DEFAULT_CONNECTION_STRING class MemoryFrontendNode(FrontendNode): @@ -10,7 +11,7 @@ class MemoryFrontendNode(FrontendNode): def add_extra_fields(self) -> None: # chat history should have another way to add common field? # prevent adding incorect field in ChatMessageHistory - if "BaseChatMessageHistory" in self.base_classes: + if "BaseChatMessageHistory" in self.base_classes: return # add return_messages field @@ -95,6 +96,7 @@ class PostgresChatMessageHistoryFrontendNode(MemoryFrontendNode): required=True, show=True, name="connection_string", + value=DEFAULT_CONNECTION_STRING, ), TemplateField( field_type="str", @@ -109,7 +111,4 @@ class PostgresChatMessageHistoryFrontendNode(MemoryFrontendNode): ], ) description: str = "Memory store with Postgres" - base_classes: list[str] = [ - "PostgresChatMessageHistory", - "BaseChatMessageHistory" - ] \ No newline at end of file + base_classes: list[str] = ["PostgresChatMessageHistory", "BaseChatMessageHistory"]