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)