@@ -1417,7 +1417,7 @@ def save_document_with_dataset_id(
14171417 assert isinstance (current_user , Account )
14181418 assert current_user .current_tenant_id is not None
14191419 assert knowledge_config .data_source
1420- assert knowledge_config .data_source .info_list . file_info_list
1420+ assert knowledge_config .data_source .info_list
14211421
14221422 features = FeatureService .get_features (current_user .current_tenant_id )
14231423
@@ -1426,6 +1426,8 @@ def save_document_with_dataset_id(
14261426 count = 0
14271427 if knowledge_config .data_source :
14281428 if knowledge_config .data_source .info_list .data_source_type == "upload_file" :
1429+ if not knowledge_config .data_source .info_list .file_info_list :
1430+ raise ValueError ("File source info is required" )
14291431 upload_file_list = knowledge_config .data_source .info_list .file_info_list .file_ids
14301432 count = len (upload_file_list )
14311433 elif knowledge_config .data_source .info_list .data_source_type == "notion_import" :
@@ -1531,6 +1533,8 @@ def save_document_with_dataset_id(
15311533 document_ids = []
15321534 duplicate_document_ids = []
15331535 if knowledge_config .data_source .info_list .data_source_type == "upload_file" :
1536+ if not knowledge_config .data_source .info_list .file_info_list :
1537+ raise ValueError ("File source info is required" )
15341538 upload_file_list = knowledge_config .data_source .info_list .file_info_list .file_ids
15351539 for file_id in upload_file_list :
15361540 file = (
0 commit comments