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

Update CoNLLULoaderComponent configuration

Browse source
This commit is contained in:
carlosrcoelho 2024-01-09 19:38:16 -03:00
commit cbacf55955
1 changed files with 9 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

18
src/backend/langflow/components/documentloaders/CoNLLULoader.py
View file

@ -1,5 +1,5 @@
from langflow import CustomComponent from langflow import CustomComponent
from langchain.documents import Document from langchain.docstore.document import Document
from typing import Optional, Dict from typing import Optional, Dict
from langflow.field_typing import TemplateField from langflow.field_typing import TemplateField
@ -11,13 +11,13 @@ class CoNLLULoaderComponent(CustomComponent):
def build_config(self): def build_config(self):
return { return {
"file_path": TemplateField( "file_path": {
display_name="File Path", "display_name": "File Path",
required=True, "required": True,
type="file", "suffixes": [".conllu"],
file_types=["conllu"], "file_types": ["conllu"],
suffixes=[".conllu"], "field_type": "file",
), },
"metadata": TemplateField( "metadata": TemplateField(
display_name="Metadata", display_name="Metadata",
required=False, required=False,
@ -25,7 +25,7 @@ class CoNLLULoaderComponent(CustomComponent):
), ),
} }
def build(self, file_path: str, metadata: Optional[Dict[str, str]] = None) -> Document: def build(self, file_path: str, metadata: dict) -> Document:
# Here, you would use the actual class that loads CoNLL-U files. # Here, you would use the actual class that loads CoNLL-U files.
# As I don't have the specific class, I'm returning an instance of Document. # As I don't have the specific class, I'm returning an instance of Document.
# In a real scenario, you should replace the below Document with the actual loader class. # In a real scenario, you should replace the below Document with the actual loader class.