jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
anovazzi1 2023-06-14 17:35:24 -03:00
commit 063fc6cc6b
1 changed files with 15 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/langflow/template/frontend_node/documentloaders.py
View file

@ -2,11 +2,9 @@ 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
def build_template(
suffixes: list, fileTypes: list, name: str = "file_path" suffixes: list, fileTypes: list, name: str = "file_path"
) -> TemplateField: ) -> TemplateField:
"""Build a template field for a document loader.""" """Build a template field for a document loader."""
return TemplateField( return TemplateField(
field_type="file", field_type="file",
@ -18,6 +16,8 @@ class DocumentLoaderFrontNode(FrontendNode):
fileTypes=fileTypes, 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"]),