Update DirectoryLoaderComponent configuration

This commit is contained in:
carlosrcoelho 2024-01-09 19:23:02 -03:00
commit 7ecb0b6ec4

View file

@ -1,6 +1,6 @@
from langflow import CustomComponent from langflow import CustomComponent
from langchain.data_connections import Document from langchain.docstore.document import Document
from typing import Optional, Dict, Any from typing import Optional, Dict, Any
class DirectoryLoaderComponent(CustomComponent): class DirectoryLoaderComponent(CustomComponent):
@ -9,14 +9,14 @@ class DirectoryLoaderComponent(CustomComponent):
def build_config(self) -> Dict[str, Any]: def build_config(self) -> Dict[str, Any]:
return { return {
"glob": {"display_name": "Glob Pattern", "default": "**/*.txt"}, "glob": {"display_name": "Glob Pattern", "value": "**/*.txt"},
"load_hidden": {"display_name": "Load Hidden Files", "default": False, "advanced": True}, "load_hidden": {"display_name": "Load Hidden Files", "value": False, "advanced": True},
"max_concurrency": {"display_name": "Max Concurrency", "default": 10, "advanced": True}, "max_concurrency": {"display_name": "Max Concurrency", "value": 10, "advanced": True},
"metadata": {"display_name": "Metadata", "default": {}}, "metadata": {"display_name": "Metadata", "value": {}},
"path": {"display_name": "Local Directory"}, "path": {"display_name": "Local Directory"},
"recursive": {"display_name": "Recursive", "default": True, "advanced": True}, "recursive": {"display_name": "Recursive", "value": True, "advanced": True},
"silent_errors": {"display_name": "Silent Errors", "default": False, "advanced": True}, "silent_errors": {"display_name": "Silent Errors", "value": False, "advanced": True},
"use_multithreading": {"display_name": "Use Multithreading", "default": True, "advanced": True}, "use_multithreading": {"display_name": "Use Multithreading", "value": True, "advanced": True},
} }
def build( def build(
@ -25,7 +25,7 @@ class DirectoryLoaderComponent(CustomComponent):
path: str, path: str,
load_hidden: Optional[bool] = False, load_hidden: Optional[bool] = False,
max_concurrency: Optional[int] = 10, max_concurrency: Optional[int] = 10,
metadata: Optional[Dict[str, Any]] = None, metadata: Optional[dict] = {},
recursive: Optional[bool] = True, recursive: Optional[bool] = True,
silent_errors: Optional[bool] = False, silent_errors: Optional[bool] = False,
use_multithreading: Optional[bool] = True, use_multithreading: Optional[bool] = True,