github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-05 11:01:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix agent retriever resource (#2719)

Browse Source 
Co-authored-by: dong <dongzhancai@iie2.com>
Co-authored-by: Fangyin Cheng <staneyffer@gmail.com>
This commit is contained in:
Cooper
2025-05-28 14:28:37 +08:00
committed by GitHub
parent 761c75a2c3
commit bf21ae36f3
1 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
packages/dbgpt-core/src/dbgpt/agent/resource/knowledge.py
View File

@@ -38,11 +38,16 @@ class RetrieverResource(Resource[ResourceParameters]):
self._name = name self._name = name
self._retriever = retriever self._retriever = retriever
app_config = CFG.SYSTEM_APP.config.configs.get("app_config") 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.need_rerank = bool(app_config.models.rerankers)
self.reranker = RerankEmbeddingsRanker( if self.need_rerank:
rerank_embeddings, topk=app_config.rag.rerank_top_k 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 @property
def name(self) -> str: def name(self) -> str: