소스 검색

fix: dataset segements api (#2766)

crazywoola 1 년 전
부모
커밋
3c91f9b5ab
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      api/controllers/service_api/dataset/segment.py

+ 2 - 2
api/controllers/service_api/dataset/segment.py

@@ -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)
-        segment = SegmentService.update_segment(args['segments'], segment, document, dataset)
+        SegmentService.segment_create_args_validate(args, document)
+        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