topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-13 04:57:28 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[fix][headless]metadata向量检索时对modelid的过滤增加下划线 (#1570)

Browse Source
This commit is contained in:
yudong
2024-08-14 13:37:09 +08:00
committed by GitHub
parent 8d29e89317
commit 6aff51d394
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
headless/chat/src/main/java/com/tencent/supersonic/headless/chat/knowledge/MetaEmbeddingService.java
View File

@@ -2,6 +2,7 @@ package com.tencent.supersonic.headless.chat.knowledge;
import com.tencent.supersonic.common.config.EmbeddingConfig; import com.tencent.supersonic.common.config.EmbeddingConfig;
import com.tencent.supersonic.common.pojo.Constants; 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.common.service.EmbeddingService;
import com.tencent.supersonic.headless.chat.knowledge.helper.NatureHelper; import com.tencent.supersonic.headless.chat.knowledge.helper.NatureHelper;
import dev.langchain4j.store.embedding.Retrieval; import dev.langchain4j.store.embedding.Retrieval;
@@ -37,7 +38,8 @@ public class MetaEmbeddingService {
if (CollectionUtils.isNotEmpty(allModels) && allModels.size() == 1) { if (CollectionUtils.isNotEmpty(allModels) && allModels.size() == 1) {
Map<String, String> filterCondition = new HashMap<>(); Map<String, String> 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); retrieveQuery.setFilterCondition(filterCondition);
} }