feat: rag初始版

model/entity/dataset.go

@@ -0,0 +1,37 @@
+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:"文档大小"`
+}