From 1e61500f3834ccb210fb1c0f50ba10e9c4ee32cd Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 22 Jun 2023 13:48:44 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=80=20feat(chains.py):=20add=20return?= =?UTF-8?q?=5Fsource=5Fdocuments=20field=20to=20template=20The=20`return?= =?UTF-8?q?=5Fsource=5Fdocuments`=20field=20is=20added=20to=20the=20templa?= =?UTF-8?q?te=20as=20a=20boolean=20field.=20It=20is=20not=20a=20required?= =?UTF-8?q?=20field=20and=20is=20set=20to=20`True`=20by=20default.=20This?= =?UTF-8?q?=20field=20is=20used=20to=20determine=20whether=20the=20source?= =?UTF-8?q?=20documents=20should=20be=20returned=20or=20not.=20It=20is=20d?= =?UTF-8?q?isplayed=20in=20the=20frontend=20and=20can=20be=20toggled=20by?= =?UTF-8?q?=20the=20user.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../langflow/template/frontend_node/chains.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/src/backend/langflow/template/frontend_node/chains.py b/src/backend/langflow/template/frontend_node/chains.py index 1a1753df3..19ea2e0df 100644 --- a/src/backend/langflow/template/frontend_node/chains.py +++ b/src/backend/langflow/template/frontend_node/chains.py @@ -19,6 +19,18 @@ class ChainFrontendNode(FrontendNode): advanced=False, ) ) + # add return_source_documents + self.template.add_field( + TemplateField( + field_type="bool", + required=False, + show=True, + name="return_source_documents", + advanced=False, + value=True, + display_name="Return source documents", + ) + ) @staticmethod def format_field(field: TemplateField, name: Optional[str] = None) -> None: @@ -59,6 +71,12 @@ class ChainFrontendNode(FrontendNode): field.show = True field.advanced = False + if field.name == "return_source_documents": + field.required = False + field.show = True + field.advanced = True + field.value = True + class SeriesCharacterChainNode(FrontendNode): name: str = "SeriesCharacterChain"