feat: rag初始版

model/entity/keyword.go

@@ -0,0 +1,27 @@
+package entity
+
+import "gitea.com/red-future/common/beans"
+
+type keywordCol struct {
+	beans.SQLBaseCol
+	DatasetId  string
+	DocumentId string
+	Word       string
+	Weight     string
+}
+
+var KeywordCol = keywordCol{
+	SQLBaseCol: beans.DefSQLBaseCol,
+	DatasetId:  "dataset_id",
+	DocumentId: "document_id",
+	Word:       "word",
+	Weight:     "weight",
+}
+
+type Keyword struct {
+	beans.SQLBaseDO `orm:",inline"`
+	DatasetId       int64  `orm:"dataset_id" json:"datasetId" dc:"数据集ID"`
+	DocumentId      int64  `orm:"document_id" json:"documentId" dc:"文件ID"`
+	Word            string `orm:"word" json:"word" dc:"关键词"`
+	Weight          int16  `orm:"weight" json:"weight" dc:"权重"`
+}