gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Refa: add meta data to retrieval. (#6676)

Browse source 
### What problem does this PR solve?

#6619
### Type of change


- [x] Performance Improvement
This commit is contained in:
Kevin Hu 2025-03-31 11:45:56 +08:00 • committed by GitHub
parent 2793c8e4fe
commit 60b9c027c8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
agent/component/retrieval.py
View file

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