From af609e3d7064228d3bdc25e55a927209cd6591b8 Mon Sep 17 00:00:00 2001
From: wenjinbo <599483010@qq.com>
Date: Mon, 4 Aug 2025 16:39:16 +0800
Subject: [PATCH] =?UTF-8?q?=E5=BE=AE=E8=B0=83=E6=96=87=E4=BB=B6=E4=B8=8A?=
 =?UTF-8?q?=E4=BC=A0=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../brichat/mapper/postgresql/DifyDatasetsMapper.java |  1 +
 .../service/dify/impl/DifyDatasetApiServiceImpl.java  | 11 +++++++----
 .../brichat/service/impl/MarkdownServiceImpl.java     |  5 ++---
 .../brichat/mapper/postgresql/DifyDatasetMapper.xml   |  4 ++++
 4 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/chat-server/src/main/java/com/bjtds/brichat/mapper/postgresql/DifyDatasetsMapper.java b/chat-server/src/main/java/com/bjtds/brichat/mapper/postgresql/DifyDatasetsMapper.java
index 693c3f2..66a90f4 100644
--- a/chat-server/src/main/java/com/bjtds/brichat/mapper/postgresql/DifyDatasetsMapper.java
+++ b/chat-server/src/main/java/com/bjtds/brichat/mapper/postgresql/DifyDatasetsMapper.java
@@ -20,4 +20,5 @@ public interface DifyDatasetsMapper {
 
     void updateDataset(DatasetUpdateReq datasetUpdateReq);
 
+    String getDatasetNameById(String name);
 }
diff --git a/chat-server/src/main/java/com/bjtds/brichat/service/dify/impl/DifyDatasetApiServiceImpl.java b/chat-server/src/main/java/com/bjtds/brichat/service/dify/impl/DifyDatasetApiServiceImpl.java
index 86666b1..3905010 100644
--- a/chat-server/src/main/java/com/bjtds/brichat/service/dify/impl/DifyDatasetApiServiceImpl.java
+++ b/chat-server/src/main/java/com/bjtds/brichat/service/dify/impl/DifyDatasetApiServiceImpl.java
@@ -6,6 +6,7 @@ import com.bjtds.brichat.entity.dify.DatasetDto;
 import com.bjtds.brichat.entity.dify.DifyDatasetResponse;
 import com.bjtds.brichat.entity.dto.PdfConversionResponse;
 import com.bjtds.brichat.entity.dto.PdfTaskDto;
+import com.bjtds.brichat.mapper.postgresql.DifyDatasetsMapper;
 import com.bjtds.brichat.service.dify.DifyDatasetApiService;
 import com.bjtds.brichat.util.Constants;
 import com.fasterxml.jackson.core.JsonProcessingException;
@@ -74,6 +75,9 @@ public class DifyDatasetApiServiceImpl implements DifyDatasetApiService {
     @Value("${pdf.conversion.service.max-workers}")
     private Integer pdfConversionMaxWorkers;
 
+    @Resource
+    private DifyDatasetsMapper difyDatasetsMapper;
+
     @Override
     public ResponseEntity<DatasetDto> createDataset(String name, String description) {
         // 1. 设置请求URL
@@ -192,11 +196,10 @@ public class DifyDatasetApiServiceImpl implements DifyDatasetApiService {
 //                0.0, // 初始化进度为0%
 //                request.getDatasetId()
 //            );
-            DatasetInfoRequest datasetInfoRequest = new DatasetInfoRequest();
-            datasetInfoRequest.setDatasetId(request.getDatasetId());
 
-            DatasetInfoResponse datasetInfo = difyDatasetService.info(datasetInfoRequest);
-            String datasetName = datasetInfo.getName();
+
+
+            String datasetName = difyDatasetsMapper.getDatasetNameById(request.getDatasetId());
 
             request.setDeepAnalysis(false);
 
diff --git a/chat-server/src/main/java/com/bjtds/brichat/service/impl/MarkdownServiceImpl.java b/chat-server/src/main/java/com/bjtds/brichat/service/impl/MarkdownServiceImpl.java
index 9eedbf0..49bfbd1 100644
--- a/chat-server/src/main/java/com/bjtds/brichat/service/impl/MarkdownServiceImpl.java
+++ b/chat-server/src/main/java/com/bjtds/brichat/service/impl/MarkdownServiceImpl.java
@@ -2,8 +2,7 @@ package com.bjtds.brichat.service.impl;
 
 import com.bjtds.brichat.service.MarkdownService;
 import com.bjtds.brichat.util.FuzzyMatcher;
-import com.itextpdf.text.pdf.parser.clipper.Path;
-import com.sun.xml.internal.ws.policy.privateutil.PolicyUtils;
+
 import java.io.File;
 import org.springframework.stereotype.Service;
 
@@ -11,7 +10,7 @@ import java.io.IOException;
 import java.nio.charset.StandardCharsets;
 import java.nio.file.Files;
 import java.nio.file.Paths;
-import java.nio.file.StandardCopyOption;
+
 import java.nio.file.StandardOpenOption;
 import java.util.ArrayList;
 import java.util.List;
diff --git a/chat-server/src/main/resources/com/bjtds/brichat/mapper/postgresql/DifyDatasetMapper.xml b/chat-server/src/main/resources/com/bjtds/brichat/mapper/postgresql/DifyDatasetMapper.xml
index 5975c64..97a5040 100644
--- a/chat-server/src/main/resources/com/bjtds/brichat/mapper/postgresql/DifyDatasetMapper.xml
+++ b/chat-server/src/main/resources/com/bjtds/brichat/mapper/postgresql/DifyDatasetMapper.xml
@@ -43,4 +43,8 @@
         select  id,tenant_id,name,description,created_by,created_at, updated_at from datasets   ORDER BY created_at DESC
     </select>
 
+    <select id="getDatasetNameById" resultType="java.lang.String" parameterType="java.lang.String">
+        select name from datasets where id = CAST(#{id} AS UUID)
+    </select>
+
 </mapper>
\ No newline at end of file