Roc il y a 2 semaines
Parent
commit
55e2f455e6
1 fichiers modifiés avec 11 ajouts et 8 suppressions
  1. 11 8
      services/wechat_platform.go

+ 11 - 8
services/wechat_platform.go

@@ -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"})
 
 }