From 5b97d28babc5bda2c7e4755a1e07dea4ca1d5000 Mon Sep 17 00:00:00 2001 From: teddy Date: Mon, 12 Jan 2026 14:08:19 +0900 Subject: [PATCH] =?UTF-8?q?[KC-108]=20=EB=B6=84=EC=84=9D=EB=8F=84=EC=97=BD?= =?UTF-8?q?=20ai=20api=ED=98=B8=EC=B6=9C=20=ED=85=8C=EC=8A=A4=ED=8A=B8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../service/InferenceResultService.java | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) diff --git a/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java b/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java index b4558d1d..5c44b513 100644 --- a/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java +++ b/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java @@ -148,14 +148,24 @@ public class InferenceResultService { cdClsModelPath = modelInfo.getClsModelPath() + "/" + modelInfo.getClsModelFileName(); } + String modelType = ""; + + if (modelInfo.getModelType().equals("M1")) { + modelType = "G1"; + } else if (modelInfo.getModelType().equals("M2")) { + modelType = "G2"; + } else { + modelType = "G3"; + } + InferenceSendDto sendDto = new InferenceSendDto(); sendDto.setModel_version(modelInfo.getModelVer()); sendDto.setCd_model_path(cdModelPath); sendDto.setCd_model_config(cdModelConfigPath); sendDto.setCls_model_path(cdClsModelPath); - // sendDto.setCls_model_version(modelInfo.getCls); - sendDto.setCd_model_type(modelInfo.getModelType()); - // sendDto.setPriority(modelInfo.getPriority()); + sendDto.setCls_model_version(modelInfo.getModelVer()); + sendDto.setCd_model_type(modelType); + sendDto.setPriority(modelInfo.getPriority().intValue()); return sendDto; }