rasa : 修改 RasaModelManager

dev_2.1.0
xueqingkun 1 year ago
parent 1ad36e28e1
commit 2c444d8baf

@ -5,12 +5,13 @@ FROM rasa_dev:1.1.0
COPY ./docs/docker-entrypoint.sh /usr/local/bin/docker-entrypoint.sh COPY ./docs/docker-entrypoint.sh /usr/local/bin/docker-entrypoint.sh
# 设置工作目录 # 设置工作目录
WORKDIR /home/app WORKDIR /data/vp
# 复制java jar 到容器中 # 复制java jar 到容器中
COPY target/virtual-patient-rasa-1.0-SNAPSHOT.jar /data/vp/virtual-patient-rasa-1.0-SNAPSHOT.jar COPY target/virtual-patient-rasa-1.0-SNAPSHOT.jar /data/vp/virtual-patient-rasa-1.0-SNAPSHOT.jar
# 复制rasa配置文件到 rasa目录下 # 复制rasa配置文件到 rasa目录下
COPY docs/rasa /rasa COPY docs/rasa /rasa
COPY docs/1 /data/vp/rasa/models/1
# 暴漏服务端口 # 暴漏服务端口
EXPOSE 8890 EXPOSE 8890

@ -61,9 +61,10 @@ public class RasaModelManager {
log.info("wakeUpInterruptServer: rasa model path {} not exist,attempt find last ...",rasaModelPath); log.info("wakeUpInterruptServer: rasa model path {} not exist,attempt find last ...",rasaModelPath);
String modeParentPath = replaceDuplicateSeparator(String.join(File.separator, modelsPath, rasaModelInfo.getModelId())); String modeParentPath = replaceDuplicateSeparator(String.join(File.separator, modelsPath, rasaModelInfo.getModelId()));
String fixedModePath = listLastFilePath(modeParentPath, f -> f.getName().matches("-?\\d+(\\.\\d+)?.tar.gz")); String fixedModePath = listLastFilePath(modeParentPath, f -> f.getName().matches("-?\\d+(\\.\\d+)?.tar.gz"));
Assert.notEmpty(fixedModePath,"wakeUpInterruptService: rasa model path {} not exist,attempt find last ...",rasaModelPath); Assert.notEmpty(fixedModePath,"wakeUpInterruptService: no rasa model in path {} ",modeParentPath);
rasaRunParam.setRasaModelPath(fixedModePath); rasaRunParam.setRasaModelPath(fixedModePath);
} }
log.info("wakeUpInterruptServer : use fixedModePath :{}",rasaRunParam.getRasaModelPath());
List<String> outMessageList = rasaCmdService.execCmd(rasaRunParam.toList(), List<String> outMessageList = rasaCmdService.execCmd(rasaRunParam.toList(),
s -> StrUtil.isNotBlank(s) && s.contains(RasaConstant.RUN_SUCCESS_MESSAGE), 300); s -> StrUtil.isNotBlank(s) && s.contains(RasaConstant.RUN_SUCCESS_MESSAGE), 300);

Loading…
Cancel
Save