Skip to content

Commit f71af7c

Browse files
authored
fix: DocumentAddByFileApi miss data_source_type field but there is a mandatory value check (langgenius#12273)
1 parent 5b01eb9 commit f71af7c

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

api/controllers/service_api/dataset/document.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,10 @@ def post(self, tenant_id, dataset_id):
190190
user=current_user,
191191
source="datasets",
192192
)
193-
data_source = {"type": "upload_file", "info_list": {"file_info_list": {"file_ids": [upload_file.id]}}}
193+
data_source = {
194+
"type": "upload_file",
195+
"info_list": {"data_source_type": "upload_file", "file_info_list": {"file_ids": [upload_file.id]}},
196+
}
194197
args["data_source"] = data_source
195198
# validate args
196199
knowledge_config = KnowledgeConfig(**args)
@@ -254,7 +257,10 @@ def post(self, tenant_id, dataset_id, document_id):
254257
raise FileTooLargeError(file_too_large_error.description)
255258
except services.errors.file.UnsupportedFileTypeError:
256259
raise UnsupportedFileTypeError()
257-
data_source = {"type": "upload_file", "info_list": {"file_info_list": {"file_ids": [upload_file.id]}}}
260+
data_source = {
261+
"type": "upload_file",
262+
"info_list": {"data_source_type": "upload_file", "file_info_list": {"file_ids": [upload_file.id]}},
263+
}
258264
args["data_source"] = data_source
259265
# validate args
260266
args["original_document_id"] = str(document_id)

0 commit comments

Comments
 (0)