[improvement](chat) udpate EmbeddingConfig embeddingStorePersistentPa…

…th and if the persistentPath directory does not exist, create the directory. (#550)
tencentmusic · Dec 19, 2023 · f60c167 · f60c167
1 parent 1d9b6d6
commit f60c167
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/common/src/main/java/com/tencent/supersonic/common/config/EmbeddingConfig.java b/common/src/main/java/com/tencent/supersonic/common/config/EmbeddingConfig.java
@@ -32,6 +32,6 @@ public class EmbeddingConfig {
     @Value("${embedding.metric.analyzeQuery.nResult:5}")
     private int metricAnalyzeQueryResultNum;
 
-    @Value("${embeddingStore.persistent.path:/tmp}")
+    @Value("${inMemoryEmbeddingStore.persistent.path:/tmp}")
     private String embeddingStorePersistentPath;
 }
diff --git a/.../src/main/java/com/tencent/supersonic/common/util/embedding/InMemoryS2EmbeddingStore.java b/.../src/main/java/com/tencent/supersonic/common/util/embedding/InMemoryS2EmbeddingStore.java
@@ -71,6 +71,10 @@ public void persistentToFile() {
         for (Entry<String, InMemoryEmbeddingStore<EmbeddingQuery>> entry : collectionNameToStore.entrySet()) {
             Path filePath = getPersistentPath(entry.getKey());
             try {
+                Path directoryPath = filePath.getParent();
+                if (!Files.exists(directoryPath)) {
+                    Files.createDirectories(directoryPath);
+                }
                 entry.getValue().serializeToFile(filePath);
             } catch (Exception e) {
                 log.error("persistentToFile error, persistentFile:" + filePath, e);