Roc 4 روز پیش
والد
کامیت
cc46821554
1فایلهای تغییر یافته به همراه40 افزوده شده و 0 حذف شده
  1. 40 0
      services/task.go

+ 40 - 0
services/task.go

@@ -720,3 +720,43 @@ func HandleEtaReportKnowledgeLLmOp() {
 		})
 	}
 }
+
+// HandleEtaReportKnowledgeLLmOp
+// @Description: 处理微信文章加入知识库
+func Handle() {
+	defer func() {
+		if err := recover(); err != nil {
+			fmt.Println("[HandleWechatArticleLLmOp]", err)
+		}
+	}()
+	obj := rag.RagEtaReport{}
+	for {
+		utils.Rc.Brpop(utils.CACHE_ETA_REPORT_KNOWLEDGE_LLM, func(b []byte) {
+			wechatArticleOp := new(cache.RagEtaReportLlmOp)
+			if err := json.Unmarshal(b, &wechatArticleOp); err != nil {
+				fmt.Println("json unmarshal wrong!")
+				return
+			}
+			item, tmpErr := obj.GetById(wechatArticleOp.RagEtaReportId)
+			if tmpErr != nil {
+				// 找不到就处理失败
+				return
+			}
+
+			// 已经删除的就不做操作了
+			if item.IsDeleted == 1 {
+				return
+			}
+
+			// 未发布的就不操作了
+			if item.IsPublished != 1 {
+				return
+			}
+
+			// 文章加入到知识库
+			//ArticleToKnowledge(item)
+			// 生成摘要
+			GenerateRagEtaReportAbstract(item, true)
+		})
+	}
+}