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