AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-13 21:56:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix notion import limit check (#1042)

Browse Source 
Co-authored-by: jyong <jyong@dify.ai>
This commit is contained in:
Jyong 2023-08-28 16:49:03 +08:00 committed by GitHub
parent 02452421d5
commit 16199e968e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/services/dataset_service.py
View File

@ -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:
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()