|
|
|
@@ -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() |
|
|
|
|