fix: keywords (#7357)

api/controllers/service_api/dataset/segment.py

@@ -59,6 +59,7 @@ class SegmentApi(DatasetApiResource):
         parser = reqparse.RequestParser()
         parser.add_argument('segments', type=list, required=False, nullable=True, location='json')
         args = parser.parse_args()
+        if args['segments'] is not None:
             for args_item in args['segments']:
                 SegmentService.segment_create_args_validate(args_item, document)
             segments = SegmentService.multi_create_segment(args['segments'], document, dataset)
@@ -66,6 +67,8 @@ class SegmentApi(DatasetApiResource):
                 'data': marshal(segments, segment_fields),
                 'doc_form': document.doc_form
             }, 200
+        else:
+            return {"error": "Segemtns is required"}, 400
 
     def get(self, tenant_id, dataset_id, document_id):
         """Create single segment."""

api/services/dataset_service.py

@@ -1429,7 +1429,10 @@ class SegmentService:
                 segment_data_list.append(segment_document)
 
                 pre_segment_data_list.append(segment_document)
+                if 'keywords' in segment_item:
                     keywords_list.append(segment_item['keywords'])
+                else:
+                    keywords_list.append(None)
 
             try:
                 # save vector index
@@ -1482,7 +1485,7 @@ class SegmentService:
                 db.session.add(segment)
                 db.session.commit()
                 # update segment index task
-                if args['keywords']:
+                if 'keywords' in args:
                     keyword = Keyword(dataset)
                     keyword.delete_by_ids([segment.index_node_id])
                     document = RAGDocument(