diff --git a/api/core/rag/extractor/notion_extractor.py b/api/core/rag/extractor/notion_extractor.py index 38dd36361a..c40064fd1d 100644 --- a/api/core/rag/extractor/notion_extractor.py +++ b/api/core/rag/extractor/notion_extractor.py @@ -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')