WJB
/
ai-manus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

微调文件上传接口

This commit is contained in:
wenjinbo 2025-08-04 16:39:16 +08:00
parent 05086120b9
commit af609e3d70
4 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
chat-server/src/main/java/com/bjtds/brichat/mapper/postgresql/DifyDatasetsMapper.java
View File

@ -20,4 +20,5 @@ public interface DifyDatasetsMapper {
void updateDataset(DatasetUpdateReq datasetUpdateReq);
String getDatasetNameById(String name);
}

11
chat-server/src/main/java/com/bjtds/brichat/service/dify/impl/DifyDatasetApiServiceImpl.java
View File

@ -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);

5
chat-server/src/main/java/com/bjtds/brichat/service/impl/MarkdownServiceImpl.java
View File

@ -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;

4
chat-server/src/main/resources/com/bjtds/brichat/mapper/postgresql/DifyDatasetMapper.xml
View File

@ -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>