|
@@ -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')
|