From 7013594161528b5a1157bc59f9beb2a600aed8e5 Mon Sep 17 00:00:00 2001
From: Liuyang <2746366019@qq.com>
Date: Wed, 12 Feb 2025 15:46:10 +0800
Subject: [PATCH] =?UTF-8?q?refactor(yudao-module-llm):=20=E8=B0=83?=
 =?UTF-8?q?=E6=95=B4=E4=B8=8A=E4=BC=A0=E6=96=87=E4=BB=B6=E7=9A=84=20socket?=
 =?UTF-8?q?=20=E8=B6=85=E6=97=B6=E6=97=B6=E9=97=B4-=20=E5=B0=86=20Unirest?=
 =?UTF-8?q?=20=E9=85=8D=E7=BD=AE=E4=B8=AD=E7=9A=84=20socketTimeout=20?=
 =?UTF-8?q?=E4=BB=8E=2060000=20=E6=AF=AB=E7=A7=92=EF=BC=881=E5=88=86?=
 =?UTF-8?q?=E9=92=9F=EF=BC=89=E5=A2=9E=E5=8A=A0=E5=88=B0=2086400000?=
 =?UTF-8?q?=E6=AF=AB=E7=A7=92=EF=BC=8824=E5=B0=8F=E6=97=B6=EF=BC=89-=20?=
 =?UTF-8?q?=E6=AD=A4=E4=BF=AE=E6=94=B9=E5=8F=AF=E8=83=BD=E6=98=AF=E4=B8=BA?=
 =?UTF-8?q?=E4=BA=86=E9=80=82=E5=BA=94=E5=A4=A7=E6=96=87=E4=BB=B6=E4=B8=8A?=
 =?UTF-8?q?=E4=BC=A0=E6=88=96=E7=BD=91=E7=BB=9C=E6=9D=A1=E4=BB=B6=E8=BE=83?=
 =?UTF-8?q?=E5=B7=AE=E7=9A=84=E5=9C=BA=E6=99=AF=EF=BC=8C=E7=BB=99=E4=BA=88?=
 =?UTF-8?q?=E4=B8=8A=E4=BC=A0=E8=BF=87=E7=A8=8B=E6=9B=B4=E5=A4=9A=E7=9A=84?=
 =?UTF-8?q?=E6=97=B6=E9=97=B4=E5=AE=8C=E6=88=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../yudao/module/llm/service/http/RagHttpService.java       | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

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 ead8e0dc4..166533df7 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
@@ -117,7 +117,7 @@ public class RagHttpService {
 
                         // 上传文件
                         Unirest.config().reset();
-                        Unirest.config().socketTimeout(60000);
+                        Unirest.config().socketTimeout(86400000);
                         HttpResponse<String> uploadResponse = Unirest.post(ragUploadReqVO.getUrl())
                                 .field("file_id", ragUploadReqVO.getFileId())
                                 .field("file", new ByteArrayInputStream(utf8Bytes), ragUploadReqVO.getFileName()) // 使用文件名 "file.txt" 作为示例
@@ -133,7 +133,7 @@ public class RagHttpService {
                         } else {
                             // 修改状态为 上传失败
                             updateFileState(documents, KnowledgeStatusEnum.UPLOAD_FAILED);
-                            throw new RuntimeException("文件上传失败:"+ragEmbedRespVO.getMessage());
+                            throw new RuntimeException("文件上传失败:" + ragEmbedRespVO.getMessage());
                         }
                     }
                 }
@@ -273,7 +273,7 @@ public class RagHttpService {
         JSONObject parseObject = JSON.parseObject(responseBody);
         log.info(" ========= Response Body Result: {}", responseBody);
 
-        if (parseObject.getBoolean("status")) {
+        if (parseObject.containsKey("status") && parseObject.getBoolean("status")) {
             // 修改状态为 上传成功
             updateFileState(documents, KnowledgeStatusEnum.UPLOAD_SUCCESS);
         } else {