From 60b9c027c86bacf3f32efb0b874e36410f8dfca2 Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Mon, 31 Mar 2025 11:45:56 +0800 Subject: [PATCH] Refa: add meta data to retrieval. (#6676) ### What problem does this PR solve? #6619 ### Type of change - [x] Performance Improvement --- agent/component/retrieval.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) 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()