feat: adds NotionDirectoryLoader

2023-05-28 02:05:32 -03:00 · 2023-05-28 02:05:32 -03:00 · 209e31da26
commit 209e31da26
parent a758f48b88
3 changed files with 100 additions and 1 deletions
--- a/src/backend/langflow/config.yaml
+++ b/src/backend/langflow/config.yaml
@ -42,6 +42,7 @@ documentloaders:
  - IMSDbLoader
  - GitbookLoader
  - ReadTheDocsLoader
+  - NotionDirectoryLoader
 embeddings:
  - OpenAIEmbeddings
  - HuggingFaceEmbeddings
--- a/src/backend/langflow/interface/document_loaders/base.py
+++ b/src/backend/langflow/interface/document_loaders/base.py
@ -118,7 +118,7 @@ class DocumentLoaderCreator(LangChainTypeCreator):
                    "value": "",
                    "display_name": "Web Page",
                }
-            elif name in {"ReadTheDocsLoader"}:
+            elif name in {"ReadTheDocsLoader", "NotionDirectoryLoader"}:
                signature["template"]["path"] = {
                    "type": "str",
                    "required": True,