From 53c1f4d84be9910d96dd796f67e4f01796d573e0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=B0=A2=E6=AC=A3=E6=82=A6?= <2289014031@qq.com>
Date: Tue, 25 Feb 2025 17:40:02 +0800
Subject: [PATCH] 111

---
 .../aliyun/bailian/data/UploadDataManagementMode.java     | 7 +------
 .../aliyun/bailian/data/fixed/UploadLeaseParamsMode.java  | 4 ++++
 .../aliyun/controller/KnowledgeBaseUploadController.java  | 4 ++--
 .../muyu/aliyun/service/KnowledgeBaseUploadService.java   | 2 +-
 .../service/impl/KnowledgeBaseUploadServiceImpl.java      | 8 ++++++--
 5 files changed, 14 insertions(+), 11 deletions(-)

diff --git a/src/main/java/com/muyu/aliyun/bailian/data/UploadDataManagementMode.java b/src/main/java/com/muyu/aliyun/bailian/data/UploadDataManagementMode.java
index 77d5c00..eaa496d 100644
--- a/src/main/java/com/muyu/aliyun/bailian/data/UploadDataManagementMode.java
+++ b/src/main/java/com/muyu/aliyun/bailian/data/UploadDataManagementMode.java
@@ -19,12 +19,7 @@ public class UploadDataManagementMode {
      * 用于在系统中唯一标识一个工作空间，便于在不同模块和功能中引用和操作特定的工作空间
      */
     private String workspaceId;
-    /**
-     * 租约的标识符
-     *
-     * 用于在系统中唯一标识一个租约，便于在不同模块和功能中引用和操作特定的租约
-     */
-    private String LeaseId;
+
     /**
      * 区域ID，用于标识不同的区域
      */
diff --git a/src/main/java/com/muyu/aliyun/bailian/data/fixed/UploadLeaseParamsMode.java b/src/main/java/com/muyu/aliyun/bailian/data/fixed/UploadLeaseParamsMode.java
index a4fb340..7842d20 100644
--- a/src/main/java/com/muyu/aliyun/bailian/data/fixed/UploadLeaseParamsMode.java
+++ b/src/main/java/com/muyu/aliyun/bailian/data/fixed/UploadLeaseParamsMode.java
@@ -28,4 +28,8 @@ public class UploadLeaseParamsMode {
      * 存储HTTP请求的目标URL，即请求的地址
      */
     private String Url;
+
+    private String LeaseId;
+
+
 }
diff --git a/src/main/java/com/muyu/aliyun/controller/KnowledgeBaseUploadController.java b/src/main/java/com/muyu/aliyun/controller/KnowledgeBaseUploadController.java
index 213ce94..29eda40 100644
--- a/src/main/java/com/muyu/aliyun/controller/KnowledgeBaseUploadController.java
+++ b/src/main/java/com/muyu/aliyun/controller/KnowledgeBaseUploadController.java
@@ -13,9 +13,9 @@ public class KnowledgeBaseUploadController {
     private KnowledgeBaseUploadService knowledgeBaseUploadService;
 
     @PostMapping("/KnowledgeBaseUpload")
-    public void uploadFile(@RequestParam("file") MultipartFile file) {
+    public String uploadFile(@RequestParam("file") MultipartFile file) {
         String workspaceId = "llm-98lxu358qfvxdxvt";
         String categoryId = "cate_997567e8a7e94f4caab94e78c580adb4_10062067";
-        knowledgeBaseUploadService.uploadDocument(file, workspaceId, categoryId);
+        return knowledgeBaseUploadService.uploadDocument(file, workspaceId, categoryId);
     }
 }
diff --git a/src/main/java/com/muyu/aliyun/service/KnowledgeBaseUploadService.java b/src/main/java/com/muyu/aliyun/service/KnowledgeBaseUploadService.java
index 876129c..57b2fdc 100644
--- a/src/main/java/com/muyu/aliyun/service/KnowledgeBaseUploadService.java
+++ b/src/main/java/com/muyu/aliyun/service/KnowledgeBaseUploadService.java
@@ -3,6 +3,6 @@ package com.muyu.aliyun.service;
 import org.springframework.web.multipart.MultipartFile;
 
 public interface KnowledgeBaseUploadService {
-   void uploadDocument(MultipartFile file, String workspaceId, String categoryId);
+   String uploadDocument(MultipartFile file, String workspaceId, String categoryId);
 
 }
diff --git a/src/main/java/com/muyu/aliyun/service/impl/KnowledgeBaseUploadServiceImpl.java b/src/main/java/com/muyu/aliyun/service/impl/KnowledgeBaseUploadServiceImpl.java
index 731622e..8b53a39 100644
--- a/src/main/java/com/muyu/aliyun/service/impl/KnowledgeBaseUploadServiceImpl.java
+++ b/src/main/java/com/muyu/aliyun/service/impl/KnowledgeBaseUploadServiceImpl.java
@@ -49,7 +49,7 @@ public class KnowledgeBaseUploadServiceImpl implements KnowledgeBaseUploadServic
     private ReviewResolution reviewResolution;
 
     @Override
-    public void uploadDocument(MultipartFile file, String workspaceId, String categoryId) {
+    public String uploadDocument(MultipartFile file, String workspaceId, String categoryId) {
         try {
             // 1. 保存临时文件
             File tempFile = convertMultipartFileToFile(file);
@@ -67,6 +67,8 @@ public class KnowledgeBaseUploadServiceImpl implements KnowledgeBaseUploadServic
             fileUploadLeaseModel.setMd5(md5);
             fileUploadLeaseModel.setSizeInBytes(String.valueOf(sizeInBytes));
 
+
+
             UploadLeaseParamsMode leaseParams = applyLease.applyFileUploadLease(fileUploadLeaseModel);
             if (leaseParams == null) {
                 throw new RuntimeException("申请文档上传租约失败");
@@ -87,7 +89,7 @@ public class KnowledgeBaseUploadServiceImpl implements KnowledgeBaseUploadServic
             UploadDataManagementMode uploadDataManagementMode = new UploadDataManagementMode();
             uploadDataManagementMode.setWorkspaceId(workspaceId);
             // 这里需要设置 LeaseId，假设 LeaseId 在 UploadLeaseParamsMode 中有对应的 getter
-            uploadDataManagementMode.setLeaseId("");
+            uploadDataManagementMode.setLeaseId(leaseParams.getLeaseId());
             AddFileResponseBodyDataMode addFileResult = addDataManagement.addFile(uploadDataManagementMode);
             if (addFileResult == null) {
                 throw new RuntimeException("将文档添加至百炼的数据管理失败");
@@ -108,8 +110,10 @@ public class KnowledgeBaseUploadServiceImpl implements KnowledgeBaseUploadServic
         } catch (IOException e) {
             e.printStackTrace();
         }
+        return null;
     }
 
+
     private File convertMultipartFileToFile(MultipartFile file) throws IOException {
         Path tempFilePath = Files.createTempFile(null, null);
         File tempFile = tempFilePath.toFile();