知识库rerank设置
This commit is contained in:
@@ -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;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user