From a05a9202e87c632fe5f061274a3a5a52401e2ae7 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 23 Jun 2023 09:04:35 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(chains.py):=20set=20required?= =?UTF-8?q?=3DTrue=20for=20TemplateField=20'memory'=20=E2=9C=A8=20feat(cha?= =?UTF-8?q?ins.py):=20add=20TemplateField=20'chain=5Ftype'=20to=20support?= =?UTF-8?q?=20different=20types=20of=20QA=20chains=20The=20'memory'=20fiel?= =?UTF-8?q?d=20was=20previously=20set=20to=20required=3DFalse,=20but=20it?= =?UTF-8?q?=20is=20actually=20required=20for=20the=20chain=20to=20function?= =?UTF-8?q?=20properly.=20This=20fix=20sets=20required=3DTrue=20for=20the?= =?UTF-8?q?=20'memory'=20field.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit A new TemplateField 'chain_type' has been added to support different types of QA chains. This field is of type 'str', is required, and is a list of options. It allows the user to select the type of QA chain they want to use. --- .../langflow/template/frontend_node/chains.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) 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: