Merge branch 'db' of personal:logspace-ai/langflow into db

This commit is contained in:
anovazzi1 2023-06-14 17:35:24 -03:00
commit 063fc6cc6b

View file

@ -2,22 +2,22 @@ from langflow.template.field.base import TemplateField
from langflow.template.frontend_node.base import FrontendNode from langflow.template.frontend_node.base import FrontendNode
class DocumentLoaderFrontNode(FrontendNode): def build_template(
@staticmethod suffixes: list, fileTypes: list, name: str = "file_path"
def build_template( ) -> TemplateField:
suffixes: list, fileTypes: list, name: str = "file_path" """Build a template field for a document loader."""
) -> TemplateField: return TemplateField(
"""Build a template field for a document loader.""" field_type="file",
return TemplateField( required=True,
field_type="file", show=True,
required=True, name=name,
show=True, value="",
name=name, suffixes=suffixes,
value="", fileTypes=fileTypes,
suffixes=suffixes, )
fileTypes=fileTypes,
)
class DocumentLoaderFrontNode(FrontendNode):
file_path_templates = { file_path_templates = {
"AirbyteJSONLoader": build_template(suffixes=[".json"], fileTypes=["json"]), "AirbyteJSONLoader": build_template(suffixes=[".json"], fileTypes=["json"]),
"CoNLLULoader": build_template(suffixes=[".csv"], fileTypes=["csv"]), "CoNLLULoader": build_template(suffixes=[".csv"], fileTypes=["csv"]),