| @@ -190,7 +190,10 @@ class DocumentAddByFileApi(DatasetApiResource): | |||
| user=current_user, | |||
| source="datasets", | |||
| ) | |||
| data_source = {"type": "upload_file", "info_list": {"file_info_list": {"file_ids": [upload_file.id]}}} | |||
| data_source = { | |||
| "type": "upload_file", | |||
| "info_list": {"data_source_type": "upload_file", "file_info_list": {"file_ids": [upload_file.id]}}, | |||
| } | |||
| args["data_source"] = data_source | |||
| # validate args | |||
| knowledge_config = KnowledgeConfig(**args) | |||
| @@ -254,7 +257,10 @@ class DocumentUpdateByFileApi(DatasetApiResource): | |||
| raise FileTooLargeError(file_too_large_error.description) | |||
| except services.errors.file.UnsupportedFileTypeError: | |||
| raise UnsupportedFileTypeError() | |||
| data_source = {"type": "upload_file", "info_list": {"file_info_list": {"file_ids": [upload_file.id]}}} | |||
| data_source = { | |||
| "type": "upload_file", | |||
| "info_list": {"data_source_type": "upload_file", "file_info_list": {"file_ids": [upload_file.id]}}, | |||
| } | |||
| args["data_source"] = data_source | |||
| # validate args | |||
| args["original_document_id"] = str(document_id) | |||