Просмотр исходного кода

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

### What problem does this PR solve?

#6619
### Type of change


- [x] Performance Improvement
tags/v0.18.0
Kevin Hu 7 месяцев назад
Родитель
Сommit
60b9c027c8
Аккаунт пользователя с таким Email не найден
1 измененных файлов: 2 добавлений и 3 удалений
  1. 2
    3
      agent/component/retrieval.py

+ 2
- 3
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()


Загрузка…
Отмена
Сохранить