llm.go 1006 B

12345678910111213141516171819202122232425262728293031323334353637
  1. package cache
  2. import (
  3. "eta/eta_task/utils"
  4. "fmt"
  5. )
  6. type RagEtaReportOpOp struct {
  7. Source string
  8. ReportId int
  9. ReportChapterId int
  10. }
  11. // RagEtaReportOpToCache
  12. // @Description: 将eta报告入知识库操作加入缓存
  13. // @author: Roc
  14. // @datetime 2025-04-07 15:05:22
  15. // @param reportId int
  16. // @param reportChapterId int
  17. // @param source string
  18. // @return bool
  19. func RagEtaReportOpToCache(reportId, reportChapterId int, source string) bool {
  20. record := new(RagEtaReportOpOp)
  21. record.Source = source
  22. record.ReportId = reportId
  23. record.ReportChapterId = reportChapterId
  24. if utils.Re == nil {
  25. err := utils.Rc.LPush(utils.CACHE_ETA_REPORT_KNOWLEDGE, record)
  26. utils.FileLog.Info(fmt.Sprintf("将eta报告入知识库操作加入缓存 加入缓存 RagEtaReportOpToCache LPush: 操作类型:%s,报告id:%d,章节id:%d", source, reportId, reportChapterId))
  27. if err != nil {
  28. fmt.Println("RagEtaReportOpToCache LPush Err:" + err.Error())
  29. }
  30. return true
  31. }
  32. return false
  33. }