eta_api/models/ai_summary/ai_summary.go

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
}