Skip to content

Commit cdf8f24

Browse files
committed
fix wrong assert
1 parent dc7ce12 commit cdf8f24

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

api/services/dataset_service.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)