Fix/shared lock (#210)

This commit is contained in:
John Wang
2023-05-25 21:31:11 +08:00
committed by GitHub
parent 4ef6392de5
commit 1a5acf43aa
3 changed files with 18 additions and 3 deletions

View File

@@ -34,5 +34,9 @@ class DatasetIndexToolCallbackHandler(IndexToolCallbackHandler):
db.session.query(DocumentSegment).filter(
DocumentSegment.dataset_id == self.dataset_id,
DocumentSegment.index_node_id == index_node_id
).update({DocumentSegment.hit_count: DocumentSegment.hit_count + 1}, synchronize_session=False)
).update(
{DocumentSegment.hit_count: DocumentSegment.hit_count + 1},
synchronize_session=False
)
db.session.commit()