From 6aff51d394352b8e2f378885a530bc9bdc6ee5e3 Mon Sep 17 00:00:00 2001 From: yudong Date: Wed, 14 Aug 2024 13:37:09 +0800 Subject: [PATCH] =?UTF-8?q?[fix][headless]metadata=E5=90=91=E9=87=8F?= =?UTF-8?q?=E6=A3=80=E7=B4=A2=E6=97=B6=E5=AF=B9modelid=E7=9A=84=E8=BF=87?= =?UTF-8?q?=E6=BB=A4=E5=A2=9E=E5=8A=A0=E4=B8=8B=E5=88=92=E7=BA=BF=20(#1570?= =?UTF-8?q?)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../headless/chat/knowledge/MetaEmbeddingService.java | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/knowledge/MetaEmbeddingService.java b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/knowledge/MetaEmbeddingService.java index 55cbe8abb..22ea68a57 100644 --- a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/knowledge/MetaEmbeddingService.java +++ b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/knowledge/MetaEmbeddingService.java @@ -2,6 +2,7 @@ package com.tencent.supersonic.headless.chat.knowledge; import com.tencent.supersonic.common.config.EmbeddingConfig; import com.tencent.supersonic.common.pojo.Constants; +import com.tencent.supersonic.common.pojo.enums.DictWordType; import com.tencent.supersonic.common.service.EmbeddingService; import com.tencent.supersonic.headless.chat.knowledge.helper.NatureHelper; import dev.langchain4j.store.embedding.Retrieval; @@ -37,7 +38,8 @@ public class MetaEmbeddingService { if (CollectionUtils.isNotEmpty(allModels) && allModels.size() == 1) { Map filterCondition = new HashMap<>(); - filterCondition.put("modelId", allModels.stream().findFirst().get().toString()); + String modelId = allModels.stream().findFirst().get().toString(); + filterCondition.put("modelId", modelId + DictWordType.NATURE_SPILT); retrieveQuery.setFilterCondition(filterCondition); }