DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

refactor(module-llm):优化 HTTP 请求和响应的日志记录

Browse Source 
- 移除重复的日志记录代码
- 添加请求耗时的记录
- 优化日志输出格式
- 封装日志打印逻辑到单独的方法中
This commit is contained in:
Liuyang 2025-02-21 17:05:39 +08:00
parent 241cefdac9
commit 758b00fefb
1 changed files with 33 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/RagHttpService.java
View File

@ -167,33 +167,22 @@ public class RagHttpService {
String curlCommand = String.format("curl -X POST -F \"file_id=%s\" -F \"file=@%s\" \"%s\"", ragUploadReqVO.getFileId(), ragUploadReqVO.getFileName(), ragEmbed);
log.info("生成的 curl 命令: {}", curlCommand);
long startTime = System.currentTimeMillis(); // 记录开始时间
String body = HttpRequest.post(ragUploadReqVO.getUrl())
.form("file", result, ragUploadReqVO.getFileName())
.form("file_id", ragUploadReqVO.getFileId())
.timeout(60000)
.executeAsync().body();
// 打印响应内容
log.info("===============================响应成功===============================");
log.info("===============================响应成功===============================");
log.info("===============================响应成功===============================");
log.info("===============================响应成功===============================");
log.info("===============================响应成功===============================");
long endTime = System.currentTimeMillis();
long duration = endTime - startTime;
printLogs();
log.info("请求耗时: {}", formatDuration(duration));
log.info("响应原始内容: {}", body);
log.info("===============================响应成功===============================");
log.info("===============================响应成功===============================");
log.info("===============================响应成功===============================");
log.info("===============================响应成功===============================");
log.info("===============================响应成功===============================");
// // 发送上传请求
// HttpResponse<String> uploadResponse = Unirest.post(ragUploadReqVO.getUrl())
// .field("file_id", ragUploadReqVO.getFileId())
// .field("file", new ByteArrayInputStream(utf8Bytes), ragUploadReqVO.getFileName())
// .asString();
printLogs();
// 检查响应状态
// log.info("Response Body: {}", uploadResponse.getBody());
ragEmbedRespVO = JSON.parseObject(body, RagEmbedRespVO.class);
log.info("ragEmbedRespVO:{}", ragEmbedRespVO);
@ -217,6 +206,20 @@ public class RagHttpService {
}
}
public void printLogs(){
for (int i = 0; i < 5; i++){
log.info("===============================响应成功===============================");
}
}
public static String formatDuration(long durationMillis) {
long minutes = durationMillis / 60000;
long seconds = (durationMillis % 60000) / 1000;
long millis = durationMillis % 1000;
return String.format("%d分%d秒%d毫秒", minutes, seconds, millis);
}
private static String detectCharset (InputStream inputStream) throws IOException {
byte[] buffer = new byte[4096];
int nread;
@ -364,6 +367,9 @@ public class RagHttpService {
)
.build();
// 记录开始时间
long startTime = System.currentTimeMillis();
// 创建请求
Request sendRequest = new Request.Builder()
.url(ragEmbed)
@ -375,17 +381,15 @@ public class RagHttpService {
try (Response sendResponse = client.newCall(sendRequest).execute()) {
if (sendResponse.body() != null) {
String body = sendResponse.body().string();
log.info("===============================响应成功===============================");
log.info("===============================响应成功===============================");
log.info("===============================响应成功===============================");
log.info("===============================响应成功===============================");
log.info("===============================响应成功===============================");
// 打印响应内容
long endTime = System.currentTimeMillis();
long duration = endTime - startTime;
printLogs();
log.info("请求耗时: {}", formatDuration(duration));
log.info("!!!!!!!!!! 响应原始内容 Response: {}", body);
log.info("===============================响应成功===============================");
log.info("===============================响应成功===============================");
log.info("===============================响应成功===============================");
log.info("===============================响应成功===============================");
log.info("===============================响应成功===============================");
printLogs();
JSONObject resJson = JSONObject.parseObject(body);
// 1: 先判断是否存在 detail，