fix: dataset segements api (#2766)

This commit is contained in:
crazywoola 2024-03-11 09:26:15 +08:00 committed by GitHub
parent f073dca22a
commit 3c91f9b5ab
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -200,8 +200,8 @@ class DatasetSegmentApi(DatasetApiResource):
parser.add_argument('segments', type=dict, required=False, nullable=True, location='json') parser.add_argument('segments', type=dict, required=False, nullable=True, location='json')
args = parser.parse_args() args = parser.parse_args()
SegmentService.segment_create_args_validate(args['segments'], document) SegmentService.segment_create_args_validate(args, document)
segment = SegmentService.update_segment(args['segments'], segment, document, dataset) segment = SegmentService.update_segment(args, segment, document, dataset)
return { return {
'data': marshal(segment, segment_fields), 'data': marshal(segment, segment_fields),
'doc_form': document.doc_form 'doc_form': document.doc_form