From 32b1fd4c1e242850a6171ab93dd6a3754bb354c9 Mon Sep 17 00:00:00 2001 From: tam Date: Mon, 21 Apr 2025 09:58:01 +0800 Subject: [PATCH] feat(knowledge): add reranking functionality for simple_assistant_agent and summary_assis_agent - Implement reranking of retrieved chunks when needed - Improve relevance of search results by ranking chunks based on query --- packages/dbgpt-core/src/dbgpt/agent/resource/knowledge.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/packages/dbgpt-core/src/dbgpt/agent/resource/knowledge.py b/packages/dbgpt-core/src/dbgpt/agent/resource/knowledge.py index 82a1f3c43..03f6a42ad 100644 --- a/packages/dbgpt-core/src/dbgpt/agent/resource/knowledge.py +++ b/packages/dbgpt-core/src/dbgpt/agent/resource/knowledge.py @@ -111,6 +111,9 @@ class RetrieverResource(Resource[ResourceParameters]): if not question: raise ValueError("Question is required for knowledge resource.") chunks = await self.retrieve(question) + if self.need_rerank and len(chunks) > 1: + chunks = self.reranker.rank(candidates_with_scores=chunks, query=question) + prompt_template = """Resources-{name}:\n {content}""" prompt_template_zh = """资源-{name}:\n {content}""" if lang == "en":