feat: add support for DirectoryLoader
This commit is contained in:
parent
272be9f566
commit
372463912c
2 changed files with 16 additions and 2 deletions
|
|
@ -48,6 +48,7 @@ documentloaders:
|
||||||
- ReadTheDocsLoader
|
- ReadTheDocsLoader
|
||||||
- SlackDirectoryLoader
|
- SlackDirectoryLoader
|
||||||
- NotionDirectoryLoader
|
- NotionDirectoryLoader
|
||||||
|
- DirectoryLoader
|
||||||
embeddings:
|
embeddings:
|
||||||
- OpenAIEmbeddings
|
- OpenAIEmbeddings
|
||||||
- HuggingFaceEmbeddings
|
- HuggingFaceEmbeddings
|
||||||
|
|
|
||||||
|
|
@ -51,6 +51,7 @@ class DocumentLoaderFrontNode(FrontendNode):
|
||||||
|
|
||||||
def add_extra_fields(self) -> None:
|
def add_extra_fields(self) -> None:
|
||||||
name = None
|
name = None
|
||||||
|
display_name = "Web Page"
|
||||||
if self.template.type_name in self.file_path_templates:
|
if self.template.type_name in self.file_path_templates:
|
||||||
self.template.add_field(self.file_path_templates[self.template.type_name])
|
self.template.add_field(self.file_path_templates[self.template.type_name])
|
||||||
elif self.template.type_name in {
|
elif self.template.type_name in {
|
||||||
|
|
@ -64,8 +65,9 @@ class DocumentLoaderFrontNode(FrontendNode):
|
||||||
name = "web_path"
|
name = "web_path"
|
||||||
elif self.template.type_name in {"GitbookLoader"}:
|
elif self.template.type_name in {"GitbookLoader"}:
|
||||||
name = "web_page"
|
name = "web_page"
|
||||||
elif self.template.type_name in {"ReadTheDocsLoader"}:
|
elif self.template.type_name in {"DirectoryLoader", "ReadTheDocsLoader"}:
|
||||||
name = "path"
|
name = "path"
|
||||||
|
display_name = "Local directory"
|
||||||
if name:
|
if name:
|
||||||
self.template.add_field(
|
self.template.add_field(
|
||||||
TemplateField(
|
TemplateField(
|
||||||
|
|
@ -74,6 +76,17 @@ class DocumentLoaderFrontNode(FrontendNode):
|
||||||
show=True,
|
show=True,
|
||||||
name=name,
|
name=name,
|
||||||
value="",
|
value="",
|
||||||
display_name="Web Page",
|
display_name=display_name,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
if self.template.type_name in {"DirectoryLoader"}:
|
||||||
|
self.template.add_field(
|
||||||
|
TemplateField(
|
||||||
|
field_type="str",
|
||||||
|
required=True,
|
||||||
|
show=True,
|
||||||
|
name="glob",
|
||||||
|
value="**/*.txt",
|
||||||
|
display_name="glob",
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue