From ba57e373fe75362c67f53520abef2c62ecac1d75 Mon Sep 17 00:00:00 2001
From: Liuyang <2746366019@qq.com>
Date: Sat, 8 Feb 2025 17:46:25 +0800
Subject: [PATCH] =?UTF-8?q?refactor(module-llm):=E7=A7=BB=E9=99=A4?=
 =?UTF-8?q?=E7=9F=A5=E8=AF=86=E5=BA=93=E5=B5=8C=E5=85=A5=E4=B8=AD=E7=9A=84?=
 =?UTF-8?q?=E7=94=A8=E6=88=B7ID=E5=AD=97=E6=AE=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 从 AsyncKnowledgeBase 类中移除了 getEntityId 方法
- 从 KnowledgeRagEmbedReqVO 类中移除了 entityId 字段
- 从 RagHttpService 的 knowledgeEmbed 方法调用中移除了 entityId 参数
---
 .../module/llm/service/async/AsyncKnowledgeBase.java | 12 +-----------
 .../module/llm/service/http/RagHttpService.java      |  1 -
 .../llm/service/http/vo/KnowledgeRagEmbedReqVO.java  |  5 -----
 3 files changed, 1 insertion(+), 17 deletions(-)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncKnowledgeBase.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncKnowledgeBase.java
index 521b2bd43..9989d255b 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncKnowledgeBase.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncKnowledgeBase.java
@@ -80,8 +80,7 @@ public class AsyncKnowledgeBase {
                     KnowledgeRagEmbedReqVO ragEmbedReqVo = new KnowledgeRagEmbedReqVO()
                             .setFileId(String.valueOf(knowledge.getId()))
                             .setFileName(knowledge.getDocumentName())
-                            .setFileInputStream(new ByteArrayInputStream(Objects.requireNonNull(getFileByte(knowledge.getFileUrl()))))
-                            .setEntityId(String.valueOf(getEntityId()));
+                            .setFileInputStream(new ByteArrayInputStream(Objects.requireNonNull(getFileByte(knowledge.getFileUrl()))));
 
                     ragHttpService.knowledgeEmbed(ragEmbedReqVo);
 
@@ -92,15 +91,6 @@ public class AsyncKnowledgeBase {
         }
     }
 
-    /**
-     * 获取当前用户ID
-     *
-     * @return 用户ID
-     */
-    public Long getEntityId () {
-        return SecurityFrameworkUtils.getLoginUserId();
-    }
-
     /**
      * 获取文件字节数组
      *
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 98a48f93b..dcf6a5d75 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
@@ -215,7 +215,6 @@ public class RagHttpService {
         // 构建请求参数
         HttpResponse<String> response = Unirest.post(ragEmbed)
                 .field("file_id", reqVO.getFileId())
-                .field("entity_id", reqVO.getEntityId())
                 .field("file", reqVO.getFileInputStream(), reqVO.getFileName())
                 .asString();
 
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/vo/KnowledgeRagEmbedReqVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/vo/KnowledgeRagEmbedReqVO.java
index 7f33d9087..d10ba1dde 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/vo/KnowledgeRagEmbedReqVO.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/vo/KnowledgeRagEmbedReqVO.java
@@ -31,9 +31,4 @@ public class KnowledgeRagEmbedReqVO {
      * 文件流
      */
     private ByteArrayInputStream fileInputStream;
-
-    /**
-     * 用户Id
-     */
-    private String entityId;
 }