From e04eb0fd7e68b9bedd980c0aee03a1ac6d253c45 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 1 Jul 2023 09:43:17 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(memories.py):=20add=20condit?= =?UTF-8?q?ion=20to=20only=20add=20output=5Fkey=20field=20if=20template=20?= =?UTF-8?q?type=20is=20not=20VectorStoreRetrieverMemory=20The=20output=5Fk?= =?UTF-8?q?ey=20field=20is=20now=20only=20added=20to=20the=20template=20if?= =?UTF-8?q?=20the=20template=20type=20is=20not=20VectorStoreRetrieverMemor?= =?UTF-8?q?y.=20This=20ensures=20that=20the=20output=5Fkey=20field=20is=20?= =?UTF-8?q?not=20added=20unnecessarily=20for=20templates=20of=20this=20typ?= =?UTF-8?q?e.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../template/frontend_node/memories.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/src/backend/langflow/template/frontend_node/memories.py b/src/backend/langflow/template/frontend_node/memories.py index 37177c72e..6d490212f 100644 --- a/src/backend/langflow/template/frontend_node/memories.py +++ b/src/backend/langflow/template/frontend_node/memories.py @@ -37,16 +37,17 @@ class MemoryFrontendNode(FrontendNode): value="", ) ) - self.template.add_field( - TemplateField( - field_type="str", - required=False, - show=True, - name="output_key", - advanced=True, - value="", + if self.template.type_name not in {"VectorStoreRetrieverMemory"}: + self.template.add_field( + TemplateField( + field_type="str", + required=False, + show=True, + name="output_key", + advanced=True, + value="", + ) ) - ) @staticmethod def format_field(field: TemplateField, name: Optional[str] = None) -> None: