Skip to content

D
dify
Unverified Commit 2191312b authored by takatost's avatar takatost Committed by GitHub
Browse files
Options

fix: segments query missing idx hit (#1072)

parent fcc6b41a
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
api/core/tool/dataset_retriever_tool.py
View file @ 2191312b
...@@ -98,7 +98,8 @@ class DatasetRetrieverTool(BaseTool): ...@@ -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)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment