eta_api/models/rag/rag_eta_report_abstract.go

package rag

import (
	"database/sql"
	"eta/eta_api/global"
	"eta/eta_api/utils"
	"fmt"
	"time"
)

// RagEtaReportAbstract 报告摘要
type RagEtaReportAbstract struct {
	RagEtaReportAbstractId int       `gorm:"primaryKey;column:rag_eta_report_abstract_id" description:"-"`
	RagEtaReportId         int       `gorm:"column:rag_eta_report_id" description:"ETA报告id"`
	Content                string    `gorm:"column:content" description:"摘要内容"`
	QuestionId             int       `gorm:"column:question_id" description:"提示词Id"`
	QuestionContent        string    `gorm:"column:question_content" description:"questionContent"`
	Version                int       `gorm:"column:version" description:"版本号"`
	Tags                   string    `gorm:"column:tags" description:"标签"`
	TagsName               string    `gorm:"column:tags_name" description:"标签名，多个用英文逗号隔开"`
	VectorKey              string    `gorm:"column:vector_key" description:"向量key标识"`
	ModifyTime             time.Time `gorm:"column:modify_time" description:"modifyTime"`
	CreateTime             time.Time `gorm:"column:create_time" description:"createTime"`
}

// TableName get sql table name.获取数据库表名
func (m *RagEtaReportAbstract) TableName() string {
	return "rag_eta_report_abstract"
}

// RagEtaReportAbstractColumns get sql column name.获取数据库列名
var RagEtaReportAbstractColumns = struct {
	RagEtaReportAbstractId string
	RagEtaReportId         string
	Content                string
	QuestionId             string
	QuestionContent        string
	Version                string
	Tags                   string
	TagsName               string
	VectorKey              string
	ModifyTime             string
	CreateTime             string
}{
	RagEtaReportAbstractId: "rag_eta_report_abstract_id",
	RagEtaReportId:         "rag_eta_report_id",
	Content:                "content",
	QuestionId:             "question_id",
	QuestionContent:        "question_content",
	Version:                "version",
	Tags:                   "tags",
	TagsName:               "tags_name",
	VectorKey:              "vector_key",
	ModifyTime:             "modify_time",
	CreateTime:             "create_time",
}

func (m *RagEtaReportAbstract) Create() (err error) {
	err = global.DbMap[utils.DbNameAI].Create(&m).Error

	return
}

func (m *RagEtaReportAbstract) Update(updateCols []string) (err error) {
	err = global.DbMap[utils.DbNameAI].Select(updateCols).Updates(&m).Error

	return
}

func (m *RagEtaReportAbstract) Del() (err error) {
	err = global.DbMap[utils.DbNameAI].Delete(&m).Error

	return
}

func (m *RagEtaReportAbstract) GetById(id int) (item *RagEtaReportAbstract, err error) {
	err = global.DbMap[utils.DbNameAI].Where(fmt.Sprintf("%s = ?", RagEtaReportAbstractColumns.RagEtaReportAbstractId), id).First(&item).Error

	return
}

func (m *RagEtaReportAbstract) GetByIdList(idList []int) (items []*RagEtaReportAbstract, err error) {
	err = global.DbMap[utils.DbNameAI].Where(fmt.Sprintf("%s in (?) ", RagEtaReportAbstractColumns.RagEtaReportAbstractId), idList).Find(&items).Error

	return
}

func (m *RagEtaReportAbstract) GetListByQuestionId(questionId int) (items []*RagEtaReportAbstract, err error) {
	err = global.DbMap[utils.DbNameAI].Where(fmt.Sprintf("%s = ? ", RagEtaReportAbstractColumns.QuestionId), questionId).Find(&items).Error

	return
}

func (m *RagEtaReportAbstract) GetListByCondition(field, condition string, pars []interface{}, startSize, pageSize int) (items []*RagEtaReportAbstract, err error) {
	if field == "" {
		field = "*"
	}
	sqlStr := fmt.Sprintf(`SELECT %s FROM %s WHERE 1=1 %s  order by rag_eta_report_abstract_id desc LIMIT ?,?`, field, m.TableName(), condition)
	pars = append(pars, startSize, pageSize)
	err = global.DbMap[utils.DbNameAI].Raw(sqlStr, pars...).Find(&items).Error

	return
}

func (m *RagEtaReportAbstract) DelByIdList(idList []int) (err error) {
	if len(idList) <= 0 {
		return
	}
	sqlStr := fmt.Sprintf(`delete from %s where %s in (?)`, m.TableName(), RagEtaReportAbstractColumns.RagEtaReportAbstractId)
	err = global.DbMap[utils.DbNameAI].Exec(sqlStr, idList).Error

	return
}

// GetByRagEtaReportId
// @Description: 根据报告id获取摘要
// @author: Roc
// @receiver m
// @datetime 2025-03-07 10:00:59
// @param id int
// @return item *RagEtaReportAbstract
// @return err error
func (m *RagEtaReportAbstract) GetByRagEtaReportId(id int) (item *RagEtaReportAbstract, err error) {
	err = global.DbMap[utils.DbNameAI].Where(fmt.Sprintf("%s = ?", RagEtaReportAbstractColumns.RagEtaReportId), id).Order(fmt.Sprintf(`%s DESC`, RagEtaReportAbstractColumns.RagEtaReportAbstractId)).First(&item).Error

	return
}

// GetByRagEtaReportIdAndQuestionId
// @Description: 根据报告id和提示词ID获取摘要
// @author: Roc
// @receiver m
// @datetime 2025-04-17 17:39:27
// @param articleId int
// @param questionId int
// @return item *RagEtaReportAbstract
// @return err error
func (m *RagEtaReportAbstract) GetByRagEtaReportIdAndQuestionId(articleId, questionId int) (item *RagEtaReportAbstract, err error) {
	err = global.DbMap[utils.DbNameAI].Where(fmt.Sprintf("%s = ? AND %s = ? ", RagEtaReportAbstractColumns.RagEtaReportId, RagEtaReportAbstractColumns.QuestionId), articleId, questionId).Order(fmt.Sprintf(`%s DESC`, RagEtaReportAbstractColumns.RagEtaReportAbstractId)).First(&item).Error

	return
}

type RagEtaReportAbstractView struct {
	RagEtaReportAbstractId int    `gorm:"primaryKey;column:rag_eta_report_abstract_id" description:"-"`
	RagEtaReportId         int    `gorm:"column:rag_eta_report_id" description:"ETA报告id"`
	Abstract               string `gorm:"column:abstract;type:longtext;comment:摘要内容;" description:"摘要内容"`
	QuestionId             int    `gorm:"column:question_id" description:"提示词Id"`
	QuestionContent        string `gorm:"column:question_content" description:"questionContent"`
	Version                int    `gorm:"column:version" description:"版本号"`
	Tags                   string `gorm:"column:tags" description:"标签"`
	TagsName               string `gorm:"column:tags_name" description:"标签名，多个用英文逗号隔开"`
	VectorKey              string `gorm:"column:vector_key" description:"向量key标识"`
	ModifyTime             string `gorm:"column:modify_time;type:datetime;default:NULL;" description:"modify_time"`
	CreateTime             string `gorm:"column:create_time;type:datetime;default:NULL;" description:"create_time"`
	Title                  string `gorm:"column:title;type:varchar(255);comment:标题;" description:"标题"`
}

type RagEtaReportAbstractItem struct {
	RagEtaReportAbstractId int       `gorm:"primaryKey;column:rag_eta_report_abstract_id" description:"-"`
	RagEtaReportId         int       `gorm:"column:rag_eta_report_id" description:"ETA报告id"`
	Content                string    `gorm:"column:content" description:"摘要内容"`
	QuestionId             int       `gorm:"column:question_id" description:"提示词Id"`
	QuestionContent        string    `gorm:"column:question_content" description:"questionContent"`
	Version                int       `gorm:"column:version" description:"版本号"`
	Tags                   string    `gorm:"column:tags" description:"标签"`
	TagsName               string    `gorm:"column:tags_name" description:"标签名，多个用英文逗号隔开"`
	VectorKey              string    `gorm:"column:vector_key" description:"向量key标识"`
	ModifyTime             time.Time `gorm:"column:modify_time;type:datetime;default:NULL;" description:"modify_time"`
	CreateTime             time.Time `gorm:"column:create_time;type:datetime;default:NULL;" description:"create_time"`
	Title                  string    `gorm:"column:title;type:varchar(255);comment:标题;" description:"标题"`
}

func (m *RagEtaReportAbstractItem) ToView() RagEtaReportAbstractView {
	return RagEtaReportAbstractView{
		RagEtaReportAbstractId: m.RagEtaReportAbstractId,
		RagEtaReportId:         m.RagEtaReportId,
		Abstract:               m.Content,
		Version:                m.Version,
		VectorKey:              m.VectorKey,
		ModifyTime:             utils.DateStrToDateTimeStr(m.ModifyTime),
		CreateTime:             utils.DateStrToDateTimeStr(m.CreateTime),
		Title:                  m.Title,
		QuestionId:             m.QuestionId,
		Tags:                   m.Tags,
		TagsName:               m.TagsName,
		QuestionContent:        m.QuestionContent,
	}
}

func (m *RagEtaReportAbstract) EtaReportAbstractItem(list []*RagEtaReportAbstractItem) (etaReportAbstractViewList []RagEtaReportAbstractView) {
	etaReportAbstractViewList = make([]RagEtaReportAbstractView, 0)

	for _, v := range list {
		etaReportAbstractViewList = append(etaReportAbstractViewList, v.ToView())
	}
	return
}

func (m *RagEtaReportAbstract) GetListByPlatformCondition(field, condition string, pars []interface{}, startSize, pageSize int) (items []*RagEtaReportAbstractItem, err error) {
	if field == "" {
		field = "*"
	}
	sqlStr := fmt.Sprintf(`SELECT %s FROM %s AS a 
          WHERE 1=1  %s order by a.modify_time DESC,a.rag_eta_report_abstract_id DESC LIMIT ?,?`, field, m.TableName(), condition)
	pars = append(pars, startSize, pageSize)
	err = global.DbMap[utils.DbNameAI].Raw(sqlStr, pars...).Find(&items).Error

	return
}

func (m *RagEtaReportAbstract) GetCountByPlatformCondition(condition string, pars []interface{}) (total int, err error) {
	var intNull sql.NullInt64
	sqlStr := fmt.Sprintf(`SELECT COUNT(1) total FROM %s AS a 
          WHERE 1=1  %s`, m.TableName(), condition)
	err = global.DbMap[utils.DbNameAI].Raw(sqlStr, pars...).Scan(&intNull).Error
	if err == nil && intNull.Valid {
		total = int(intNull.Int64)
	}

	return
}

func (m *RagEtaReportAbstract) GetPageListByPlatformCondition(condition string, pars []interface{}, startSize, pageSize int) (total int, items []*RagEtaReportAbstractItem, err error) {

	total, err = m.GetCountByPlatformCondition(condition, pars)
	if err != nil {
		return
	}
	if total > 0 {
		items, err = m.GetListByPlatformCondition(``, condition, pars, startSize, pageSize)
	}

	return
}

func (m *RagEtaReportAbstract) GetListByTagAndPlatformCondition(field, condition string, pars []interface{}, startSize, pageSize int) (items []*RagEtaReportAbstractItem, err error) {
	if field == "" {
		field = "*"
	}
	sqlStr := fmt.Sprintf(`SELECT %s FROM %s AS a 
          JOIN wechat_article AS b ON a.rag_eta_report_id=b.rag_eta_report_id
          JOIN wechat_platform AS c ON b.wechat_platform_id=c.wechat_platform_id
          JOIN wechat_platform_tag_mapping AS d ON c.wechat_platform_id=d.wechat_platform_id
          WHERE 1=1 AND b.is_deleted=0 %s  order by a.modify_time DESC,a.rag_eta_report_abstract_id DESC LIMIT ?,?`, field, m.TableName(), condition)
	pars = append(pars, startSize, pageSize)
	err = global.DbMap[utils.DbNameAI].Raw(sqlStr, pars...).Find(&items).Error

	return
}

func (m *RagEtaReportAbstract) GetCountByTagAndPlatformCondition(condition string, pars []interface{}) (total int, err error) {
	var intNull sql.NullInt64
	sqlStr := fmt.Sprintf(`SELECT COUNT(1) total FROM %s AS a 
          JOIN wechat_article AS b ON a.rag_eta_report_id=b.rag_eta_report_id
          JOIN wechat_platform AS c ON b.wechat_platform_id=c.wechat_platform_id
          JOIN wechat_platform_tag_mapping AS d ON c.wechat_platform_id=d.wechat_platform_id
          WHERE 1=1 AND b.is_deleted=0 %s`, m.TableName(), condition)
	err = global.DbMap[utils.DbNameAI].Raw(sqlStr, pars...).Scan(&intNull).Error
	if err == nil && intNull.Valid {
		total = int(intNull.Int64)
	}

	return
}

func (m *RagEtaReportAbstract) GetPageListByTagAndPlatformCondition(condition string, pars []interface{}, startSize, pageSize int) (total int, items []*RagEtaReportAbstractItem, err error) {

	total, err = m.GetCountByTagAndPlatformCondition(condition, pars)
	if err != nil {
		return
	}
	if total > 0 {
		items, err = m.GetListByTagAndPlatformCondition(`a.rag_eta_report_abstract_id,a.rag_eta_report_id,a.content AS abstract,a.version,a.vector_key,a.modify_time,a.create_time,b.title,b.link,d.tag_id`, condition, pars, startSize, pageSize)
	}

	return
}

// DelVectorKey
// @Description: 批量删除向量库
// @author: Roc
// @receiver m
// @datetime 2025-03-12 16:47:52
// @param ragEtaReportAbstractIdList []int
// @return err error
func (m *RagEtaReportAbstract) DelVectorKey(ragEtaReportAbstractIdList []int) (err error) {
	sqlStr := fmt.Sprintf(`UPDATE %s set vector_key = '' WHERE rag_eta_report_abstract_id IN (?)`, m.TableName())
	err = global.DbMap[utils.DbNameAI].Exec(sqlStr, ragEtaReportAbstractIdList).Error

	return
}