diff --git a/agent/component/retrieval.py b/agent/component/retrieval.py
index 40d16ce8f..90cb8c9eb 100644
--- a/agent/component/retrieval.py
+++ b/agent/component/retrieval.py
@@ -24,6 +24,7 @@ from api.db.services.llm_service import LLMBundle
 from api import settings
 from agent.component.base import ComponentBase, ComponentParamBase
 from rag.app.tag import label_question
+from rag.prompts import kb_prompt
 from rag.utils.tavily_conn import Tavily
 
 
@@ -102,9 +103,7 @@ class Retrieval(ComponentBase, ABC):
                 df["empty_response"] = self._param.empty_response
             return df
 
-        df = pd.DataFrame(kbinfos["chunks"])
-        df["content"] = df["content_with_weight"]
-        del df["content_with_weight"]
+        df = pd.DataFrame({"content": kb_prompt(kbinfos, 200000)})
         logging.debug("{} {}".format(query, df))
         return df.dropna()