Procházet zdrojové kódy
fix dataset_id and index_node_id idx missed in document_segments tabl… (#7681)
tags/0.7.3
Jyong
před 1 rokem
1 změnil soubory, kde provedl 2 přidání a 0 odebrání
+ 2
- 0
api/core/indexing_runner.py
Zobrazit soubor
| 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({ | ||||
| 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({ |
Načítá se…