优质内容生成计划

long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/remote/aigc/AIGCProducePlanSaveService.java

@@ -2,6 +2,7 @@ package com.tzld.longarticle.recommend.server.remote.aigc;
 
 import com.alibaba.fastjson.JSONObject;
 import com.tzld.longarticle.recommend.server.common.HttpPoolFactory;
+import com.tzld.longarticle.recommend.server.common.enums.aigc.ProducePlanInputSourceTypeEnum;
 import com.tzld.longarticle.recommend.server.model.vo.IdNameVO;
 import com.tzld.longarticle.recommend.server.model.vo.aigc.ProducePlanDetailVO;
 import com.tzld.longarticle.recommend.server.model.vo.aigc.ProducePlanInputSourceParam;
@@ -39,6 +40,10 @@ public class AIGCProducePlanSaveService {
         newInputSource.setInputSourceModal(3);
         newInputSource.setInputSourceChannel(5);
         inputSources.add(newInputSource);
+        if (Objects.equals(inputSourceType, ProducePlanInputSourceTypeEnum.contentID.getVal())) {
+            detail.setDailyProduceNum(inputSources.size());
+            detail.setTotalProduceNum(inputSources.size());
+        }
 
         // 准备请求参数
         JSONObject payload = new JSONObject();