eta_api/models/ai_summary/ai_pormpt.go

package ai_summary

import (
	"github.com/beego/beego/v2/client/orm"
	"time"
)

type AiPrompt struct {
	AiPromptId    int       `orm:"column(ai_prompt_id);pk"` // ai纪要提示词id
	PromptContent string    // 原始内容
	SysAdminId    int       // 创建人ID
	SysAdminName  string    // 创建人姓名
	Title         string    // 文档标题
	CreateTime    time.Time // 创建时间
	ModifyTime    time.Time // 更新时间
	IsShare       int       // 是否分享，0：不分享，1：分享
}

type AiPromptItem struct {
	AiPromptId    int    `orm:"column(ai_prompt_id);pk"` // ai纪要提示词id
	PromptContent string // 原始内容
	SysAdminId    int    // 创建人ID
	SysAdminName  string // 创建人姓名
	Title         string // 文档标题
	CreateTime    string // 创建时间
	ModifyTime    string // 更新时间
	IsShare       int    // 是否分享，0：不分享，1：分享
}

func GetAiPromptList(sysUserId, isShare int) (items []*AiPromptItem, err error) {
	sql := ``
	if isShare == 1 {
		sql = ` SELECT * FROM ai_prompt WHERE sys_admin_id=? and is_share=1 ORDER BY create_time DESC `
	} else {
		sql = ` SELECT * FROM ai_prompt WHERE sys_admin_id=? ORDER BY create_time DESC `
	}
	o := orm.NewOrm()
	_, err = o.Raw(sql, sysUserId).QueryRows(&items)
	return
}

func GetAiPromptShareList() (items []*AiPromptItem, err error) {
	sql := ``
	sql = ` SELECT * FROM ai_prompt WHERE is_share=1 ORDER BY create_time DESC `

	o := orm.NewOrm()
	_, err = o.Raw(sql).QueryRows(&items)
	return
}

type RespGroupListItem struct {
	GroupId    int64  `description:"目录id"`
	GroupName  string `description:"目录名称"`
	AdminId    int    `description:"目录创建者账号ID"`
	IsShare    int8   `description:"是否共享，0私有，1共享"`
	PromptList []*AiPromptItem
}

type AiPromptAddReq struct {
	PromptContent string // 原始内容
	Title         string // 文档标题
}

func AddPropmt(item *AiPrompt) (lastId int64, err error) {
	o := orm.NewOrm()
	lastId, err = o.Insert(item)
	return
}

type AiPromptEditReq struct {
	PromptContent string // 原始内容
	Title         string // 文档标题
	AiPromptId    int    // ai纪要提示词id
	IsShare       int    // 是否分享，0：不分享，1：分享
}

// Update
func (aiPrompt *AiPrompt) Update(cols []string) (err error) {
	o := orm.NewOrm()
	_, err = o.Update(aiPrompt, cols...)
	return
}

// DelAiSummaryById 根据纪要id删除纪要
func DelAiPromptyId(aiPromptId int) (err error) {
	o := orm.NewOrm()
	sql := `delete from ai_prompt where ai_prompt_id = ? `
	_, err = o.Raw(sql, aiPromptId).Exec()
	return
}

func GetAiPromptById(pptId int) (item *AiPrompt, err error) {
	o := orm.NewOrm()
	sql := `SELECT * FROM ai_prompt WHERE 1=1 AND ai_prompt_id=? `
	err = o.Raw(sql, pptId).QueryRow(&item)
	return
}

type DeleteAipromptReq struct {
	AiPromptId int // ai纪要提示词id
}