24 lines
1.2 KiB
Go
24 lines
1.2 KiB
Go
package entity
|
|
|
|
import "github.com/gogf/gf/v2/os/gtime"
|
|
|
|
type Dataset struct {
|
|
Id int64 `orm:"id" json:"id"`
|
|
Name string `orm:"name" json:"name"`
|
|
Description string `orm:"description" json:"description"`
|
|
EmbeddingCfgId int64 `orm:"embedding_cfg_id" json:"embedding_cfg_id"`
|
|
ChunkSize int `orm:"chunk_size" json:"chunk_size"`
|
|
ChunkOverlap int `orm:"chunk_overlap" json:"chunk_overlap"`
|
|
UnitPattern string `orm:"unit_pattern" json:"unit_pattern"`
|
|
ContextPattern string `orm:"context_pattern" json:"context_pattern"`
|
|
ReactRounds int `orm:"react_rounds" json:"react_rounds"`
|
|
// 召回数量配置(0=全局默认,-1=尽量多,>0=固定值)
|
|
VecTopK int `orm:"vec_top_k" json:"vec_top_k"` // 向量原始召回数
|
|
FtsTopK int `orm:"fts_top_k" json:"fts_top_k"` // 全文原始召回数
|
|
RerankTopK int `orm:"rerank_top_k" json:"rerank_top_k"` // 重排候选数
|
|
RecallTopK int `orm:"recall_top_k" json:"recall_top_k"` // 最终返回数
|
|
Status int `orm:"status" json:"status"`
|
|
CreatedAt *gtime.Time `orm:"created_at" json:"created_at"`
|
|
UpdatedAt *gtime.Time `orm:"updated_at" json:"updated_at"`
|
|
}
|