eta_api/services/llm/promote_service.go

package llm

import (
	"eta/eta_api/models/rag"
	"eta/eta_api/utils"
	"fmt"
	"os"
)

func CreateArticleFile(item *rag.WechatArticle) (filePath string, err error) {
	if item.TextContent == `` {
		err = fmt.Errorf("生成文章原文文本失败,文章内容为空")
		return
	}
	defer func() {
		if err != nil {
			utils.FileLog.Error("上传文章原文到知识库失败，err:%v", err)
			fmt.Println("上传文章原文到知识库失败，err:", err)
		}
	}()

	// 生成临时文件
	uploadDir := utils.STATIC_DIR + "ai/article"
	err = os.MkdirAll(uploadDir, utils.DIR_MOD)
	if err != nil {
		err = fmt.Errorf("存储目录创建失败,Err:" + err.Error())
		return
	}
	fileName := utils.RemoveSpecialChars(item.Title) + `.md`
	tmpFilePath := uploadDir + "/" + fileName
	err = utils.SaveToFile(item.TextContent, tmpFilePath)
	if err != nil {
		err = fmt.Errorf("生成临时文件失败，Err:" + err.Error())
		return
	}
	defer func() {
		_ = os.Remove(tmpFilePath)
	}()
	return
}