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

chore: remove useless code in knowledge_retrieval_node (#4412)

Browse Source
This commit is contained in:
VoidIsVoid 2024-05-15 15:24:40 +08:00 committed by GitHub
parent 97b65f9b4b
commit a76ae2d756
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
View File

@ -146,7 +146,6 @@ class KnowledgeRetrievalNode(BaseNode):
if 'score' in item.metadata and item.metadata['score']: if 'score' in item.metadata and item.metadata['score']:
document_score_list[item.metadata['doc_id']] = item.metadata['score'] document_score_list[item.metadata['doc_id']] = item.metadata['score']
document_context_list = []
index_node_ids = [document.metadata['doc_id'] for document in all_documents] index_node_ids = [document.metadata['doc_id'] for document in all_documents]
segments = DocumentSegment.query.filter( segments = DocumentSegment.query.filter(
DocumentSegment.dataset_id.in_(dataset_ids), DocumentSegment.dataset_id.in_(dataset_ids),
@ -160,11 +159,6 @@ class KnowledgeRetrievalNode(BaseNode):
sorted_segments = sorted(segments, sorted_segments = sorted(segments,
key=lambda segment: index_node_id_to_position.get(segment.index_node_id, key=lambda segment: index_node_id_to_position.get(segment.index_node_id,
float('inf'))) float('inf')))
for segment in sorted_segments:
if segment.answer:
document_context_list.append(f'question:{segment.content} answer:{segment.answer}')
else:
document_context_list.append(segment.content)
for segment in sorted_segments: for segment in sorted_segments:
dataset = Dataset.query.filter_by( dataset = Dataset.query.filter_by(