小bug

application-webadmin/src/main/java/com/yice/webadmin/app/service/impl/TuningRunServiceImpl.java

@@ -33,6 +33,7 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import java.io.File;
 import java.net.URI;
 import java.util.List;
 
@@ -59,7 +60,7 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement
     private IdGeneratorWrapper idGenerator;
     @Autowired
     private PythonConfig pythonConfig;
-    private String modelName;
+    private String modelVersionURl;
 
     /**
      * 返回当前Service的主表Mapper对象。
@@ -197,14 +198,18 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement
             modelManageS.setModelDescribe(runPublishDto.getModelDescribe());
             modelManageS.setModelName(runPublishDto.getModelName());
             modelManageS.setModelType(runPublishDto.getModelType());
+            modelVersionURl = pythonConfig.getModelOutputFileBaseDir() + runPublishDto.getModelName() + "_V1";
+            modelVersionS.setModelUrl(modelVersionURl);
             this.modelManageService.saveAndCreateVersion(modelManageS, modelVersionS);
-            modelName = runPublishDto.getModelName() + "_V1";
         } else {
             modelVersionS.setModelId(runPublishDto.getModelId());
             ModelManage modelManageS = this.modelManageService.getById(runPublishDto.getModelId());
             modelManageS.setModelDescribe(runPublishDto.getModelDescribe());
             this.modelManageService.updateById(modelManageS);
-            modelName = this.modelVersionService.saveNew(modelVersionS).getVersionName();
+            ModelVersion modelVersionR = this.modelVersionService.saveNew(modelVersionS);
+            modelVersionURl = pythonConfig.getModelOutputFileBaseDir() + modelVersionR.getVersionName();
+            modelVersionR.setModelUrl(modelVersionURl);
+            this.modelVersionService.updateById(modelVersionR);
         }
         new WebSocketClient(new URI(this.pythonConfig.getPythonWebsocketUri()), new Draft_6455()) {
             @Override
@@ -237,7 +242,7 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement
                     array.add(tuningRun.getTrainMethod());
                     array.add(jsonObject.get("promptTemplate"));
                     array.add(2);
-                    String newModelUrl = pythonConfig.getModelOutputFileBaseDir() + modelName;
+                    String newModelUrl = modelVersionURl;
                     array.add(newModelUrl);
                     array.add("none");
                     sendJson.put("data", array);