Unverified Commit a4d86496 authored by takatost's avatar takatost Committed by GitHub

fix: notion extractor raise 'NoneType' object has no attribute 'curre… (#2608)

parent 90bdc85f
...@@ -4,7 +4,6 @@ from typing import Any, Optional ...@@ -4,7 +4,6 @@ from typing import Any, Optional
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
...@@ -43,7 +42,7 @@ class NotionExtractor(BaseExtractor): ...@@ -43,7 +42,7 @@ class NotionExtractor(BaseExtractor):
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')
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment