package com.storlead.knowledge.entity;

import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

@Data
@ApiModel(value = "RetrievalModelEntity", description = "知识库检索模型字典")
public class RetrievalModelDictEntity {
    @ApiModelProperty(value = "搜索方法")
    private String searchMethod;
    @ApiModelProperty(value = "是否启用重排")
    private boolean rerankingEnable;
    @ApiModelProperty(value = "重排模式")
    private String rerankingMode;
    @ApiModelProperty(value = "重排模型")
    private RerankingModelEntity rerankingModel;
    @ApiModelProperty(value = "权重")
    private double weights;
    @ApiModelProperty(value = "topK")
    private int topK;
    @ApiModelProperty(value = "是否启用分数阈值")
    private boolean scoreThresholdEnabled;
    @ApiModelProperty(value = "分数阈值")
    private double scoreThreshold;

}
