35 lines
929 B
Go
35 lines
929 B
Go
package entity
|
|
|
|
import (
|
|
"rag/consts/keyword"
|
|
|
|
"gitea.redpowerfuture.com/red-future/common/beans"
|
|
)
|
|
|
|
type keywordCol struct {
|
|
beans.SQLBaseCol
|
|
DatasetId string
|
|
DocumentId string
|
|
Word string
|
|
Weight string
|
|
KeywordType string
|
|
}
|
|
|
|
var KeywordCol = keywordCol{
|
|
SQLBaseCol: beans.DefSQLBaseCol,
|
|
DatasetId: "dataset_id",
|
|
DocumentId: "document_id",
|
|
Word: "word",
|
|
Weight: "weight",
|
|
KeywordType: "keyword_type",
|
|
}
|
|
|
|
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:"权重"`
|
|
KeywordType keyword.KeywordType `orm:"keyword_type" json:"keywordType" dc:"类型"`
|
|
}
|