estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix dataset_id and index_node_id idx missed in document_segments tabl… (#7681)

Browse Source
This commit is contained in:
Jyong
2024-08-27 10:25:24 +08:00
committed by GitHub
parent 7b7576ad55
commit 35431bce0d
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/indexing_runner.py
View File

@@ -720,6 +720,7 @@ class IndexingRunner:
document_ids = [document.metadata['doc_id'] for document in documents] document_ids = [document.metadata['doc_id'] for document in documents]
db.session.query(DocumentSegment).filter( db.session.query(DocumentSegment).filter(
DocumentSegment.document_id == document_id, DocumentSegment.document_id == document_id,
DocumentSegment.dataset_id == dataset_id,
DocumentSegment.index_node_id.in_(document_ids), DocumentSegment.index_node_id.in_(document_ids),
DocumentSegment.status == "indexing" DocumentSegment.status == "indexing"
).update({ ).update({
@@ -751,6 +752,7 @@ class IndexingRunner:
document_ids = [document.metadata['doc_id'] for document in chunk_documents] document_ids = [document.metadata['doc_id'] for document in chunk_documents]
db.session.query(DocumentSegment).filter( db.session.query(DocumentSegment).filter(
DocumentSegment.document_id == dataset_document.id, DocumentSegment.document_id == dataset_document.id,
DocumentSegment.dataset_id == dataset.id,
DocumentSegment.index_node_id.in_(document_ids), DocumentSegment.index_node_id.in_(document_ids),
DocumentSegment.status == "indexing" DocumentSegment.status == "indexing"
).update({ ).update({