From 26749e9e730b71a7a8d90ca41ed73c965436e829 Mon Sep 17 00:00:00 2001
From: chenzhihang <709011834@qq.com>
Date: Thu, 17 Oct 2024 17:05:44 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E6=9F=A5=E8=AF=A2=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E7=89=88=E6=9C=AC=E6=8C=87=E6=A0=87=E5=AF=B9=E6=AF=94?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/impl/ModelsServiceImpl.java       | 58 +++++++++++--------
 1 file changed, 33 insertions(+), 25 deletions(-)

diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java
index 216824f7..da21bf35 100644
--- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java
+++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java
@@ -986,34 +986,42 @@ public class ModelsServiceImpl implements ModelsService {
             String meta = modelDependency1Dao.getMeta(identifier, owner, (String) branch.get("name"));
             if (StringUtils.isNotEmpty(meta)) {
                 ModelMetaVo modelMetaVo = JSON.parseObject(meta, ModelMetaVo.class);
-                if (modelMetaVo.getParams() != null) {
-                    HashMap<String, Object> params = modelMetaVo.getParams();
-                    branch.put("params", params);
+                if (Constant.Source_Add.equals(modelMetaVo.getModelSource())) {
+                    branch.put("params", null);
+                    branch.put("params_names", new ArrayList<>());
+                    branch.put("metrics", null);
+                    branch.put("metrics_names", new ArrayList<>());
+                } else {
+                    if (modelMetaVo.getParams() != null) {
+                        HashMap<String, Object> params = modelMetaVo.getParams();
+                        branch.put("params", params);
 
-                    ArrayList<String> params_names = new ArrayList<>();
-                    for (String key : params.keySet()) {
-                        params_names.add(key);
-                    }
-                    branch.put("params_names", params_names);
-                }
-                if (modelMetaVo.getMetrics() != null) {
-                    HashMap<String, Object> metrics = modelMetaVo.getMetrics();
-                    if (Constant.Type_Train.equals(type)) {
-                        Map<String, Object> trainMetrics = (Map<String, Object>) metrics.get(Constant.Type_Train);
-                        ArrayList<String> metrics_names = new ArrayList<>();
-                        for (String key : trainMetrics.keySet()) {
-                            metrics_names.add(key);
+                        ArrayList<String> params_names = new ArrayList<>();
+                        for (String key : params.keySet()) {
+                            params_names.add(key);
                         }
-                        branch.put("metrics_names", metrics_names);
-                        branch.put("metrics",trainMetrics);
-                    } else {
-                        Map<String, Object> evaluateMetrics = (Map<String, Object>) metrics.get(Constant.Type_Evaluate);
-                        ArrayList<String> metrics_names = new ArrayList<>();
-                        for (String key : evaluateMetrics.keySet()) {
-                            metrics_names.add(key);
+                        branch.put("params_names", params_names);
+                    }
+                    if (modelMetaVo.getMetrics() != null) {
+                        HashMap<String, Object> metrics = modelMetaVo.getMetrics();
+                        if (Constant.Type_Train.equals(type)) {
+                            Map<String, Object> trainMetrics = (Map<String, Object>) metrics.get(Constant.Type_Train);
+                            branch.put("metrics", trainMetrics);
+
+                            ArrayList<String> metrics_names = new ArrayList<>();
+                            for (String key : trainMetrics.keySet()) {
+                                metrics_names.add(key);
+                            }
+                            branch.put("metrics_names", metrics_names);
+                        } else {
+                            Map<String, Object> evaluateMetrics = (Map<String, Object>) metrics.get(Constant.Type_Evaluate);
+                            ArrayList<String> metrics_names = new ArrayList<>();
+                            for (String key : evaluateMetrics.keySet()) {
+                                metrics_names.add(key);
+                            }
+                            branch.put("metrics_names", metrics_names);
+                            branch.put("metrics", evaluateMetrics);
                         }
-                        branch.put("metrics_names", metrics_names);
-                        branch.put("metrics",evaluateMetrics);
                     }
                 }
             }