38 lines
1.0 KiB
Go
38 lines
1.0 KiB
Go
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:"文档大小"`
|
|
}
|