highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-03-26 17:51:54 +01:00
Code Issues Packages Projects Releases Wiki Activity

Notion connector test separately (#372)

Browse Source
This commit is contained in:
Yuhong Sun 2023-08-31 18:18:19 -07:00 committed by GitHub
parent 43efa9da94
commit 6bae93ad3c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
backend/danswer/connectors/notion/connector.py
View File

@ -1,4 +1,3 @@
"""Notion reader."""
import time
from dataclasses import dataclass
from dataclasses import fields
@ -231,3 +230,14 @@ class NotionConnector(LoadConnector, PollConnector):
query_dict["start_cursor"] = db_res.next_cursor
else:
break
if __name__ == "__main__":
import os
connector = NotionConnector()
connector.load_credentials(
{"notion_integration_token": os.environ.get("NOTION_INTEGRATION_TOKEN")}
)
document_batches = connector.load_from_state()
print(next(document_batches))