promote_service.go 751 B

123456789101112131415161718192021222324252627282930
  1. package llm
  2. import (
  3. "eta/eta_api/models/rag"
  4. "eta/eta_api/utils"
  5. "fmt"
  6. "os"
  7. )
  8. func CreateArticleFile(item *rag.WechatArticle) (tmpFilePath string, err error) {
  9. if item.TextContent == `` {
  10. err = fmt.Errorf("生成文章原文文本失败,文章内容为空")
  11. return
  12. }
  13. // 生成临时文件
  14. uploadDir := utils.STATIC_DIR + "ai/article"
  15. err = os.MkdirAll(uploadDir, utils.DIR_MOD)
  16. if err != nil {
  17. err = fmt.Errorf("存储目录创建失败,Err:" + err.Error())
  18. return
  19. }
  20. fileName := utils.RemoveSpecialChars(item.Title) + `.md`
  21. tmpFilePath = uploadDir + "/" + fileName
  22. err = utils.SaveToFile(item.TextContent, tmpFilePath)
  23. if err != nil {
  24. err = fmt.Errorf("生成临时文件失败,Err:" + err.Error())
  25. return
  26. }
  27. return
  28. }