fix notion import limit check

api/services/dataset_service.py

@@ -402,8 +402,9 @@ class DocumentService:
                 upload_file_list = document_data["data_source"]["info_list"]['file_info_list']['file_ids']
                 count = len(upload_file_list)
             elif document_data["data_source"]["type"] == "notion_import":
-                notion_page_list = document_data["data_source"]['info_list']['notion_info_list']['pages']
-                count = len(notion_page_list)
+                notion_info_list = document_data["data_source"]['info_list']['notion_info_list']
+                for notion_info in notion_info_list:
+                    count = count + len(notion_info['pages'])
             documents_count = DocumentService.get_tenant_documents_count()
             total_count = documents_count + 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']
             count = len(upload_file_list)
         elif document_data["data_source"]["type"] == "notion_import":
-            notion_page_list = document_data["data_source"]['info_list']['notion_info_list']['pages']
-            count = len(notion_page_list)
+            notion_info_list = document_data["data_source"]['info_list']['notion_info_list']
+            for notion_info in notion_info_list:
+                count = count + len(notion_info['pages'])
         # check document limit
         if current_app.config['EDITION'] == 'CLOUD':
             documents_count = DocumentService.get_tenant_documents_count()