Yaxin
/
virtual-patient
31
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

优化测试服务

Browse Source
dev_2.1.0
liu 1 year ago
parent d726a73ad0
commit ebfa898b2a
3 changed files with 84 additions and 14 deletions
Show Stats Download Patch File Download Diff File

2
virtual-patient-graph/src/main/java/com/supervision/service/impl/GraphNebulaServiceImpl.java
Unescape Escape View File

@ -336,7 +336,7 @@ public class GraphNebulaServiceImpl implements GraphNebulaService {
;
}
}
preNode.setChildNodeList(childNode);
preNode.setChildren(childNode);
for (TreeNodeVO treeNodeVO : childNode) {
recursionBuildTree(treeNodeVO, treeNodeMap, edgeList);
}

2
virtual-patient-graph/src/main/java/com/supervision/vo/TreeNodeVO.java
Unescape Escape View File

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

94
virtual-patient-web/src/test/java/com/supervision/AskTemplateIdTest.java
Unescape Escape View File

@ -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<CommonDic> aqtList = commonDicService.lambdaQuery().eq(CommonDic::getGroupCode, "AQT").isNotNull(CommonDic::getParentId).ne(CommonDic::getParentId, 179).list();
Map<Long, CommonDic> dictMap = aqtList.stream().collect(Collectors.toMap(CommonDic::getId, Function.identity()));
List<AskPatientAnswer> list = askPatientAnswerService.lambdaQuery().isNotNull(AskPatientAnswer::getQuestion).eq(AskPatientAnswer::getAnswerType, 1).list();
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;
}

Write Preview
Loading…
Cancel
Save