package ai_summary
import (
"eta_gn/eta_api/global"
"fmt"
"time"
)
type AiSummary struct {
AiSummaryId int `gorm:"primaryKey;column:ai_summary_id;type:int(10) unsigned;not null"` // ai纪要id
SaDocId int `gorm:"index:idx_doc_id;column:sa_doc_id;type:int(10) unsigned;not null;default:0"` // 文档ID
OriginTitle string `gorm:"column:origin_title;type:varchar(255)"` // 原始内容标题
OriginContent string `gorm:"column:origin_content;type:text"` // 原始内容
ClassifyId int `gorm:"column:classify_id;type:int(11);not null"` // 分类id
Title string `gorm:"column:title;type:varchar(255);not null;default:''"` // 文档标题
CreateTime time.Time `gorm:"column:create_time;type:datetime"` // 创建时间
ModifyTime time.Time `gorm:"column:modify_time;type:datetime"` // 更新时间
OpenaiFileName string `gorm:"column:openai_file_name;type:varchar(255);default:''"` // 文件名称
OpenaiFilePath string `gorm:"column:openai_file_path;type:varchar(255);default:''"` // 文件路径
SummaryContent string `gorm:"column:summary_content;type:longtext"` // ai纪要内容
Sort int `gorm:"column:sort;type:int(9) unsigned;default:10"` // 排序字段，越小越靠前，默认值：10
SysUserId int `gorm:"column:sys_user_id;type:int(11)"` // 创建人id
SysUserRealName string `gorm:"column:sys_user_real_name;type:varchar(50)"` // 创建人姓名
}
type AiSummaryItems struct {
AiSummaryId int `gorm:"primaryKey;column:ai_summary_id;type:int(10) unsigned;not null"` // ai纪要id
SaDocId int `gorm:"index:idx_doc_id;column:sa_doc_id;type:int(10) unsigned;not null;default:0"` // 文档ID
OriginTitle string `gorm:"column:origin_title;type:varchar(255)"` // 原始内容标题
OriginContent string `gorm:"column:origin_content;type:text"` // 原始内容
ClassifyId int `gorm:"column:classify_id;type:int(11);not null"` // 分类id
Title string `gorm:"column:title;type:varchar(255);not null;default:''"` // 文档标题
CreateTime string `gorm:"column:create_time;type:datetime"` // 创建时间
ModifyTime string `gorm:"column:modify_time;type:datetime"` // 更新时间
OpenaiFileName string `gorm:"column:openai_file_name;type:varchar(255);default:''"` // 文件名称
OpenaiFilePath string `gorm:"column:openai_file_path;type:varchar(255);default:''"` // 文件路径
SummaryContent string `gorm:"column:summary_content;type:longtext"` // ai纪要内容
Sort int `gorm:"column:sort;type:int(9) unsigned;default:10"` // 排序字段，越小越靠前，默认值：10
SysUserId int `gorm:"column:sys_user_id;type:int(11)"` // 创建人id
SysUserRealName string `gorm:"column:sys_user_real_name;type:varchar(50)"` // 创建人姓名
ParentIds string `gorm:"-"`
}
// GetAiSummaryById 根据纪要id获取详情
func GetAiSummaryById(aiSummaryId int) (sandboxInfo *AiSummary, err error) {
sql := `select * from ai_summary where ai_summary_id = ? `
err = global.DEFAULT_DmSQL.Raw(sql, aiSummaryId).First(&sandboxInfo).Error
return
}
// GetAiSummaryItemById 根据纪要id获取详情
func GetAiSummaryItemById(aiSummaryId int) (sandboxInfo *AiSummaryItems, err error) {
sql := `select * from ai_summary where ai_summary_id = ? `
err = global.DEFAULT_DmSQL.Raw(sql, aiSummaryId).First(&sandboxInfo).Error
return
}
// DelAiSummaryById 根据纪要id删除纪要
func DelAiSummaryById(aiSummaryId int) (err error) {
sql := `delete from ai_summary where ai_summary_id = ? `
err = global.DEFAULT_DmSQL.Exec(sql, aiSummaryId).Error
return
}
func MoveAiSummary(aiSummaryId, classifyId int) (err error) {
sql := `UPDATE ai_summary SET classify_id = ? WHERE ai_summary_id = ?`
err = global.DEFAULT_DmSQL.Exec(sql, classifyId, aiSummaryId).Error
return
}
// GetFirstAiSummaryByClassifyId 获取当前分类下，且排序数相同 的排序第一条的数据
func GetFirstAiSummaryByClassifyId(classifyId int) (item *AiSummary, err error) {
sql := ` SELECT * FROM ai_summary WHERE classify_id=? order by sort asc,ai_summary_id asc limit 1`
err = global.DEFAULT_DmSQL.Raw(sql, classifyId).First(&item).Error
return
}
func (aiSummary *AiSummary) Update(cols []string) (err error) {
err = global.DEFAULT_DmSQL.Select(cols).Updates(aiSummary).Error
return
}
// UpdateAiSummarySortByClassifyId 根据纪要id更新排序
func UpdateAiSummarySortByClassifyId(classifyId, nowSort, prevAiSummaryId int, updateSort string) (err error) {
sql := ` update ai_summary set sort = ` + updateSort + ` WHERE classify_id=? AND `
if prevAiSummaryId > 0 {
sql += ` (sort > ? or (ai_summary_id > ` + fmt.Sprint(prevAiSummaryId) + ` and sort = ` + fmt.Sprint(nowSort) + `))`
}
err = global.DEFAULT_DmSQL.Exec(sql, classifyId, nowSort).Error
return
}
func GetAiSummaryListByCondition(condition string, pars []interface{}, startSize, pageSize int) (item []*AiSummaryItems, err error) {
sql := ` SELECT * FROM ai_summary WHERE 1=1 `
if condition != "" {
sql += condition
}
sql += " ORDER BY create_time DESC LIMIT ?,? "
pars = append(pars, startSize)
pars = append(pars, pageSize)
err = global.DEFAULT_DmSQL.Raw(sql, pars...).Find(&item).Error
return
}
func GetAiSummaryListCountByCondition(condition string, pars []interface{}) (count int, err error) {
sql := ` SELECT COUNT(1) AS count FROM ai_summary WHERE 1=1 `
if condition != "" {
sql += condition
}
err = global.DEFAULT_DmSQL.Raw(sql, pars...).Scan(&count).Error
return
}
func AddAiSummary(item *AiSummary) (lastId int64, err error) {
err = global.DEFAULT_DmSQL.Create(item).Error
lastId = int64(item.AiSummaryId)
return
}
type AddAiSummaryReq struct {
SaDocId int // 文档ID
OriginContent string // 原始内容
SummaryContent string // 纪要内容
ClassifyId int // 分类id
Title string // 文档标题
OpenaiFileName string // 文件名称
OpenaiFilePath string // 文件路径
OriginTitle string // 原文标题
}
type GenerateAiSummaryReq struct {
AiChatTopicId int `description:"主题id"`
OriginContent string // 原始内容
OpenaiFileId []string `description:"openai返回的文件id"`
Model string `description:"模型名称"`
Prompt string `description:"提示词"`
SaDocId int // 文档ID
}
// GetAiSummaryMaxSort 获取ai纪要下最大的排序数
func GetAiSummaryMaxSort(classifyId int) (sort int, err error) {
sql := `SELECT COALESCE(MAX(sort), 0) AS sort FROM ai_summary WHERE parent_id=? `
err = global.DEFAULT_DmSQL.Raw(sql, classifyId).Scan(&sort).Error
return
}
type AiSummaryAddResp struct {
AiSummaryId int
UniqueCode string // 唯一编码
ParentId int
}
type AiSummaryDetailResp struct {
*AiSummaryItems
SaDocTitle string `description:"文档库标题"`
SaDocClassifyId int `description:"文档库分类id"`
}
func GetSummaryCountByCondition(condition string, pars []interface{}) (count int, err error) {
sql := ` SELECT COUNT(1) AS count FROM ai_summary WHERE 1=1 `
if condition != "" {
sql += condition
}
err = global.DEFAULT_DmSQL.Raw(sql, pars...).Scan(&count).Error
return
}