Files
rag-local/kb/model/entity/dataset.go
T
2026-08-07 13:18:53 +08:00

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"`
}