src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java

@@ -120,6 +120,15 @@ public class ModelTrainMngCoreService {
     entity.setTrainType(addReq.getTrainType()); // 일반, 전이
     entity.setBeforeModelId(addReq.getBeforeModelId());
 
+    // 전이학습일 때 beforeModelUuid 추가하기
+    if (addReq.getBeforeModelId() != null) {
+      ModelMasterEntity beforeModel =
+          modelMngRepository
+              .findById(addReq.getBeforeModelId())
+              .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
+      entity.setBeforeModelUuid(beforeModel.getUuid());
+    }
+
     entity.setStatusCd(TrainStatusType.READY.getId());
     entity.setStep1State(TrainStatusType.READY.getId());