Merge branch 'feat/support-parent-child-chunk' into deploy/dev

langgenius · Dec 6, 2024 · f33c02b · f33c02b
2 parents 6a148bb + e205e8f
commit f33c02b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/api/services/dataset_service.py b/api/services/dataset_service.py
@@ -1121,14 +1121,14 @@ def save_document_without_dataset_id(tenant_id: str, knowledge_config: Knowledge
         if features.billing.enabled:
             count = 0
             if knowledge_config.data_source.info_list.data_source_type == "upload_file":
-                upload_file_list = knowledge_config.data_source.file_info_list.file_ids
+                upload_file_list = knowledge_config.data_source.info_list.file_info_list.file_ids
                 count = len(upload_file_list)
             elif knowledge_config.data_source.info_list.data_source_type == "notion_import":
-                notion_info_list = knowledge_config.data_source.notion_info_list
+                notion_info_list = knowledge_config.data_source.info_list.notion_info_list
                 for notion_info in notion_info_list:
                     count = count + len(notion_info["pages"])
             elif knowledge_config.data_source.info_list.data_source_type == "website_crawl":
-                website_info = knowledge_config.data_source.website_info_list
+                website_info = knowledge_config.data_source.info_list.website_info_list
                 count = len(website_info.urls)
             batch_upload_limit = int(dify_config.BATCH_UPLOAD_LIMIT)
             if count > batch_upload_limit: