package entity

import (
	"gitea.com/red-future/common/beans"
)

type datasetCol struct {
	beans.SQLBaseCol
	Name          string
	Description   string
	Embedding     string
	Dimension     string
	DocumentCount string
	DocumentSize  string
}

var DatasetCol = datasetCol{
	SQLBaseCol:    beans.DefSQLBaseCol,
	Name:          "name",
	Description:   "description",
	Embedding:     "embedding",
	Dimension:     "dimension",
	DocumentCount: "document_count",
	DocumentSize:  "document_size",
}

// Dataset 数据集表
type Dataset struct {
	beans.SQLBaseDO `orm:",inline"`

	Name          string `orm:"name" json:"name" dc:"数据集名称"`
	Description   string `orm:"description" json:"description" dc:"数据集描述"`
	Embedding     string `orm:"embedding" json:"embedding" dc:"向量模型"`
	Dimension     int    `orm:"dimension" json:"dimension" dc:"向量维度"`
	DocumentCount int64  `orm:"document_count" json:"documentCount" dc:"文档数量"`
	DocumentSize  int64  `orm:"document_size" json:"documentSize" dc:"文档大小"`
}