知识库rerank设置

This commit is contained in:
2025-11-07 11:21:27 +08:00
parent d9947e273c
commit b98450df96
8 changed files with 360 additions and 34 deletions

View File

@@ -93,6 +93,11 @@ public class TbAiKnowledge extends BaseDTO {
*/
private String rerankModelProvider;
/**
* @description 是否启用Rerank
*/
private Boolean rerankingEnable;
/**
* @description 检索Top K返回前K个结果
*/
@@ -322,6 +327,14 @@ public class TbAiKnowledge extends BaseDTO {
this.rerankModelProvider = rerankModelProvider;
}
public Boolean getRerankingEnable() {
return rerankingEnable;
}
public void setRerankingEnable(Boolean rerankingEnable) {
this.rerankingEnable = rerankingEnable;
}
public Integer getRetrievalTopK() {
return retrievalTopK;
}