adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #1051 from SuanmoSuanyangTechnology/feature/agent-tool_xjn

Browse Source 
feat(workflow)
This commit is contained in:
山程漫悟
2026-05-07 19:12:39 +08:00
committed by GitHub
parent 4c59e41b95 0dc8d8cbeb
commit 24d2fe726a
3 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
api/app/core/workflow/nodes/knowledge/node.py
View File

@@ -363,11 +363,12 @@ class KnowledgeRetrievalNode(BaseNode):
seen_doc_ids = set()
for chunk in final_rs:
meta = chunk.metadata or {}
doc_id = meta.get("document_id") or meta.get("doc_id")
if doc_id and doc_id not in seen_doc_ids:
seen_doc_ids.add(doc_id)
document_id = meta.get("document_id")
if document_id and document_id not in seen_doc_ids:
seen_doc_ids.add(document_id)
citations.append({
"document_id": str(doc_id),
"document_id": str(document_id),
"doc_id": meta.get("doc_id", ""),
"file_name": meta.get("file_name", ""),
"knowledge_id": str(meta.get("knowledge_id", kb_config.kb_id)),
"score": meta.get("score", 0.0),