forked from lxh/go-wxhelper
40 lines
1.1 KiB
Go
40 lines
1.1 KiB
Go
|
package entity
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/google/uuid"
|
||
|
|
"go-wechat/common/types"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// AiAssistant
|
||
|
|
// @description: AI助手表
|
||
|
|
type AiAssistant struct {
|
||
|
|
Id string `json:"id" gorm:"type:varchar(32);primarykey"`
|
||
|
|
CreatedAt types.DateTime `json:"createdAt"`
|
||
|
|
Name string `json:"name" gorm:"type:varchar(10);not null;comment:'名称'"`
|
||
|
|
Personality string `json:"personality" gorm:"type:varchar(999);not null;comment:'人设'"`
|
||
|
|
Model string `json:"model" gorm:"type:varchar(50);not null;comment:'使用的模型'"`
|
||
|
|
Enable bool `json:"enable" gorm:"type:tinyint(1);not null;default:1;comment:'是否启用'"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// TableName
|
||
|
|
// @description: 表名
|
||
|
|
// @receiver AiAssistant
|
||
|
|
// @return string
|
||
|
|
func (AiAssistant) TableName() string {
|
||
|
|
return "t_ai_assistant"
|
||
|
|
}
|
||
|
|
|
||
|
|
// BeforeCreate
|
||
|
|
// @description: 创建数据库对象之前生成UUID
|
||
|
|
// @receiver m
|
||
|
|
// @param *gorm.DB
|
||
|
|
// @return err
|
||
|
|
func (m *AiAssistant) BeforeCreate(*gorm.DB) (err error) {
|
||
|
|
if m.Id == "" {
|
||
|
|
m.Id = strings.ReplaceAll(uuid.New().String(), "-", "")
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|