From ad346a7e1f1c35b09a5550c1b60cebe68f0619bf Mon Sep 17 00:00:00 2001
From: 云 <2163098428@qq.com>
Date: 星期二, 09 六月 2026 15:26:33 +0800
Subject: [PATCH] feat(ai): 集成 Pinecone 向量数据库并实现知识库 RAG 功能
---
src/main/java/com/ruoyi/approve/controller/KnowledgeBaseController.java | 152 ++++++++++++++++++++++++++++++++++++++++++--------
1 files changed, 128 insertions(+), 24 deletions(-)
diff --git a/src/main/java/com/ruoyi/approve/controller/KnowledgeBaseController.java b/src/main/java/com/ruoyi/approve/controller/KnowledgeBaseController.java
index b6ede4a..ba93fb4 100644
--- a/src/main/java/com/ruoyi/approve/controller/KnowledgeBaseController.java
+++ b/src/main/java/com/ruoyi/approve/controller/KnowledgeBaseController.java
@@ -2,66 +2,73 @@
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
-import com.ruoyi.approve.mapper.KnowledgeBaseMapper;
+import com.ruoyi.approve.dto.KnowledgeBaseVectorVO;
import com.ruoyi.approve.pojo.KnowledgeBase;
-import com.ruoyi.approve.pojo.RpaProcessAutomation;
+import com.ruoyi.approve.pojo.KnowledgeBaseVector;
import com.ruoyi.approve.service.KnowledgeBaseService;
+import com.ruoyi.approve.service.KnowledgeBaseVectorService;
+import com.ruoyi.basic.dto.StorageAttachmentDTO;
+import com.ruoyi.basic.dto.StorageBlobDTO;
+import com.ruoyi.basic.pojo.StorageBlob;
+import com.ruoyi.basic.service.StorageAttachmentService;
+import com.ruoyi.basic.service.StorageBlobService;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.framework.web.domain.AjaxResult;
-import io.swagger.annotations.Api;
-import io.swagger.annotations.ApiOperation;
+import io.swagger.v3.oas.annotations.Operation;
+import io.swagger.v3.oas.annotations.tags.Tag;
+import jakarta.servlet.http.HttpServletResponse;
import lombok.AllArgsConstructor;
-import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
-import javax.servlet.http.HttpServletResponse;
+import java.util.ArrayList;
import java.util.List;
@RestController
@RequestMapping("/knowledgeBase")
@AllArgsConstructor
-@Api(tags = "鐭ヨ瘑搴撶鐞�")
+@Tag(name = "鐭ヨ瘑搴撶鐞�")
public class KnowledgeBaseController {
- @Autowired
private KnowledgeBaseService knowledgeBaseService;
+ private KnowledgeBaseVectorService knowledgeBaseVectorService;
+ private StorageAttachmentService storageAttachmentService;
+ private StorageBlobService storageBlobService;
- /**銆�
+ /**
* 鑾峰彇鍒楄〃
- * @return
*/
@GetMapping("/getList")
public AjaxResult getList(@RequestParam(defaultValue = "1") long current,
@RequestParam(defaultValue = "10") long size, KnowledgeBase knowledgeBase) {
Page page = new Page(current, size);
- return AjaxResult.success(knowledgeBaseService.listpage(page,knowledgeBase));
+ return AjaxResult.success(knowledgeBaseService.listpage(page, knowledgeBase));
}
- /**銆�
- * 澧炴坊
- * @return
+
+ /**
+ * 鏂板鐭ヨ瘑搴�
*/
@PostMapping("/add")
- public AjaxResult add(@RequestBody KnowledgeBase knowledgeBase){
+ public AjaxResult add(@RequestBody KnowledgeBase knowledgeBase) {
return AjaxResult.success(knowledgeBaseService.save(knowledgeBase));
}
+
/**
- * 鏇存柊
- * @return
+ * 鏇存柊鐭ヨ瘑搴�
*/
@PostMapping("/update")
- public AjaxResult update(@RequestBody KnowledgeBase knowledgeBase){
+ public AjaxResult update(@RequestBody KnowledgeBase knowledgeBase) {
return AjaxResult.success(knowledgeBaseService.updateById(knowledgeBase));
}
+
/**
- * 鍒犻櫎
- * @return
+ * 鍒犻櫎鐭ヨ瘑搴�
*/
@DeleteMapping("/delete")
- public AjaxResult delete(@RequestBody List<Long> ids){
- if(CollectionUtils.isEmpty(ids)) return AjaxResult.error("璇蜂紶鍏ヨ鍒犻櫎鐨処D");
+ public AjaxResult delete(@RequestBody List<Long> ids) {
+ if (CollectionUtils.isEmpty(ids)) return AjaxResult.error("璇蜂紶鍏ヨ鍒犻櫎鐨処D");
return AjaxResult.success(knowledgeBaseService.removeByIds(ids));
}
- @ApiOperation(value = "鐭ヨ瘑搴撶鐞嗗鍑�")
+ @Operation(summary = "鐭ヨ瘑搴撶鐞嗗鍑�")
@PostMapping("/export")
public void export(HttpServletResponse response) {
List<KnowledgeBase> accountExpenses = knowledgeBaseService.list();
@@ -69,4 +76,101 @@
util.exportExcel(response, accountExpenses, "鐭ヨ瘑搴撶鐞嗗鍑�");
}
-}
+ /**
+ * 鏌ヨ鐭ヨ瘑搴撴枃浠跺悜閲忓寲鐘舵��
+ */
+ @GetMapping("/vector/status/{knowledgeBaseId}")
+ @Operation(summary = "鏌ヨ鐭ヨ瘑搴撴枃浠跺悜閲忓寲鐘舵��")
+ public AjaxResult getVectorStatus(@PathVariable Long knowledgeBaseId) {
+ List<KnowledgeBaseVectorVO> list = knowledgeBaseVectorService.getVectorStatusByKnowledgeBaseId(knowledgeBaseId);
+ return AjaxResult.success(list);
+ }
+
+ /**
+ * 閲嶆柊鍚戦噺鍖栨枃浠�
+ */
+ @PostMapping("/vector/reprocess/{vectorId}")
+ @Operation(summary = "閲嶆柊鍚戦噺鍖栨枃浠�")
+ public AjaxResult reprocessVector(@PathVariable Long vectorId) {
+ knowledgeBaseVectorService.reprocessVector(vectorId);
+ return AjaxResult.success("宸查噸鏂版彁浜ゅ悜閲忓寲浠诲姟");
+ }
+
+ /**
+ * 淇濆瓨鐭ヨ瘑搴撴枃浠跺叧鑱旓紙鏂囦欢涓婁紶鍚庤皟鐢級
+ * 涓婁紶娴佺▼锛�
+ * 1. 鍏堣皟鐢� /common/upload 涓婁紶鏂囦欢锛岃幏鍙� storageBlobDTOs
+ * 2. 鍐嶈皟鐢ㄦ鎺ュ彛鍏宠仈鏂囦欢鍒扮煡璇嗗簱骞惰Е鍙戝悜閲忓寲
+ */
+ @PostMapping("/file/save")
+ @Operation(summary = "淇濆瓨鐭ヨ瘑搴撴枃浠跺叧鑱�")
+ public AjaxResult saveKnowledgeBaseFiles(@RequestBody KnowledgeBaseFileDTO dto) {
+ if (dto.getKnowledgeBaseId() == null) {
+ return AjaxResult.error("鐭ヨ瘑搴揑D涓嶈兘涓虹┖");
+ }
+ if (CollectionUtils.isEmpty(dto.getStorageBlobIds())) {
+ return AjaxResult.error("鏂囦欢ID涓嶈兘涓虹┖");
+ }
+
+ // 淇濆瓨闄勪欢鍏宠仈
+ StorageAttachmentDTO attachmentDTO = new StorageAttachmentDTO();
+ attachmentDTO.setRecordType("knowledge_base");
+ attachmentDTO.setRecordId(dto.getKnowledgeBaseId());
+ attachmentDTO.setApplication("rag_file");
+ List<StorageBlobDTO> blobDTOs = new ArrayList<>();
+ for (Long blobId : dto.getStorageBlobIds()) {
+ StorageBlobDTO blobDTO = new StorageBlobDTO();
+ blobDTO.setId(blobId);
+ blobDTOs.add(blobDTO);
+ }
+ attachmentDTO.setStorageBlobDTOs(blobDTOs);
+ storageAttachmentService.saveStorageAttachment(attachmentDTO);
+
+ // 鍒涘缓鍚戦噺璁板綍骞惰Е鍙戝悜閲忓寲
+ for (Long blobId : dto.getStorageBlobIds()) {
+ StorageBlob blob = storageBlobService.getById(blobId);
+ if (blob != null) {
+ String fileName = blob.getOriginalFilename();
+ String fileType = getFileExtension(fileName);
+
+ knowledgeBaseVectorService.createVectorRecord(
+ dto.getKnowledgeBaseId(),
+ blobId,
+ fileName,
+ fileType
+ );
+ }
+ }
+
+ return AjaxResult.success();
+ }
+
+ private String getFileExtension(String fileName) {
+ if (fileName == null || !fileName.contains(".")) {
+ return "unknown";
+ }
+ return fileName.substring(fileName.lastIndexOf('.') + 1).toLowerCase();
+ }
+
+ /**
+ * 鍒犻櫎鐭ヨ瘑搴撴枃浠�
+ */
+ @DeleteMapping("/file/delete")
+ @Operation(summary = "鍒犻櫎鐭ヨ瘑搴撴枃浠�")
+ public AjaxResult deleteKnowledgeBaseFiles(@RequestBody List<Long> vectorIds) {
+ if (CollectionUtils.isEmpty(vectorIds)) {
+ return AjaxResult.error("璇烽�夋嫨瑕佸垹闄ょ殑鏂囦欢");
+ }
+ knowledgeBaseVectorService.deleteVectors(vectorIds);
+ return AjaxResult.success();
+ }
+
+ /**
+ * 鐭ヨ瘑搴撴枃浠禗TO
+ */
+ @lombok.Data
+ public static class KnowledgeBaseFileDTO {
+ private Long knowledgeBaseId;
+ private List<Long> storageBlobIds;
+ }
+}
\ No newline at end of file
--
Gitblit v1.9.3