详情返回完整两层证据分类Id

topo_dev
DESKTOP-DDTUS3E\yaxin 6 months ago
parent f2a342cd7c
commit a0e73c4ce3

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

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

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

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

Loading…
Cancel
Save