From a0e73c4ce338aba755bc6ec5ee3fd9dab63a6000 Mon Sep 17 00:00:00 2001
From: "DESKTOP-DDTUS3E\\yaxin" <daixiaoyi0622@gmail.com>
Date: Thu, 24 Oct 2024 14:16:47 +0800
Subject: [PATCH] =?UTF-8?q?=E8=AF=A6=E6=83=85=E8=BF=94=E5=9B=9E=E5=AE=8C?=
 =?UTF-8?q?=E6=95=B4=E4=B8=A4=E5=B1=82=E8=AF=81=E6=8D=AE=E5=88=86=E7=B1=BB?=
 =?UTF-8?q?Id?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../police/controller/PromptController.java   |  4 +--
 .../supervision/police/dto/NotePromptDTO.java |  4 +++
 .../police/service/NotePromptService.java     |  2 ++
 .../service/impl/NotePromptServiceImpl.java   | 27 ++++++++++++++-----
 4 files changed, 29 insertions(+), 8 deletions(-)

diff --git a/src/main/java/com/supervision/police/controller/PromptController.java b/src/main/java/com/supervision/police/controller/PromptController.java
index 8bd898d..fe4307f 100644
--- a/src/main/java/com/supervision/police/controller/PromptController.java
+++ b/src/main/java/com/supervision/police/controller/PromptController.java
@@ -45,7 +45,7 @@ public class PromptController {
 
     @PostMapping("/getById")
     @Operation(summary = "提示词调试")
-    public R<NotePrompt> getById(@RequestParam String id) {
-        return R.ok(promptService.getById(id));
+    public R<NotePromptDTO> getById(@RequestBody NotePrompt notePrompt) {
+        return R.ok(promptService.getById(notePrompt.getId()));
     }
 }
diff --git a/src/main/java/com/supervision/police/dto/NotePromptDTO.java b/src/main/java/com/supervision/police/dto/NotePromptDTO.java
index f7bebf1..c49113a 100644
--- a/src/main/java/com/supervision/police/dto/NotePromptDTO.java
+++ b/src/main/java/com/supervision/police/dto/NotePromptDTO.java
@@ -4,10 +4,14 @@ import com.supervision.police.domain.NotePrompt;
 import lombok.Data;
 import lombok.EqualsAndHashCode;
 
+import java.util.ArrayList;
+import java.util.List;
+
 @EqualsAndHashCode(callSuper = true)
 @Data
 public class NotePromptDTO extends NotePrompt {
     private Integer matchNum;
     private String fileId;
     private String text;
+    private List<String> evidenceCategoryIdList = new ArrayList<>();
 }
diff --git a/src/main/java/com/supervision/police/service/NotePromptService.java b/src/main/java/com/supervision/police/service/NotePromptService.java
index 800a09f..af45fb6 100644
--- a/src/main/java/com/supervision/police/service/NotePromptService.java
+++ b/src/main/java/com/supervision/police/service/NotePromptService.java
@@ -16,4 +16,6 @@ public interface NotePromptService extends IService<NotePrompt> {
     IPage<NotePromptDTO> listPrompt(int page, int size, NotePrompt notePrompt);
 
     List promptDebugging(NotePromptDTO notePromptDTO);
+
+    NotePromptDTO getById(String id);
 }
diff --git a/src/main/java/com/supervision/police/service/impl/NotePromptServiceImpl.java b/src/main/java/com/supervision/police/service/impl/NotePromptServiceImpl.java
index 549d6dc..1e46dda 100644
--- a/src/main/java/com/supervision/police/service/impl/NotePromptServiceImpl.java
+++ b/src/main/java/com/supervision/police/service/impl/NotePromptServiceImpl.java
@@ -6,16 +6,12 @@ import cn.hutool.json.JSONUtil;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
-import com.fasterxml.jackson.databind.ObjectMapper;
 import com.supervision.common.constant.NotePromptConstants;
 import com.supervision.common.utils.StringUtils;
 import com.supervision.demo.dto.QARecordNodeDTO;
 import com.supervision.minio.domain.MinioFile;
 import com.supervision.minio.service.MinioService;
-import com.supervision.police.domain.ModelRecordType;
-import com.supervision.police.domain.NotePrompt;
-import com.supervision.police.domain.NotePromptTypeRel;
-import com.supervision.police.domain.NoteRecordSplit;
+import com.supervision.police.domain.*;
 import com.supervision.police.dto.LLMExtractDto;
 import com.supervision.police.dto.NotePromptDTO;
 import com.supervision.police.mapper.NotePromptMapper;
@@ -25,11 +21,11 @@ import com.supervision.utils.RecordRegexUtil;
 import com.supervision.utils.WordReadUtil;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
-import org.apache.poi.ss.formula.functions.T;
 import org.springframework.ai.chat.ChatClient;
 import org.springframework.ai.chat.ChatResponse;
 import org.springframework.ai.chat.messages.UserMessage;
 import org.springframework.ai.chat.prompt.Prompt;
+import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
@@ -61,6 +57,8 @@ public class NotePromptServiceImpl extends ServiceImpl<NotePromptMapper, NotePro
     private LLMExtractService llmExtractService;
     @Autowired
     private ChatClient chatClient;
+    @Autowired
+    private EvidenceCategoryService evidenceCategoryService;
 
     @Override
     public List<NotePrompt> listPromptBySplitId(String recordSplitId) {
@@ -181,4 +179,21 @@ public class NotePromptServiceImpl extends ServiceImpl<NotePromptMapper, NotePro
         return null;
 
     }
+
+    @Override
+    public NotePromptDTO getById(String id) {
+        NotePrompt notePrompt = super.getById(id);
+        NotePromptDTO notePromptDTO = new NotePromptDTO();
+        BeanUtils.copyProperties(notePrompt, notePromptDTO);
+        String evidenceCategoryId = notePromptDTO.getEvidenceCategoryId();
+        if (StringUtils.isNotEmpty(evidenceCategoryId)) {
+            EvidenceCategory category = evidenceCategoryService.getById(evidenceCategoryId);
+            if (category != null) {
+                String parentId = category.getParentId();
+                notePromptDTO.setEvidenceCategoryIdList(Arrays.asList(parentId, evidenceCategoryId));
+            }
+        }
+        return notePromptDTO;
+    }
+
 }