fix: does not save segment vector when there is no attachment_ids

This commit is contained in:
jyong
2025-12-15 13:48:35 +08:00
parent 2058186f22
commit bf14d04bb1

View File

@@ -2817,20 +2817,20 @@ class SegmentService:
db.session.add(binding) db.session.add(binding)
db.session.commit() db.session.commit()
# save vector index # save vector index
try: try:
VectorService.create_segments_vector( VectorService.create_segments_vector(
[args["keywords"]], [segment_document], dataset, document.doc_form [args["keywords"]], [segment_document], dataset, document.doc_form
) )
except Exception as e: except Exception as e:
logger.exception("create segment index failed") logger.exception("create segment index failed")
segment_document.enabled = False segment_document.enabled = False
segment_document.disabled_at = naive_utc_now() segment_document.disabled_at = naive_utc_now()
segment_document.status = "error" segment_document.status = "error"
segment_document.error = str(e) segment_document.error = str(e)
db.session.commit() db.session.commit()
segment = db.session.query(DocumentSegment).where(DocumentSegment.id == segment_document.id).first() segment = db.session.query(DocumentSegment).where(DocumentSegment.id == segment_document.id).first()
return segment return segment
except LockNotOwnedError: except LockNotOwnedError:
pass pass