ai_pormpt.go 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. package ai_summary
  2. import (
  3. "github.com/beego/beego/v2/client/orm"
  4. "time"
  5. )
  6. type AiPrompt struct {
  7. AiPromptId int `orm:"column(ai_prompt_id);pk"` // ai纪要提示词id
  8. PromptContent string // 原始内容
  9. SysAdminId int // 创建人ID
  10. SysAdminName string // 创建人姓名
  11. Title string // 文档标题
  12. CreateTime time.Time // 创建时间
  13. ModifyTime time.Time // 更新时间
  14. IsShare int // 是否分享,0:不分享,1:分享
  15. }
  16. type AiPromptItem struct {
  17. AiPromptId int `orm:"column(ai_prompt_id);pk"` // ai纪要提示词id
  18. PromptContent string // 原始内容
  19. SysAdminId int // 创建人ID
  20. SysAdminName string // 创建人姓名
  21. Title string // 文档标题
  22. CreateTime string // 创建时间
  23. ModifyTime string // 更新时间
  24. IsShare int // 是否分享,0:不分享,1:分享
  25. }
  26. func GetAiPromptList(sysUserId, isShare int) (items []*AiPromptItem, err error) {
  27. sql := ``
  28. if isShare == 1 {
  29. sql = ` SELECT * FROM ai_prompt WHERE sys_admin_id=? and is_share=1 ORDER BY create_time DESC `
  30. } else {
  31. sql = ` SELECT * FROM ai_prompt WHERE sys_admin_id=? ORDER BY create_time DESC `
  32. }
  33. o := orm.NewOrm()
  34. _, err = o.Raw(sql, sysUserId).QueryRows(&items)
  35. return
  36. }
  37. func GetAiPromptShareList() (items []*AiPromptItem, err error) {
  38. sql := ``
  39. sql = ` SELECT * FROM ai_prompt WHERE is_share=1 ORDER BY create_time DESC `
  40. o := orm.NewOrm()
  41. _, err = o.Raw(sql).QueryRows(&items)
  42. return
  43. }
  44. type RespGroupListItem struct {
  45. GroupId int64 `description:"目录id"`
  46. GroupName string `description:"目录名称"`
  47. AdminId int `description:"目录创建者账号ID"`
  48. IsShare int8 `description:"是否共享,0私有,1共享"`
  49. PromptList []*AiPromptItem
  50. }