diff --git a/src/main/java/com/ai/da/service/CreditsService.java b/src/main/java/com/ai/da/service/CreditsService.java index ab8ac126..aba4d027 100644 --- a/src/main/java/com/ai/da/service/CreditsService.java +++ b/src/main/java/com/ai/da/service/CreditsService.java @@ -33,6 +33,8 @@ public interface CreditsService extends IService { void addRecordToCreditsDeduction(Long accountId, String taskId, CreditsEventsEnum creditsEventsEnum); + void addRecordToCreditsDeduction(Long accountId, String taskId, CreditsEventsEnum creditsEventsEnum, Integer num); + Boolean taskCreditsDeduction(Long accountId, String taskId); CreditsDetail getByAccountIdAndChangeEvent(Long accountId, String changeEvent, String changedCredits); diff --git a/src/main/java/com/ai/da/service/impl/CreditsServiceImpl.java b/src/main/java/com/ai/da/service/impl/CreditsServiceImpl.java index b6ca3287..3f2c57e6 100644 --- a/src/main/java/com/ai/da/service/impl/CreditsServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/CreditsServiceImpl.java @@ -266,6 +266,12 @@ public class CreditsServiceImpl extends ServiceImpl impleme toProductImageResultMapper.insert(toProductImageResult); result.add(toProductImageResult); // 添加需要扣除的积分到预扣除区 - creditsService.addRecordToCreditsDeduction(userHolder.getId(), taskId, CreditsEventsEnum.TO_PRODUCT_IMAGE); + creditsService.addRecordToCreditsDeduction(userHolder.getId(), taskId, CreditsEventsEnum.TO_PRODUCT_IMAGE, cloudTaskDTO.getNums()); i ++; } pythonService.toProductImageBatch(batchTaskId, paramList, userHolder.getId().toString()); @@ -2083,7 +2083,7 @@ public class DesignServiceImpl extends ServiceImpl impleme toProductImageResultMapper.insert(toProductImageResult); result.add(toProductImageResult); // 添加需要扣除的积分到预扣除区 - creditsService.addRecordToCreditsDeduction(userHolder.getId(), taskId, CreditsEventsEnum.TO_PRODUCT_IMAGE); + creditsService.addRecordToCreditsDeduction(userHolder.getId(), taskId, CreditsEventsEnum.TO_PRODUCT_IMAGE, cloudTaskDTO.getNums()); i ++; } pythonService.toProductImageBatch(batchTaskId, paramList, userHolder.getId().toString()); @@ -2208,7 +2208,7 @@ public class DesignServiceImpl extends ServiceImpl impleme result.add(toProductImageResult); // 添加需要扣除的积分到预扣除区 - creditsService.addRecordToCreditsDeduction(userHolder.getId(), taskId, CreditsEventsEnum.RELIGHT); + creditsService.addRecordToCreditsDeduction(userHolder.getId(), taskId, CreditsEventsEnum.RELIGHT, cloudTaskDTO.getNums()); i ++; } // 走模型 @@ -2251,7 +2251,7 @@ public class DesignServiceImpl extends ServiceImpl impleme result.add(toProductImageResult); // 添加需要扣除的积分到预扣除区 - creditsService.addRecordToCreditsDeduction(userHolder.getId(), taskId, CreditsEventsEnum.RELIGHT); + creditsService.addRecordToCreditsDeduction(userHolder.getId(), taskId, CreditsEventsEnum.RELIGHT, cloudTaskDTO.getNums()); i ++; } // 走模型 @@ -2325,7 +2325,7 @@ public class DesignServiceImpl extends ServiceImpl impleme Boolean b = pythonService.poseTransformationBatch(poseTransformBatchDTO.getProductImage(), poseTransformBatchDTO.getPoseId(), taskId); if (b){ // 6、添加预扣除积分到redis - creditsService.addRecordToCreditsDeduction(accountId, uuid, creditsEventsEnum); + creditsService.addRecordToCreditsDeduction(accountId, uuid, creditsEventsEnum, cloudTaskDTO.getNums()); // 6.1 添加积分扣除记录到db creditsService.preInsert(accountId, creditsEventsEnum.getName(), uuid, Boolean.TRUE, null); }else { diff --git a/src/main/java/com/ai/da/service/impl/LLMServiceImpl.java b/src/main/java/com/ai/da/service/impl/LLMServiceImpl.java index 8bb8e980..e87c2291 100644 --- a/src/main/java/com/ai/da/service/impl/LLMServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/LLMServiceImpl.java @@ -353,6 +353,7 @@ public class LLMServiceImpl implements LLMService { chatMessageMapper.insert(systemMessage); systemMessage.setId(null); responseContentBuilder = new StringBuilder(); + contentType = null; } String toolsName = json.getString("tools_name"); JSONObject toolsData = json.getJSONObject("content");