diff --git a/virtual-patient-graph/src/main/java/com/supervision/service/impl/GraphNebulaServiceImpl.java b/virtual-patient-graph/src/main/java/com/supervision/service/impl/GraphNebulaServiceImpl.java index b9fe54c7..da8d7a03 100644 --- a/virtual-patient-graph/src/main/java/com/supervision/service/impl/GraphNebulaServiceImpl.java +++ b/virtual-patient-graph/src/main/java/com/supervision/service/impl/GraphNebulaServiceImpl.java @@ -336,7 +336,7 @@ public class GraphNebulaServiceImpl implements GraphNebulaService { ; } } - preNode.setChildNodeList(childNode); + preNode.setChildren(childNode); for (TreeNodeVO treeNodeVO : childNode) { recursionBuildTree(treeNodeVO, treeNodeMap, edgeList); } diff --git a/virtual-patient-graph/src/main/java/com/supervision/vo/TreeNodeVO.java b/virtual-patient-graph/src/main/java/com/supervision/vo/TreeNodeVO.java index 39384497..6a0c45f0 100644 --- a/virtual-patient-graph/src/main/java/com/supervision/vo/TreeNodeVO.java +++ b/virtual-patient-graph/src/main/java/com/supervision/vo/TreeNodeVO.java @@ -31,5 +31,5 @@ public class TreeNodeVO { * 子节点 */ @ApiModelProperty("子节点") - private List childNodeList; + private List children; } diff --git a/virtual-patient-web/src/test/java/com/supervision/AskTemplateIdTest.java b/virtual-patient-web/src/test/java/com/supervision/AskTemplateIdTest.java index 4abd7794..0433e612 100644 --- a/virtual-patient-web/src/test/java/com/supervision/AskTemplateIdTest.java +++ b/virtual-patient-web/src/test/java/com/supervision/AskTemplateIdTest.java @@ -1,8 +1,11 @@ package com.supervision; import cn.hutool.core.collection.CollUtil; +import cn.hutool.core.map.MapUtil; import cn.hutool.core.util.ObjectUtil; import cn.hutool.core.util.ReUtil; +import cn.hutool.http.HttpRequest; +import cn.hutool.http.HttpResponse; import cn.hutool.http.HttpUtil; import cn.hutool.json.JSONObject; import cn.hutool.json.JSONUtil; @@ -12,12 +15,11 @@ import com.baomidou.mybatisplus.core.incrementer.DefaultIdentifierGenerator; import com.supervision.model.AskPatientAnswer; import com.supervision.model.AskTemplateQuestionLibrary; import com.supervision.model.CommonDic; -import com.supervision.pojo.vo.TalkResultResVO; -import com.supervision.pojo.vo.TalkVideoReqVO; import com.supervision.service.AskPatientAnswerService; import com.supervision.service.AskService; import com.supervision.service.AskTemplateQuestionLibraryService; import com.supervision.service.CommonDicService; +import lombok.Data; import lombok.extern.slf4j.Slf4j; import org.junit.Test; import org.junit.runner.RunWith; @@ -25,7 +27,6 @@ import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.test.context.SpringBootTest; import org.springframework.test.context.junit4.SpringJUnit4ClassRunner; -import java.io.IOException; import java.util.HashMap; import java.util.List; import java.util.Map; @@ -46,6 +47,9 @@ public class AskTemplateIdTest { @Autowired private CommonDicService commonDicService; + @Autowired + private AskService askService; + @Test public void creatAskId() { Object o = new Object(); @@ -125,18 +129,84 @@ public class AskTemplateIdTest { } } - - @Autowired - private AskService askService; + @Test + public void generateByGpt() { + String api_key = "sk-FDNQ1bhd7007e62e714eT3BLbKFJ004fcC3ebDeA4542a516"; + String url = "https://aigptx.top/v1/chat/completions"; + + String question = "假设你是一个精通RASA NLU调优的工程师且具备丰富医疗经验;" + + "我现在有一个意图,请你根据这个意图,针对这个问题示例,提出10条医生在问诊时,可能根据这个意图来提问患者的问题.\n" + + "注意,问题不要超出这个意图的范围,始终契合意图的关键词\n" + + "回答请使用json array的格式,示例:[\"相似问题1\",\"相似问题2\"]\n" + + "### 下面是问题示例\n" + + "这种感觉持续多久了?"; + + GptParam gptParam = new GptParam(); + GptMessage gptMessage = new GptMessage(); + gptMessage.setContent(question); + gptParam.setMessages(CollUtil.newArrayList(gptMessage)); + + HttpResponse response = HttpRequest.post(url) + .header("Authorization", "Bearer " + api_key) + .body(JSONUtil.toJsonStr(gptParam)) + .execute(); + String body = response.body(); + System.out.println(body); + } @Test - public void testRasa() throws IOException { - TalkVideoReqVO talkVideoReqVO = new TalkVideoReqVO(); - talkVideoReqVO.setText("你现在感觉怎么样?"); - talkVideoReqVO.setProcessId("1749312510591934465"); - TalkResultResVO talkResultResVO = askService.talkByVideo(talkVideoReqVO); + public void testRasa() { + List aqtList = commonDicService.lambdaQuery().eq(CommonDic::getGroupCode, "AQT").isNotNull(CommonDic::getParentId).ne(CommonDic::getParentId, 179).list(); + Map dictMap = aqtList.stream().collect(Collectors.toMap(CommonDic::getId, Function.identity())); + List list = askPatientAnswerService.lambdaQuery().isNotNull(AskPatientAnswer::getQuestion).eq(AskPatientAnswer::getAnswerType, 1).list(); + List libraryList = askTemplateQuestionLibraryService.list(); + Map libraryMap = libraryList.stream().collect(Collectors.toMap(AskTemplateQuestionLibrary::getId, Function.identity())); + for (AskPatientAnswer answer : list) { + Map build = MapUtil.builder().put("text", answer.getQuestion()).build(); + String post = HttpUtil.post("http://localhost:5005/model/parse", JSONUtil.toJsonStr(build)); + RasaResult bean = JSONUtil.toBean(post, RasaResult.class); + ResaIntentResult intent = bean.getIntent(); + if (intent.getName().startsWith("Q")) { + String id = intent.getName().split("_")[1]; + if (!id.equals(answer.getLibraryQuestionId())) { + log.info("问题:{}匹配不正确,走了其他回答,期望ID为:{},实际ID为:{},实际分类为:{},期望分类为:{}", bean.getText(), answer.getLibraryQuestionId(), id, + dictMap.get(libraryMap.get(id).getDictId()).getNameZhPath(), + dictMap.get(libraryMap.get(answer.getLibraryQuestionId()).getDictId()).getNameZhPath() + + ); + } + } else { + log.info("问题:{}匹配不正确,走了默认回答", bean.getText()); + } + + } + + } + + @Data + private static class GptParam { + private List messages; + // # 如果需要切换模型,在这里修改 + private String model = "gpt-3.5-turbo"; + } + + @Data + private static class GptMessage { + private String role = "user"; + private String content; + } + + @Data + private static class RasaResult { + private String text; + private ResaIntentResult intent; + private List intent_ranking; + } - System.out.println(JSONUtil.toJsonStr(talkResultResVO)); + @Data + private static class ResaIntentResult { + private String name; + private Double confidence; }