From 7b57ab5dea7559567f6c4306a581f4048a93c8ed Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Thu, 8 May 2025 09:20:34 +0800 Subject: [PATCH] Fix: retrieval component for shared KB issue. (#7513) ### What problem does this PR solve? #7483 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- agent/component/retrieval.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/agent/component/retrieval.py b/agent/component/retrieval.py index 22a4b6827..0a34b5d66 100644 --- a/agent/component/retrieval.py +++ b/agent/component/retrieval.py @@ -97,7 +97,7 @@ class Retrieval(ComponentBase, ABC): kbinfos = settings.retrievaler.retrieval( query, embd_mdl, - kbs[0].tenant_id, + [kb.tenant_id for kb in kbs], filtered_kb_ids, 1, self._param.top_n, @@ -111,7 +111,7 @@ class Retrieval(ComponentBase, ABC): kbinfos = {"chunks": [], "doc_aggs": []} if self._param.use_kg and kbs: - ck = settings.kg_retrievaler.retrieval(query, [kbs[0].tenant_id], filtered_kb_ids, embd_mdl, LLMBundle(kbs[0].tenant_id, LLMType.CHAT)) + ck = settings.kg_retrievaler.retrieval(query, [kb.tenant_id for kb in kbs], filtered_kb_ids, embd_mdl, LLMBundle(kbs[0].tenant_id, LLMType.CHAT)) if ck["content_with_weight"]: kbinfos["chunks"].insert(0, ck)