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

[improvement](chat) udpate EmbeddingConfig embeddingStorePersistentPath and if the persistentPath directory does not exist, create the directory. (#550)

Browse Source
This commit is contained in:
lexluo09
2023-12-19 16:12:11 +08:00
committed by GitHub
parent 1d9b6d6877
commit f60c1675cd
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
common/src/main/java/com/tencent/supersonic/common/config/EmbeddingConfig.java
View File

@@ -32,6 +32,6 @@ public class EmbeddingConfig {
@Value("${embedding.metric.analyzeQuery.nResult:5}") @Value("${embedding.metric.analyzeQuery.nResult:5}")
private int metricAnalyzeQueryResultNum; private int metricAnalyzeQueryResultNum;
@Value("${embeddingStore.persistent.path:/tmp}") @Value("${inMemoryEmbeddingStore.persistent.path:/tmp}")
private String embeddingStorePersistentPath; private String embeddingStorePersistentPath;
} }

4
common/src/main/java/com/tencent/supersonic/common/util/embedding/InMemoryS2EmbeddingStore.java
View File

@@ -71,6 +71,10 @@ public class InMemoryS2EmbeddingStore implements S2EmbeddingStore {
for (Entry<String, InMemoryEmbeddingStore<EmbeddingQuery>> entry : collectionNameToStore.entrySet()) { for (Entry<String, InMemoryEmbeddingStore<EmbeddingQuery>> entry : collectionNameToStore.entrySet()) {
Path filePath = getPersistentPath(entry.getKey()); Path filePath = getPersistentPath(entry.getKey());
try { try {
Path directoryPath = filePath.getParent();
if (!Files.exists(directoryPath)) {
Files.createDirectories(directoryPath);
}
entry.getValue().serializeToFile(filePath); entry.getValue().serializeToFile(filePath);
} catch (Exception e) { } catch (Exception e) {
log.error("persistentToFile error, persistentFile:" + filePath, e); log.error("persistentToFile error, persistentFile:" + filePath, e);