diff --git a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java index d9983f4e..f824032f 100644 --- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java @@ -721,8 +721,12 @@ public class GenerateServiceImpl extends ServiceImpl i // 创建生成任务 List taskIds = createGenerationTasks(generateDTO, config.times); - // 处理积分扣除(使用第一个任务的UUID前缀) - processCreditDeduction(generateDTO.getUserId(), taskIds.get(0).split("-")[0], config.creditsEvent); + // 截取任务id + String original = taskIds.get(0); + String taskId = original.substring(0, original.lastIndexOf("-", original.lastIndexOf("-") - 1)); + + // 处理积分扣除 + processCreditDeduction(generateDTO.getUserId(), taskId, config.creditsEvent); return new PrepareForGenerateVO(taskIds, 200); }