From e49bc929e027a30014fb9875291e56afdb9e0c9b Mon Sep 17 00:00:00 2001
From: limin <limin@qq.com>
Date: Thu, 20 Feb 2025 13:39:32 +0800
Subject: [PATCH] =?UTF-8?q?refactor(yudao-module-llm):=20=E9=87=8D?=
 =?UTF-8?q?=E6=9E=84=E8=87=AA=E5=8A=A8=E8=AF=84=E4=BC=B0=E5=8A=9F=E8=83=BD?=
 =?UTF-8?q?=E4=BB=A5=E6=94=AF=E6=8C=81=E5=A4=9A=E6=A8=A1=E5=9E=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 修改 AsyncModelAccessManualService 中的 autoEvaluation 方法，增加 modelUrl 参数
- 更新 TrainHttpService 中的 autoEvaluation 方法，添加 url 参数
- 优化日志输出，增加自动响应内容的打印
---
 .../llm/service/async/AsyncModelAccessManualService.java      | 3 ++-
 .../yudao/module/llm/service/http/TrainHttpService.java       | 4 ++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelAccessManualService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelAccessManualService.java
index 772868c9b..c6ab9568d 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelAccessManualService.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelAccessManualService.java
@@ -121,8 +121,9 @@ public class AsyncModelAccessManualService {
                 modelCompletionsReqVO.setMessages(messages);
                 modelCompletionsReqVO.setModel(modelName);
                 ModelCompletionsRespVO modelCompletionsRespVO = modelService.modelCompletions(modelUrl,modelCompletionsReqVO);
+                log.info("auto response {}",modelCompletionsRespVO);
                 String prompt = modelCompletionsRespVO.getAnswer();
-                String res = trainHttpService.autoEvaluation(prompt, datasetPrompt);
+                String res = trainHttpService.autoEvaluation(modelUrl,prompt, datasetPrompt);
 
                 ModelAssessTaskAutoInfoDO infoDO = new ModelAssessTaskAutoInfoDO();
                 infoDO.setPrompt(prompt);
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 c5c254cdd..486a49539 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
@@ -211,7 +211,7 @@ public class TrainHttpService {
     }
 
 
-    public String autoEvaluation(String prompt,String dataSetPrompt) {
+    public String autoEvaluation(String url,String prompt,String dataSetPrompt) {
         Map<String,String> map = new HashMap<>();
         map.put("accept","application/json");
         map.put("Content-Type","application/json");
@@ -222,7 +222,7 @@ public class TrainHttpService {
         List<String> dimension = Stream.of("事实性错误","情感倾向性","语义连贯性").collect(Collectors.toList());
         jsonObject.put("dimension",dimension);
         String body = JSONObject.toJSONString(jsonObject);
-        String res = HttpUtils.post(autoEvaluation, map, body);
+        String res = HttpUtils.post(url+autoEvaluation, map, body);
         log.info(" autoEvaluation:{}", res);
         JSONObject resJson = JSONObject.parseObject(res);
         JSONObject response = resJson.getJSONObject("response");