diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/RagHttpService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/RagHttpService.java index 161051960..58a970d8f 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/RagHttpService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/RagHttpService.java @@ -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 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,