From 707d3981db424b7386136b872d8216afd58245c9 Mon Sep 17 00:00:00 2001
From: limin <limin@qq.com>
Date: Tue, 18 Feb 2025 13:36:58 +0800
Subject: [PATCH] =?UTF-8?q?=E9=87=8D=E6=96=B0=E8=B0=83=E8=AF=95=20?=
 =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=97=A5=E5=BF=97=E8=BF=94=E5=9B=9E=E6=8E=A5?=
 =?UTF-8?q?=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/finetuningtask/FineTuningTaskServiceImpl.java   | 6 +++++-
 .../yudao/module/llm/service/http/TrainHttpService.java     | 4 ++--
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java
index 49580ae00..3e9854711 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java
@@ -316,7 +316,11 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService {
             String query = "?log_path="+fineTuningTaskDO.getTrainLog();
             Map<String, String> map = new HashMap<>();
             map.put("log_path", fineTuningTaskDO.getTrainDuration());
-            return trainHttpService.getTrainLog(query);
+            ServerNameDO serverNameDO = serverNameMapper.selectById(fineTuningTaskDO.getGpuType());
+            if (serverNameDO == null){
+                return "";
+            }
+            return trainHttpService.getTrainLog(serverNameDO.getHost(),query);
         }
         return null;
     }
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java
index c5b29eac0..1e86a4386 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java
@@ -332,9 +332,9 @@ public class TrainHttpService {
 
     }
 
-    public String getTrainLog(String params) {
+    public String getTrainLog(String url,String params) {
         String finetuningLog = llmBackendProperties.getFinetuningLog();
-        return HttpUtils.get(finetuningLog+params, null);
+        return HttpUtils.get(url+finetuningLog+params, null);
     }
 
     public String getCheckFileList(String name){