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"