mirror of https://github.com/langgenius/dify.git
fix: does not save segment vector when there is no attachment_ids (#29520)
Co-authored-by: sai <> Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
parent
323e0c4d30
commit
1e47ffb50c
|
|
@ -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(
|
keywords = args.get("keywords")
|
||||||
[args["keywords"]], [segment_document], dataset, document.doc_form
|
keywords_list = [keywords] if keywords is not None else None
|
||||||
)
|
VectorService.create_segments_vector(keywords_list, [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
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue