opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-05 17:18:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: shared lock

Browse Source
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 index_node_id = node.node.doc_id
# add hit count to document segment # add hit count to document segment
db.session.query(DocumentSegment).filter( update_stmt = (
DocumentSegment.dataset_id == self.dataset_id, db.session.query(DocumentSegment)
DocumentSegment.index_node_id == index_node_id .filter(
).update({DocumentSegment.hit_count: DocumentSegment.hit_count + 1}, synchronize_session=False) 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()