From 8f620480c68d1d78adf9912d7c2851e65c1b9015 Mon Sep 17 00:00:00 2001 From: JackLiang Date: Fri, 26 Jul 2024 18:07:49 +0800 Subject: [PATCH] =?UTF-8?q?fix=EF=BC=9A=E5=8A=A0=E8=BD=BD=E6=A8=A1?= =?UTF-8?q?=E5=9E=8B=E5=88=B0=E5=86=85=E5=AD=98=E6=97=B6=E5=86=85=E5=AD=98?= =?UTF-8?q?=E6=B3=84=E9=9C=B2bug=20(#1465)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * fix existSegment bug which embedding search not has min-score,so that it will return true. --- .../langchain4j/model/embedding/S2OnnxEmbeddingModel.java | 6 ++++-- .../supersonic/headless/chat/corrector/TimeCorrector.java | 3 +-- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/common/src/main/java/dev/langchain4j/model/embedding/S2OnnxEmbeddingModel.java b/common/src/main/java/dev/langchain4j/model/embedding/S2OnnxEmbeddingModel.java index 5a376786f..92f7374ce 100644 --- a/common/src/main/java/dev/langchain4j/model/embedding/S2OnnxEmbeddingModel.java +++ b/common/src/main/java/dev/langchain4j/model/embedding/S2OnnxEmbeddingModel.java @@ -19,7 +19,7 @@ import java.nio.file.Paths; */ public class S2OnnxEmbeddingModel extends AbstractInProcessEmbeddingModel { - private final OnnxBertBiEncoder model; + private static OnnxBertBiEncoder model = null; /** * @param pathToModel The path to the .onnx model file (e.g., "/home/me/model.onnx"). @@ -33,7 +33,9 @@ public class S2OnnxEmbeddingModel extends AbstractInProcessEmbeddingModel { throw new RuntimeException(e); } } - model = loadFromFileSystem(Paths.get(pathToModel), resource); + if (model == null) { + model = loadFromFileSystem(Paths.get(pathToModel), resource); + } } /** diff --git a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/corrector/TimeCorrector.java b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/corrector/TimeCorrector.java index 4629f0e40..bd68e2eeb 100644 --- a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/corrector/TimeCorrector.java +++ b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/corrector/TimeCorrector.java @@ -108,5 +108,4 @@ public class TimeCorrector extends BaseSemanticCorrector { semanticParseInfo.getSqlInfo().setCorrectedS2SQL(correctS2SQL); } } - -} +} \ No newline at end of file