优化测试服务

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

virtual-patient-graph/src/main/java/com/supervision/vo/TreeNodeVO.java

@@ -31,5 +31,5 @@ public class TreeNodeVO {
      * 子节点
      */
     @ApiModelProperty("子节点")
-    private List<TreeNodeVO> childNodeList;
+    private List<TreeNodeVO> children;
 }

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 {
         }
     }
 
-
+    @Test
-    @Autowired
+    public void generateByGpt() {
-    private AskService askService;
+        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 {
+    public void testRasa() {
-        TalkVideoReqVO talkVideoReqVO = new TalkVideoReqVO();
+        List<CommonDic> aqtList = commonDicService.lambdaQuery().eq(CommonDic::getGroupCode, "AQT").isNotNull(CommonDic::getParentId).ne(CommonDic::getParentId, 179).list();
-        talkVideoReqVO.setText("你现在感觉怎么样？");
+        Map<Long, CommonDic> dictMap = aqtList.stream().collect(Collectors.toMap(CommonDic::getId, Function.identity()));
-        talkVideoReqVO.setProcessId("1749312510591934465");
+        List<AskPatientAnswer> list = askPatientAnswerService.lambdaQuery().isNotNull(AskPatientAnswer::getQuestion).eq(AskPatientAnswer::getAnswerType, 1).list();
-        TalkResultResVO talkResultResVO = askService.talkByVideo(talkVideoReqVO);
+        List<AskTemplateQuestionLibrary> libraryList = askTemplateQuestionLibraryService.list();
+        Map<String, AskTemplateQuestionLibrary> libraryMap = libraryList.stream().collect(Collectors.toMap(AskTemplateQuestionLibrary::getId, Function.identity()));
+        for (AskPatientAnswer answer : list) {
+            Map<Object, Object> 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<GptMessage> 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<ResaIntentResult> intent_ranking;
+    }
 
-        System.out.println(JSONUtil.toJsonStr(talkResultResVO));
+    @Data
+    private static class ResaIntentResult {
+        private String name;
+        private Double confidence;
     }