| @@ -4,7 +4,6 @@ from typing import Any, Optional | |||
| import requests | |||
| from flask import current_app | |||
| from flask_login import current_user | |||
| from core.rag.extractor.extractor_base import BaseExtractor | |||
| from core.rag.models.document import Document | |||
| @@ -43,7 +42,7 @@ class NotionExtractor(BaseExtractor): | |||
| if notion_access_token: | |||
| self._notion_access_token = notion_access_token | |||
| 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) | |||
| if not self._notion_access_token: | |||
| integration_token = current_app.config.get('NOTION_INTEGRATION_TOKEN') | |||