test/dify
1
0
Fork 0
mirror of https://gitee.com/dify_ai/dify.git synced 2024-12-03 03:38:08 +08:00
Code Issues Actions 4 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats 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({