From 56b16dac60e46ba157b258c68aef3b63f790fb4e Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 14 Jun 2023 17:15:54 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A8=20refactor(documentloaders.py):=20?= =?UTF-8?q?move=20build=5Ftemplate=20function=20outside=20of=20DocumentLoa?= =?UTF-8?q?derFrontNode=20class=20The=20build=5Ftemplate=20function=20is?= =?UTF-8?q?=20now=20outside=20of=20the=20DocumentLoaderFrontNode=20class?= =?UTF-8?q?=20to=20improve=20code=20organization=20and=20make=20it=20more?= =?UTF-8?q?=20modular.=20This=20change=20also=20makes=20it=20easier=20to?= =?UTF-8?q?=20reuse=20the=20function=20in=20other=20parts=20of=20the=20cod?= =?UTF-8?q?ebase.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../template/frontend_node/documentloaders.py | 30 +++++++++---------- 1 file changed, 15 insertions(+), 15 deletions(-) diff --git a/src/backend/langflow/template/frontend_node/documentloaders.py b/src/backend/langflow/template/frontend_node/documentloaders.py index 48c07a188..b8f26a9ca 100644 --- a/src/backend/langflow/template/frontend_node/documentloaders.py +++ b/src/backend/langflow/template/frontend_node/documentloaders.py @@ -2,22 +2,22 @@ from langflow.template.field.base import TemplateField from langflow.template.frontend_node.base import FrontendNode -class DocumentLoaderFrontNode(FrontendNode): - @staticmethod - def build_template( - suffixes: list, fileTypes: list, name: str = "file_path" - ) -> TemplateField: - """Build a template field for a document loader.""" - return TemplateField( - field_type="file", - required=True, - show=True, - name=name, - value="", - suffixes=suffixes, - fileTypes=fileTypes, - ) +def build_template( + suffixes: list, fileTypes: list, name: str = "file_path" +) -> TemplateField: + """Build a template field for a document loader.""" + return TemplateField( + field_type="file", + required=True, + show=True, + name=name, + value="", + suffixes=suffixes, + fileTypes=fileTypes, + ) + +class DocumentLoaderFrontNode(FrontendNode): file_path_templates = { "AirbyteJSONLoader": build_template(suffixes=[".json"], fileTypes=["json"]), "CoNLLULoader": build_template(suffixes=[".csv"], fileTypes=["csv"]),