From 51e7a9bcd762855476a6fdfec4fad173b29fad7e Mon Sep 17 00:00:00 2001 From: yudong Date: Wed, 11 Sep 2024 22:07:23 +0800 Subject: [PATCH] =?UTF-8?q?(fix)(chat)=20=E4=BF=AE=E5=A4=8D=E5=81=9Aembedd?= =?UTF-8?q?ing=E6=A3=80=E7=B4=A2=E6=97=B6=EF=BC=8C=E5=87=BA=E7=8E=B0?= =?UTF-8?q?=E9=87=8D=E5=A4=8D=E6=95=B0=E6=8D=AE=E7=9A=84=E9=97=AE=E9=A2=98?= =?UTF-8?q?=20(#1643)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../supersonic/common/service/impl/EmbeddingServiceImpl.java | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/common/src/main/java/com/tencent/supersonic/common/service/impl/EmbeddingServiceImpl.java b/common/src/main/java/com/tencent/supersonic/common/service/impl/EmbeddingServiceImpl.java index 6d46ce824..796ebaa44 100644 --- a/common/src/main/java/com/tencent/supersonic/common/service/impl/EmbeddingServiceImpl.java +++ b/common/src/main/java/com/tencent/supersonic/common/service/impl/EmbeddingServiceImpl.java @@ -59,6 +59,7 @@ public class EmbeddingServiceImpl implements EmbeddingService { continue; } embeddingStore.add(embedding, query); + cache.put(TextSegmentConvert.getQueryId(query), true); } catch (Exception e) { log.error( "embeddingModel embed error question: {}, embeddingStore: {}", @@ -117,6 +118,9 @@ public class EmbeddingServiceImpl implements EmbeddingService { new MetadataFilterBuilder(TextSegmentConvert.QUERY_ID); Filter filter = filterBuilder.isIn(queryIds); inMemoryEmbeddingStore.removeAll(filter); + for (String queryId : queryIds) { + cache.put(queryId, false); + } } } else { throw new RuntimeException("Not supported yet.");