From bf21ae36f393b090aa11f8361e787c118eb79b94 Mon Sep 17 00:00:00 2001 From: Cooper <42096311+FOkvj@users.noreply.github.com> Date: Wed, 28 May 2025 14:28:37 +0800 Subject: [PATCH] Fix agent retriever resource (#2719) Co-authored-by: dong Co-authored-by: Fangyin Cheng --- .../src/dbgpt/agent/resource/knowledge.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/packages/dbgpt-core/src/dbgpt/agent/resource/knowledge.py b/packages/dbgpt-core/src/dbgpt/agent/resource/knowledge.py index 03f6a42ad..528d9c8c6 100644 --- a/packages/dbgpt-core/src/dbgpt/agent/resource/knowledge.py +++ b/packages/dbgpt-core/src/dbgpt/agent/resource/knowledge.py @@ -38,11 +38,16 @@ class RetrieverResource(Resource[ResourceParameters]): self._name = name self._retriever = retriever app_config = CFG.SYSTEM_APP.config.configs.get("app_config") - rerank_embeddings = RerankEmbeddingFactory.get_instance(CFG.SYSTEM_APP).create() self.need_rerank = bool(app_config.models.rerankers) - self.reranker = RerankEmbeddingsRanker( - rerank_embeddings, topk=app_config.rag.rerank_top_k - ) + if self.need_rerank: + rerank_embeddings = RerankEmbeddingFactory.get_instance( + CFG.SYSTEM_APP + ).create() + self.reranker = RerankEmbeddingsRanker( + rerank_embeddings, topk=app_config.rag.rerank_top_k + ) + else: + self.reranker = None @property def name(self) -> str: