| import requests | import requests | ||||
| from flask import current_app | from flask import current_app | ||||
| from flask_login import current_user | |||||
| from core.rag.extractor.extractor_base import BaseExtractor | from core.rag.extractor.extractor_base import BaseExtractor | ||||
| from core.rag.models.document import Document | from core.rag.models.document import Document | ||||
| if notion_access_token: | if notion_access_token: | ||||
| self._notion_access_token = notion_access_token | self._notion_access_token = notion_access_token | ||||
| else: | else: | ||||
| self._notion_access_token = self._get_access_token(current_user.current_tenant_id, | |||||
| self._notion_access_token = self._get_access_token(tenant_id, | |||||
| self._notion_workspace_id) | self._notion_workspace_id) | ||||
| if not self._notion_access_token: | if not self._notion_access_token: | ||||
| integration_token = current_app.config.get('NOTION_INTEGRATION_TOKEN') | integration_token = current_app.config.get('NOTION_INTEGRATION_TOKEN') |