|
@@ -616,13 +616,13 @@ func ArticleToKnowledge(item *rag.WechatArticle) {
|
|
|
// @author: Roc
|
|
|
// @datetime 2025-03-10 16:14:59
|
|
|
// @param wechatArticleItem *rag.WechatArticle
|
|
|
-// @param item *rag.WechatArticleAbstract
|
|
|
-func AbstractToKnowledge(wechatArticleItem *rag.WechatArticle, item *rag.WechatArticleAbstract, isReUpload bool) {
|
|
|
- if item.Content == `` {
|
|
|
+// @param abstractItem *rag.WechatArticleAbstract
|
|
|
+func AbstractToKnowledge(wechatArticleItem *rag.WechatArticle, abstractItem *rag.WechatArticleAbstract, isReUpload bool) {
|
|
|
+ if abstractItem.Content == `` {
|
|
|
return
|
|
|
}
|
|
|
// 已经生成了,那就不处理了
|
|
|
- if item.VectorKey != `` && !isReUpload {
|
|
|
+ if abstractItem.VectorKey != `` && !isReUpload {
|
|
|
return
|
|
|
}
|
|
|
var err error
|
|
@@ -631,6 +631,9 @@ func AbstractToKnowledge(wechatArticleItem *rag.WechatArticle, item *rag.WechatA
|
|
|
utils.FileLog.Error("摘要入向量库失败,err:%v", err)
|
|
|
fmt.Println("摘要入向量库失败,err:", err)
|
|
|
}
|
|
|
+
|
|
|
+ // 数据入ES库
|
|
|
+ go AddOrEditEsWechatArticleAbstract(abstractItem.WechatArticleAbstractId)
|
|
|
}()
|
|
|
|
|
|
// 生成临时文件
|
|
@@ -644,7 +647,7 @@ func AbstractToKnowledge(wechatArticleItem *rag.WechatArticle, item *rag.WechatA
|
|
|
}
|
|
|
fileName := utils.RemoveSpecialChars(wechatArticleItem.Title) + `.md`
|
|
|
tmpFilePath := uploadDir + "/" + fileName
|
|
|
- err = utils.SaveToFile(item.Content, tmpFilePath)
|
|
|
+ err = utils.SaveToFile(abstractItem.Content, tmpFilePath)
|
|
|
if err != nil {
|
|
|
err = fmt.Errorf("生成临时文件失败,Err:" + err.Error())
|
|
|
return
|
|
@@ -667,9 +670,9 @@ func AbstractToKnowledge(wechatArticleItem *rag.WechatArticle, item *rag.WechatA
|
|
|
}
|
|
|
}
|
|
|
|
|
|
- item.VectorKey = tmpFilePath
|
|
|
- item.ModifyTime = time.Now()
|
|
|
- err = item.Update([]string{"vector_key", "modify_time"})
|
|
|
+ abstractItem.VectorKey = tmpFilePath
|
|
|
+ abstractItem.ModifyTime = time.Now()
|
|
|
+ err = abstractItem.Update([]string{"vector_key", "modify_time"})
|
|
|
|
|
|
}
|
|
|
|