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

Fix: Restore Notion and Slack directory loaders. (#476)

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-06-21 13:57:00 +00:00 • committed by GitHub
commit fef8324886
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -41,6 +41,7 @@ class DocumentLoaderFrontNode(FrontendNode):
"UnstructuredPowerPointLoader": build_template( "UnstructuredPowerPointLoader": build_template(
suffixes=[".pptx", ".ppt"], fileTypes=["pptx", "ppt"] suffixes=[".pptx", ".ppt"], fileTypes=["pptx", "ppt"]
), ),
"SlackDirectoryLoader": build_template(suffixes=[".zip"], fileTypes=["zip"]),
"SRTLoader": build_template(suffixes=[".srt"], fileTypes=["srt"]), "SRTLoader": build_template(suffixes=[".srt"], fileTypes=["srt"]),
"TelegramChatLoader": build_template(suffixes=[".json"], fileTypes=["json"]), "TelegramChatLoader": build_template(suffixes=[".json"], fileTypes=["json"]),
"TextLoader": build_template(suffixes=[".txt"], fileTypes=["txt"]), "TextLoader": build_template(suffixes=[".txt"], fileTypes=["txt"]),
@ -51,6 +52,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 +66,12 @@ 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 {
"NotionDirectoryLoader",
"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 +80,6 @@ class DocumentLoaderFrontNode(FrontendNode):
show=True, show=True,
name=name, name=name,
value="", value="",
display_name="Web Page", display_name=display_name,
) )
) )