Roc пре 1 месец
родитељ
комит
24c4fae049
3 измењених фајлова са 30 додато и 0 уклоњено
  1. 21 0
      cache/wechat_platform.go
  2. 6 0
      services/llm/wechat_platform.go
  3. 3 0
      services/task.go

+ 21 - 0
cache/wechat_platform.go

@@ -26,3 +26,24 @@ func AddWechatArticleOpToCache(wechatPlatformId int, source string) bool {
 	}
 	return false
 }
+
+// AddWechatArticleLlmOpToCache
+// @Description: 将公众号文章llm操作加入缓存
+// @param wechatPlatformId
+// @param source
+// @return bool
+func AddWechatArticleLlmOpToCache(wechatPlatformId int, source string) bool {
+	record := new(llm.WechatArticleOp)
+	record.Source = source
+	record.WechatPlatformId = wechatPlatformId
+	if utils.Re == nil {
+		err := utils.Rc.LPush(utils.CACHE_WECHAT_PLATFORM_ARTICLE_KNOWLEDGE, record)
+
+		utils.FileLog.Info(fmt.Sprintf("将公众号文章llm操作加入缓存 加入缓存 AddWechatArticleLlmOpToCache LPush: 操作类型:%s,公众号id:%d", source, wechatPlatformId))
+		if err != nil {
+			fmt.Println("AddWechatArticleOpToCache LPush Err:" + err.Error())
+		}
+		return true
+	}
+	return false
+}

+ 6 - 0
services/llm/wechat_platform.go

@@ -1,6 +1,7 @@
 package llm
 
 import (
+	"eta/eta_api/cache"
 	"eta/eta_api/models"
 	"eta/eta_api/models/rag"
 	"eta/eta_api/utils"
@@ -161,6 +162,10 @@ func AddWechatArticle(item *rag.WechatPlatform, articleLink string, articleDetai
 		obj.Description = articleMenu.Digest
 	}
 	err = obj.Create()
+
+	// 文章入库成功后,需要将相关信息入摘要库
+	go cache.AddWechatArticleLlmOpToCache(item.WechatPlatformId, ``)
+
 }
 
 // BeachAddWechatArticle
@@ -210,6 +215,7 @@ func BeachAddWechatArticle(item *rag.WechatPlatform, num int) {
 		AddWechatArticle(item, articleMenu.Link, articleDetail, &articleMenu)
 
 		time.Sleep(10 * time.Second)
+
 	}
 	return
 }

+ 3 - 0
services/task.go

@@ -69,6 +69,9 @@ func Task() {
 	// 定时任务进行微信文章操作
 	go HandleWechatArticleOp()
 
+	// 定时任务进行微信文章LLM操作
+	go HandleWechatArticleLLmOp()
+
 	// TODO:数据修复
 	//FixNewEs()
 	fmt.Println("task end")