fix: segment keywords bug (#17599)

Co-authored-by: huangzhuo <huangzhuo1@xiaomi.com>
1 year ago · 07ed728605
parent d796fcc0e7
commit 07ed728605
1 changed files with 2 additions and 6 deletions
--- a/api/services/dataset_service.py
+++ b/api/services/dataset_service.py
@ -1786,12 +1786,8 @@ class SegmentService:
                    )
                elif document.doc_form in (IndexType.PARAGRAPH_INDEX, IndexType.QA_INDEX):
                    if args.enabled or keyword_changed:
-                        VectorService.create_segments_vector(
-                            [args.keywords] if args.keywords else None,
-                            [segment],
-                            dataset,
-                            document.doc_form,
-                        )
+                        # update segment vector index
+                        VectorService.update_segment_vector(args.keywords, segment, dataset)
            else:
                segment_hash = helper.generate_text_hash(content)
                tokens = 0