Roc 2 days ago
parent
commit
9be657d05c
2 changed files with 22 additions and 23 deletions
  1. 0 21
      cache/wechat_platform.go
  2. 22 2
      services/llm/wechat_platform.go

+ 0 - 21
cache/wechat_platform.go

@@ -26,24 +26,3 @@ 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
-}

+ 22 - 2
services/llm/wechat_platform.go

@@ -1,7 +1,6 @@
 package llm
 
 import (
-	"eta/eta_api/cache"
 	"eta/eta_api/models"
 	"eta/eta_api/models/rag"
 	"eta/eta_api/utils"
@@ -164,10 +163,31 @@ func AddWechatArticle(item *rag.WechatPlatform, articleLink string, articleDetai
 	err = obj.Create()
 
 	// 文章入库成功后,需要将相关信息入摘要库
-	go cache.AddWechatArticleLlmOpToCache(item.WechatPlatformId, ``)
+	go AddWechatArticleLlmOpToCache(item.WechatPlatformId, ``)
 
 }
 
+// AddWechatArticleLlmOpToCache
+// @Description: 将公众号文章llm操作加入缓存
+// @param wechatPlatformId
+// @param source
+// @return bool
+func AddWechatArticleLlmOpToCache(wechatPlatformId int, source string) bool {
+	record := new(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
+}
+
 // BeachAddWechatArticle
 // @Description: 批量添加公众号文章
 // @param item