fix: dataset segements api (#2766)

langgenius · Mar 11, 2024 · 3c91f9b · 3c91f9b
1 parent f073dca
commit 3c91f9b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/controllers/service_api/dataset/segment.py b/api/controllers/service_api/dataset/segment.py
@@ -200,8 +200,8 @@ def post(self, tenant_id, dataset_id, document_id, segment_id):
         parser.add_argument('segments', type=dict, required=False, nullable=True, location='json')
         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 {
             'data': marshal(segment, segment_fields),
             'doc_form': document.doc_form