opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: shared lock

This commit is contained in:
John Wang 2023-05-25 20:23:17 +08:00
parent 9e00e3894e
commit b70a4a9ec8
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
api/core/callback_handler/index_tool_callback_handler.py
View File

@ -31,8 +31,17 @@ class DatasetIndexToolCallbackHandler(IndexToolCallbackHandler):
index_node_id = node.node.doc_id
# add hit count to document segment
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_stmt = (
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
)
)
db.session.execute(update_stmt)
db.session.commit()