Browse Source

kimi失败内容进行重试

wangyunpeng 8 tháng trước cách đây
mục cha
commit
d8ae6287ca

+ 4 - 4
long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/service/recommend/ArticleService.java

@@ -279,7 +279,7 @@ public class ArticleService {
             // 溯源
             Article article = articleRepository.getByWxSn(task.getWxSn());
             if (Objects.isNull(article)) {
-                task.setDeleted(StatusEnum.SUCCESS.getCode());
+                task.setDeleted(StatusEnum.ONE.getCode());
                 articlePoolPromotionSourceRepository.save(task);
                 continue;
             }
@@ -291,7 +291,7 @@ public class ArticleService {
             List<PublishContent> publishContentList = aigcBaseMapper.getNearestPublishContent(publishAccount.getId(), publishTimestamp, 100);
             PublishContent publishContent = findPublishContent(publishContentList, task.getTitle(), publishTimestamp);
             if (Objects.isNull(publishContent)) {
-                task.setDeleted(StatusEnum.SUCCESS.getCode());
+                task.setDeleted(StatusEnum.ONE.getCode());
                 articlePoolPromotionSourceRepository.save(task);
                 continue;
             }
@@ -384,7 +384,7 @@ public class ArticleService {
             for (ArticleCategory articleCategory : partition) {
                 articleCategory.setKimiResult(kimiResult.getResponseStr());
                 articleCategory.setUpdateTimestamp(now);
-                if (kimiResult.isSuccess() && Objects.nonNull(obj)) {
+                if (kimiResult.isSuccess() && Objects.nonNull(obj) && obj.containsKey(articleCategory.getTitle())) {
                     articleCategory.setCategory(obj.getString(articleCategory.getTitle()));
                     articleCategory.setStatus(ArticleCategoryStatusEnum.SUCCESS.getCode());
                 } else {
@@ -455,7 +455,7 @@ public class ArticleService {
             articleCategory.setKimiResult(kimiResult.getResponseStr());
             articleCategory.setUpdateTimestamp(now);
             articleCategory.setRetryTimes(articleCategory.getRetryTimes() + 1);
-            if (kimiResult.isSuccess() && Objects.nonNull(obj)) {
+            if (kimiResult.isSuccess() && Objects.nonNull(obj) && obj.containsKey(articleCategory.getTitle())) {
                 articleCategory.setCategory(obj.getString(articleCategory.getTitle()));
                 articleCategory.setStatus(ArticleCategoryStatusEnum.SUCCESS.getCode());
                 articleCategory.setFailReason(null);