diff --git a/src/backend/langflow/template/frontend_node/chains.py b/src/backend/langflow/template/frontend_node/chains.py index 19ea2e0df..0ed8f1389 100644 --- a/src/backend/langflow/template/frontend_node/chains.py +++ b/src/backend/langflow/template/frontend_node/chains.py @@ -13,7 +13,7 @@ class ChainFrontendNode(FrontendNode): self.template.add_field( TemplateField( field_type="BaseChatMemory", - required=False, + required=True, show=True, name="memory", advanced=False, @@ -31,6 +31,19 @@ class ChainFrontendNode(FrontendNode): display_name="Return source documents", ) ) + self.template.add_field( + TemplateField( + field_type="str", + required=True, + is_list=True, + show=True, + multiline=False, + options=QA_CHAIN_TYPES, + value=QA_CHAIN_TYPES[0], + name="chain_type", + advanced=False, + ) + ) @staticmethod def format_field(field: TemplateField, name: Optional[str] = None) -> None: