|
@@ -79,7 +79,7 @@ func AIGCBaseOnPromote(aigc AIGC) (resp bus_response.AIGCEtaResponse, err error)
|
|
|
article, fileErr := ragEtaReportObj.GetById(aigc.ArticleId)
|
|
|
if fileErr != nil {
|
|
|
// 找不到就处理失败
|
|
|
- utils.FileLog.Error("公众号文章不存在")
|
|
|
+ utils.FileLog.Error("ETA文章不存在")
|
|
|
err = fmt.Errorf("ETA文章不存在")
|
|
|
return
|
|
|
}
|
|
@@ -152,20 +152,37 @@ func AIGCBaseOnPromote(aigc AIGC) (resp bus_response.AIGCEtaResponse, err error)
|
|
|
}
|
|
|
if gcResp.Code == 404 {
|
|
|
param["PrevId"] = kbId
|
|
|
- article, fileErr := rag.GetArticleById(aigc.ArticleId)
|
|
|
- if fileErr != nil {
|
|
|
- // 找不到就处理失败
|
|
|
- utils.FileLog.Error("公众号文章不存在")
|
|
|
- err = fmt.Errorf("公众号文章不存在")
|
|
|
- return
|
|
|
+ var title, textContent string
|
|
|
+ switch aigc.Source {
|
|
|
+ case 0:
|
|
|
+ article, fileErr := rag.GetArticleById(aigc.ArticleId)
|
|
|
+ if fileErr != nil {
|
|
|
+ // 找不到就处理失败
|
|
|
+ utils.FileLog.Error("公众号文章不存在")
|
|
|
+ err = fmt.Errorf("公众号文章不存在")
|
|
|
+ return
|
|
|
+ }
|
|
|
+ textContent = article.TextContent
|
|
|
+ title = article.Title
|
|
|
+ case 1:
|
|
|
+ ragEtaReportObj := rag.RagEtaReport{}
|
|
|
+ article, fileErr := ragEtaReportObj.GetById(aigc.ArticleId)
|
|
|
+ if fileErr != nil {
|
|
|
+ // 找不到就处理失败
|
|
|
+ utils.FileLog.Error("ETA文章不存在")
|
|
|
+ err = fmt.Errorf("ETA文章不存在")
|
|
|
+ return
|
|
|
+ }
|
|
|
+ textContent = article.TextContent
|
|
|
+ title = article.Title
|
|
|
}
|
|
|
- if article.TextContent == "" {
|
|
|
+ if textContent == "" {
|
|
|
utils.FileLog.Error("暂不支持纯文本以外的内容生成")
|
|
|
err = fmt.Errorf("暂不支持纯文本以外的内容生成")
|
|
|
return
|
|
|
}
|
|
|
// 文章加入到知识库
|
|
|
- path, fileErr := localService.CreateArticleFile(article)
|
|
|
+ path, fileErr := localService.CreateArticleFile(title, textContent)
|
|
|
if fileErr != nil {
|
|
|
utils.FileLog.Error("创建文章文件失败,err: %v", fileErr)
|
|
|
err = fmt.Errorf("创建文章文件失败,err: %v", fileErr)
|