From 81e7c4fd5754599fab4122bc494acd0be9761b94 Mon Sep 17 00:00:00 2001 From: xueqingkun Date: Fri, 19 Jan 2024 12:45:56 +0800 Subject: [PATCH] =?UTF-8?q?rasa:=20text2vecData=20=E4=BF=AE=E5=A4=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../com/supervision/rasa/service/Text2vecServiceImpl.java | 6 ++++-- .../supervision/rasa/service/impl/RasaCmdServiceImpl.java | 5 ++++- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/virtual-patient-rasa/src/main/java/com/supervision/rasa/service/Text2vecServiceImpl.java b/virtual-patient-rasa/src/main/java/com/supervision/rasa/service/Text2vecServiceImpl.java index 62f576ad..6ad53710 100644 --- a/virtual-patient-rasa/src/main/java/com/supervision/rasa/service/Text2vecServiceImpl.java +++ b/virtual-patient-rasa/src/main/java/com/supervision/rasa/service/Text2vecServiceImpl.java @@ -86,9 +86,11 @@ public class Text2vecServiceImpl implements Text2vecService { Map intentCodeAndIdMap = rasaCmdService.getIntentCodeAndIdMap(); // 更新text2vec数据信息 List text2vecDataVoList = intentCodeAndIdMap.entrySet().stream() - .filter(entry -> Objects.nonNull(entry.getValue()) && CollUtil.isNotEmpty(entry.getValue().getQuestionList())) + .filter(entry -> Objects.nonNull(entry.getValue()) + && CollUtil.isNotEmpty(entry.getValue().getQuestionList()) + && CollUtil.isNotEmpty(entry.getValue().getAnswerList())) .flatMap(entry -> entry.getValue().getQuestionList().stream() - .map(question -> new Text2vecDataVo(entry.getKey(), question))).collect(Collectors.toList()); + .map(question -> new Text2vecDataVo(entry.getValue().getAnswerList().get(0), question))).collect(Collectors.toList()); if (CollUtil.isEmpty(text2vecDataVoList)){ log.info("text2vec数据初始化失败，text2vec数据为空...."); return; diff --git a/virtual-patient-rasa/src/main/java/com/supervision/rasa/service/impl/RasaCmdServiceImpl.java b/virtual-patient-rasa/src/main/java/com/supervision/rasa/service/impl/RasaCmdServiceImpl.java index 2425d6ec..038a55d5 100644 --- a/virtual-patient-rasa/src/main/java/com/supervision/rasa/service/impl/RasaCmdServiceImpl.java +++ b/virtual-patient-rasa/src/main/java/com/supervision/rasa/service/impl/RasaCmdServiceImpl.java @@ -219,8 +219,11 @@ public class RasaCmdServiceImpl implements RasaCmdService { // 更新text2vec数据信息 List text2vecDataVoList = questionAnswerDTOMap.entrySet().stream() + .filter(entry -> Objects.nonNull(entry.getValue()) + && CollUtil.isNotEmpty(entry.getValue().getQuestionList()) + && CollUtil.isNotEmpty(entry.getValue().getAnswerList())) .flatMap(entry -> entry.getValue().getQuestionList().stream() - .map(question -> new Text2vecDataVo(entry.getKey(), question))).collect(Collectors.toList()); + .map(question -> new Text2vecDataVo(entry.getValue().getAnswerList().get(0), question))).collect(Collectors.toList()); text2vecService.updateDataset(text2vecDataVoList); return true;