AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 22:49:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: segments query missing idx hit (#1072)

Browse Source
This commit is contained in:
takatost 2023-08-31 19:39:44 +08:00 committed by GitHub
parent fcc6b41ab7
commit 2191312bb9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/core/tool/dataset_retriever_tool.py
View File

@ -98,7 +98,8 @@ class DatasetRetrieverTool(BaseTool):
hit_callback.on_tool_end(documents) hit_callback.on_tool_end(documents)
document_context_list = [] document_context_list = []
index_node_ids = [document.metadata['doc_id'] for document in documents] index_node_ids = [document.metadata['doc_id'] for document in documents]
segments = DocumentSegment.query.filter(DocumentSegment.completed_at.isnot(None), segments = DocumentSegment.query.filter(DocumentSegment.dataset_id == self.dataset_id,
DocumentSegment.completed_at.isnot(None),
DocumentSegment.status == 'completed', DocumentSegment.status == 'completed',
DocumentSegment.enabled == True, DocumentSegment.enabled == True,
DocumentSegment.index_node_id.in_(index_node_ids) DocumentSegment.index_node_id.in_(index_node_ids)