|
|
|
|
@ -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;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
xueqingkun
1 year ago