Skip to content

D
dify
Commit f87c0f0f authored by jyong's avatar jyong
Browse files
Options

fix app query answer

parent 2f1a2aa7
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
api/core/tool/dataset_index_tool.py
View file @ f87c0f0f
...@@ -56,11 +56,11 @@ class DatasetTool(BaseTool): ...@@ -56,11 +56,11 @@ class DatasetTool(BaseTool):
hit_callback = DatasetIndexToolCallbackHandler(self.dataset.id) hit_callback = DatasetIndexToolCallbackHandler(self.dataset.id)
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]
segments = DocumentSegment.query.filter(DocumentSegment.completed_at.isnot(None), segments = DocumentSegment.query.filter(DocumentSegment.completed_at.isnot(None),
DocumentSegment.status == 'completed', DocumentSegment.status == 'completed',
DocumentSegment.enabled == True, DocumentSegment.enabled == True,
DocumentSegment.index_node_id in [str(document.metadata['doc_id']) DocumentSegment.index_node_id.in_(index_node_ids)
for document in documents]
).all() ).all()
if segments: if segments:
......
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