github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-21 19:31:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(reranker)In the recall test scenario, the reranking logic is updated based on the application configuration

Browse Source 
The logic for determining whether to use reranking and setting the `top_k` value now depends on the `app_config` instead of directly using `CFG.RERANK_MODEL` and `CFG.RERANK_TOP_K`. This change ensures that the application's specific configuration settings are respected.
This commit is contained in:
tam 2025-04-19 15:20:10 +08:00
parent 129509fcd2
commit cc287702bc
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
packages/dbgpt-app/src/dbgpt_app/knowledge/service.py
View File

@ -340,11 +340,14 @@ class KnowledgeService:
else 0.3
)
if CFG.RERANK_MODEL is not None:
if top_k < int(CFG.RERANK_TOP_K) or top_k < 20:
app_config = CFG.SYSTEM_APP.config.configs.get("app_config")
rerank_top_k = app_config.rag.rerank_top_k
if app_config.models.rerankers:
if top_k < int(rerank_top_k) or top_k < 20:
# We use reranker, so if the top_k is less than 20,
# we need to set it to 20
top_k = max(int(CFG.RERANK_TOP_K), 20)
top_k = max(int(rerank_top_k), 20)
knowledge_space_retriever = KnowledgeSpaceRetriever(
space_id=space.id, top_k=top_k, system_app=CFG.SYSTEM_APP
@ -360,7 +363,8 @@ class KnowledgeService:
)
recall_top_k = int(doc_recall_test_request.recall_top_k)
if CFG.RERANK_MODEL is not None:
if app_config.models.rerankers:
rerank_embeddings = RerankEmbeddingFactory.get_instance(
CFG.SYSTEM_APP
).create()