代码提交

virtual-patient-web/pom.xml

@@ -24,6 +24,8 @@
             <version>${project.version}</version>
         </dependency>
 
+
+
         <dependency>
             <groupId>com.supervision</groupId>
             <artifactId>virtual-patient-common</artifactId>
@@ -57,21 +59,7 @@
             <groupId>org.springframework.boot</groupId>
             <artifactId>spring-boot-starter-websocket</artifactId>
         </dependency>
-        <dependency>
-            <groupId>org.springframework.boot</groupId>
-            <artifactId>spring-boot-test</artifactId>
-            <scope>test</scope>
-        </dependency>
-        <dependency>
-            <groupId>junit</groupId>
-            <artifactId>junit</artifactId>
-            <scope>test</scope>
-        </dependency>
-        <dependency>
-            <groupId>org.springframework</groupId>
-            <artifactId>spring-test</artifactId>
-            <scope>test</scope>
-        </dependency>
+
         <dependency>
             <groupId>org.springframework.boot</groupId>
             <artifactId>spring-boot-test</artifactId>

virtual-patient-web/src/main/resources/application.yml

@@ -1,3 +1,3 @@
 spring:
   profiles:
-    active: dev
+    active: local

virtual-patient-web/src/test/java/com/supervision/AskTemplateIdTest.java

@@ -0,0 +1,122 @@
+package com.supervision;
+
+import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.util.ArrayUtil;
+import cn.hutool.core.util.ObjectUtil;
+import cn.hutool.core.util.ReUtil;
+import cn.hutool.core.util.StrUtil;
+import cn.hutool.http.HttpUtil;
+import cn.hutool.json.JSON;
+import cn.hutool.json.JSONArray;
+import cn.hutool.json.JSONObject;
+import cn.hutool.json.JSONUtil;
+import cn.hutool.poi.excel.ExcelReader;
+import cn.hutool.poi.excel.ExcelUtil;
+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.service.AskPatientAnswerService;
+import com.supervision.service.AskTemplateQuestionLibraryService;
+import com.supervision.service.CommonDicService;
+import lombok.extern.slf4j.Slf4j;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+@Slf4j
+@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
+@RunWith(SpringJUnit4ClassRunner.class)
+public class AskTemplateIdTest {
+
+    @Autowired
+    private AskTemplateQuestionLibraryService askTemplateQuestionLibraryService;
+
+    @Autowired
+    private AskPatientAnswerService askPatientAnswerService;
+
+    @Autowired
+    private CommonDicService commonDicService;
+
+    @Test
+    public void creatAskId() {
+        Object o = new Object();
+        DefaultIdentifierGenerator defaultIdentifierGenerator = new DefaultIdentifierGenerator();
+
+        List<AskTemplateQuestionLibrary> list = askTemplateQuestionLibraryService.list();
+        for (AskTemplateQuestionLibrary askTemplateQuestionLibrary : list) {
+            askTemplateQuestionLibraryService.lambdaUpdate()
+                    .set(AskTemplateQuestionLibrary::getId, String.valueOf(defaultIdentifierGenerator.nextId(o)))
+                    .eq(AskTemplateQuestionLibrary::getId, askTemplateQuestionLibrary.getId()).update();
+        }
+    }
+
+    @Test
+    public void creatAnswerId() {
+        Object o = new Object();
+        DefaultIdentifierGenerator defaultIdentifierGenerator = new DefaultIdentifierGenerator();
+
+        List<AskPatientAnswer> list = askPatientAnswerService.list();
+        for (AskPatientAnswer answer : list) {
+            askPatientAnswerService.lambdaUpdate()
+                    .set(AskPatientAnswer::getId, String.valueOf(defaultIdentifierGenerator.nextId(o)))
+                    .eq(AskPatientAnswer::getId, answer.getId()).update();
+        }
+    }
+
+    @Test
+    public void insertDict() {
+        ExcelReader reader = ExcelUtil.getReader("/Users/flevance/Desktop/虚拟病人/语料库/标准病人语料1226-2.xlsx");
+        List<List<Object>> read = reader.read(1, 86);
+        for (List<Object> objects : read) {
+            String pathOne = String.valueOf(objects.get(0));
+            String pathTwo = String.valueOf(objects.get(1));
+            CommonDic dic = commonDicService.lambdaQuery().eq(CommonDic::getNameZhPath, pathOne + "/" + pathTwo).last("limit 1").one();
+            if (ObjectUtil.isNotEmpty(dic)) {
+                String code = String.valueOf(objects.get(5));
+                askTemplateQuestionLibraryService.lambdaUpdate().set(AskTemplateQuestionLibrary::getDictId, dic.getId())
+                        .eq(AskTemplateQuestionLibrary::getCode, code).update();
+            }
+        }
+    }
+
+    @Test
+    public void generateQuestion() {
+        // 使用文心一言,这里是付费的,不要随便用哦,注意数量
+        String post = HttpUtil.get("https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=8VdV7Hm4ZmPVzRovCQXaH8nN&client_secret=Bte6UKtrexydMuhWrbqciolzYzS8rEYm");
+        System.out.println(post);
+        JSONObject parse = JSONUtil.parseObj(post);
+        String accessToken = parse.getStr("access_token");
+
+        List<AskTemplateQuestionLibrary> list = askTemplateQuestionLibraryService.list();
+        for (AskTemplateQuestionLibrary ask : list) {
+            try {
+                String description = ask.getDescription();
+                Map<String, Object> map = new HashMap<>();
+                map.put("role", "user");
+                map.put("content", "请把下面这句话以20种不同的方式提问，以JSONARRAY的形式输出:" + description);
+                HashMap<String, Object> param = new HashMap<>();
+                param.put("messages", CollUtil.newArrayList(map));
+                String askAnswer = HttpUtil.post("https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions_pro?access_token=" + accessToken, JSONUtil.toJsonStr(param));
+                JSONObject answerJSON = JSONUtil.parseObj(askAnswer);
+                String result = answerJSON.getStr("result");
+                String s = ReUtil.get("\\[(.*?)\\]", result, 0);
+                List<String> question = JSONUtil.toList(s, String.class);
+                question.add(0, description);
+                askTemplateQuestionLibraryService.lambdaUpdate().set(AskTemplateQuestionLibrary::getQuestion, JSONUtil.toJsonStr(question)).eq(AskTemplateQuestionLibrary::getId, ask.getId()).update();
+            } catch (Exception e) {
+                log.error("{}生成错误", ask.getDescription(), e);
+            }
+
+        }
+
+    }
+
+
+}