Skip to content

D
dify
Unverified Commit 16199e96 authored by Jyong's avatar Jyong Committed by GitHub
Browse files
Options

fix notion import limit check (#1042) 

Co-authored-by: 's avatarjyong <jyong@dify.ai>
parent 02452421
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 4 deletions
api/services/dataset_service.py
View file @ 16199e96
...@@ -402,8 +402,9 @@ class DocumentService: ...@@ -402,8 +402,9 @@ class DocumentService:
upload_file_list = document_data["data_source"]["info_list"]['file_info_list']['file_ids'] upload_file_list = document_data["data_source"]["info_list"]['file_info_list']['file_ids']
count = len(upload_file_list) count = len(upload_file_list)
elif document_data["data_source"]["type"] == "notion_import": elif document_data["data_source"]["type"] == "notion_import":
notion_page_list = document_data["data_source"]['info_list']['notion_info_list']['pages'] notion_info_list = document_data["data_source"]['info_list']['notion_info_list']
count = len(notion_page_list) for notion_info in notion_info_list:
count = count + len(notion_info['pages'])
documents_count = DocumentService.get_tenant_documents_count() documents_count = DocumentService.get_tenant_documents_count()
total_count = documents_count + count total_count = documents_count + count
tenant_document_count = int(current_app.config['TENANT_DOCUMENT_COUNT']) tenant_document_count = int(current_app.config['TENANT_DOCUMENT_COUNT'])
...@@ -663,8 +664,9 @@ class DocumentService: ...@@ -663,8 +664,9 @@ class DocumentService:
upload_file_list = document_data["data_source"]["info_list"]['file_info_list']['file_ids'] upload_file_list = document_data["data_source"]["info_list"]['file_info_list']['file_ids']
count = len(upload_file_list) count = len(upload_file_list)
elif document_data["data_source"]["type"] == "notion_import": elif document_data["data_source"]["type"] == "notion_import":
notion_page_list = document_data["data_source"]['info_list']['notion_info_list']['pages'] notion_info_list = document_data["data_source"]['info_list']['notion_info_list']
count = len(notion_page_list) for notion_info in notion_info_list:
count = count + len(notion_info['pages'])
# check document limit # check document limit
if current_app.config['EDITION'] == 'CLOUD': if current_app.config['EDITION'] == 'CLOUD':
documents_count = DocumentService.get_tenant_documents_count() documents_count = DocumentService.get_tenant_documents_count()
......
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