From c5f11d6f476e14b745c59b0abdf7123d0b640a1e Mon Sep 17 00:00:00 2001 From: liu Date: Mon, 6 May 2024 15:31:54 +0800 Subject: [PATCH] =?UTF-8?q?AiUtil=E6=9B=B4=E6=8D=A2=E4=B8=BA=E6=96=B0?= =?UTF-8?q?=E7=9A=84apiKey?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../main/java/com/supervision/ai/AiUtil.java | 2 +- .../test/java/com/supervision/AskTest.java | 95 ------------------- 2 files changed, 1 insertion(+), 96 deletions(-) delete mode 100644 kbqa-graph/src/test/java/com/supervision/AskTest.java diff --git a/kbqa-graph/src/main/java/com/supervision/ai/AiUtil.java b/kbqa-graph/src/main/java/com/supervision/ai/AiUtil.java index ed70bfc..f9390e6 100644 --- a/kbqa-graph/src/main/java/com/supervision/ai/AiUtil.java +++ b/kbqa-graph/src/main/java/com/supervision/ai/AiUtil.java @@ -19,7 +19,7 @@ import java.util.Map; @Slf4j public class AiUtil { - private static final String API_KEY = "6e91c91fc9da1f19f0cbaad0d31d90ee.Rsb3KSMJskIpQjhc"; + private static final String API_KEY = "789c02fa1000afd951145efff01e0461.tGLSHnIsiEEiWNNN"; private static final String TOKEN; diff --git a/kbqa-graph/src/test/java/com/supervision/AskTest.java b/kbqa-graph/src/test/java/com/supervision/AskTest.java deleted file mode 100644 index 741070d..0000000 --- a/kbqa-graph/src/test/java/com/supervision/AskTest.java +++ /dev/null @@ -1,95 +0,0 @@ -package com.supervision; - -import cn.hutool.core.collection.CollUtil; -import cn.hutool.core.io.FileUtil; -import cn.hutool.core.io.file.FileReader; -import cn.hutool.core.util.ObjectUtil; -import cn.hutool.core.util.StrUtil; -import cn.hutool.poi.excel.ExcelReader; -import cn.hutool.poi.excel.ExcelUtil; -import cn.hutool.poi.excel.ExcelWriter; -import com.supervision.service.AskService; -import com.supervision.vo.SingleTalkReqVO; -import com.supervision.vo.SingleTalkResVO; -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.io.BufferedWriter; -import java.io.File; -import java.io.FileWriter; -import java.io.IOException; -import java.nio.charset.Charset; -import java.util.List; - -@Slf4j -@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT) -@RunWith(SpringJUnit4ClassRunner.class) -public class AskTest { - - @Autowired - private AskService askService; - - @Test - public void test() throws IOException { - // 从excel中获取问题,开始循环调用 - ExcelReader reader = ExcelUtil.getReader("/Users/flevance/Desktop/知识图谱/正常退休业务问题梳理_0422.xlsx", 1); - File file = FileUtil.file("/Users/flevance/Java/KBQA/test.log"); - ExcelWriter writer = ExcelUtil.getWriter("/Users/flevance/Desktop/知识图谱/正常退休业务问题梳理_0422_output.xlsx", "Sheet2"); - - // 从第一行开始读取 - List> readList = reader.read(1); - for (int i = 0; i < readList.size(); i++) { - List objects = readList.get(i); - SingleTalkReqVO singleTalkReqVO = new SingleTalkReqVO(); - singleTalkReqVO.setUserTalk(String.valueOf(objects.get(1))); - SingleTalkResVO singleTalkResVO = null; - try { - singleTalkResVO = askService.singleTalk(singleTalkReqVO); - } catch (Exception e) { - log.info("出现异常,结束"); - } - FileReader fileReader = new FileReader(file); - List strings = fileReader.readLines(); - int y = i + 1; - strings.stream().filter(e -> StrUtil.contains(e, "identifyIntent开始识别意图:")).findAny().ifPresent(e -> { - String s = StrUtil.subAfter(e, "identifyIntent开始识别意图:", true); - writer.writeCellValue(2, y, s); - }); - strings.stream().filter(e -> StrUtil.contains(e, "identifyIntent意图识别结果为:")).findAny().ifPresent(e -> { - String s = StrUtil.subAfter(e, "identifyIntent意图识别结果为:", true); - writer.writeCellValue(3, y, s); - }); - strings.stream().filter(e -> StrUtil.contains(e, "itemExtractByPossibleItemWithExample查询语句为:")).findAny().ifPresent(e -> { - String s = StrUtil.subAfter(e, "itemExtractByPossibleItemWithExample查询语句为:", true); - writer.writeCellValue(4, y, s); - }); - strings.stream().filter(e -> StrUtil.contains(e, "itemExtractByPossibleItemWithExample结果为:")).findAny().ifPresent(e -> { - String s = StrUtil.subAfter(e, "itemExtractByPossibleItemWithExample结果为:", true); - writer.writeCellValue(5, y, s); - }); - strings.stream().filter(e -> StrUtil.contains(e, "answerQuestion的prompt是:")).findAny().ifPresent(e -> { - String s = StrUtil.subAfter(e, "answerQuestion的prompt是:", true); - writer.writeCellValue(6, y, s); - }); - SingleTalkResVO finalSingleTalkResVO = singleTalkResVO; - strings.stream().filter(e -> StrUtil.contains(e, "answerQuestion的答案是:")).findAny().ifPresent(e -> { - String s = StrUtil.subAfter(e, "answerQuestion的答案是:", true); - if (ObjectUtil.isNotNull(finalSingleTalkResVO)) { - writer.writeCellValue(7, y, finalSingleTalkResVO.getAnswerText()); - } - }); - writer.flush(); - BufferedWriter writer1 = FileUtil.getWriter(file, Charset.defaultCharset(), false); - writer1.write(""); - writer1.flush(); - writer1.close(); - } - - - } - -}