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

fix-common-因缓存无法存记忆 (#2181)

Browse Source
This commit is contained in:
guilinlewis
2025-03-21 22:30:38 +08:00
committed by GitHub
parent 3ffc8c3d9e
commit 81cd60d2da
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
common/src/main/java/com/tencent/supersonic/common/service/impl/EmbeddingServiceImpl.java
View File

@@ -49,7 +49,7 @@ public class EmbeddingServiceImpl implements EmbeddingService {
try {
EmbeddingModel embeddingModel = ModelProvider.getEmbeddingModel();
Embedding embedding = embeddingModel.embed(question).content();
boolean existSegment = existSegment(embeddingStore, query, embedding);
boolean existSegment = existSegment(collectionName,embeddingStore, query, embedding);
if (existSegment) {
continue;
}
@@ -62,14 +62,14 @@ public class EmbeddingServiceImpl implements EmbeddingService {
}
}
private boolean existSegment(EmbeddingStore embeddingStore, TextSegment query,
private boolean existSegment(String collectionName,EmbeddingStore embeddingStore, TextSegment query,
Embedding embedding) {
String queryId = TextSegmentConvert.getQueryId(query);
if (queryId == null) {
return false;
}
// Check cache first
Boolean cachedResult = cache.getIfPresent(queryId);
Boolean cachedResult = cache.getIfPresent(collectionName+queryId);
if (cachedResult != null) {
return cachedResult;
}
@@ -82,7 +82,7 @@ public class EmbeddingServiceImpl implements EmbeddingService {
EmbeddingSearchResult result = embeddingStore.search(request);
List<EmbeddingMatch<TextSegment>> relevant = result.matches();
boolean exists = CollectionUtils.isNotEmpty(relevant);
cache.put(queryId, exists);
cache.put(collectionName+queryId, exists);
return exists;
}