mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-07-31 15:47:05 +00:00
fix(knowledge recall test): Add system_app parameter to knowledgeSpaceRetriever initialization (#2472)
Co-authored-by: tam <tanwe@fulan.com.cn>
This commit is contained in:
parent
d6eb283e41
commit
34d86d693c
@ -341,7 +341,7 @@ class KnowledgeService:
|
|||||||
top_k = max(int(CFG.RERANK_TOP_K), 20)
|
top_k = max(int(CFG.RERANK_TOP_K), 20)
|
||||||
|
|
||||||
knowledge_space_retriever = KnowledgeSpaceRetriever(
|
knowledge_space_retriever = KnowledgeSpaceRetriever(
|
||||||
space_id=space.id, top_k=top_k
|
space_id=space.id, top_k=top_k, system_app=CFG.SYSTEM_APP
|
||||||
)
|
)
|
||||||
chunks = await knowledge_space_retriever.aretrieve_with_scores(
|
chunks = await knowledge_space_retriever.aretrieve_with_scores(
|
||||||
question, score_threshold
|
question, score_threshold
|
||||||
|
Loading…
Reference in New Issue
Block a user