ソースを参照

Update embedding_manager: create dir for cache

StrayWarrior 5 ヶ月 前
コミット
e04846e0ac
1 ファイル変更3 行追加1 行削除
  1. 3 1
      applications/embedding_manager.py

+ 3 - 1
applications/embedding_manager.py

@@ -6,7 +6,7 @@ import numpy as np
 
 
 
 
 class EmbeddingManager:
 class EmbeddingManager:
-    def __init__(self, model, emb_size = 1024, cache_file="embedding_cache", save_interval=600):
+    def __init__(self, model, emb_size=1024, cache_file="cache/embedding_cache", save_interval=600):
         self.model = model
         self.model = model
         self.emb_size = emb_size
         self.emb_size = emb_size
         self.cache_file = cache_file
         self.cache_file = cache_file
@@ -67,6 +67,8 @@ class EmbeddingManager:
 
 
     def save_now(self):
     def save_now(self):
         """Manually trigger a save to disk."""
         """Manually trigger a save to disk."""
+        if os.path.dirname(self.cache_file):
+            os.makedirs(os.path.dirname(self.cache_file), 0o755, True)
         tmp_cache_file = self.cache_file + ".tmp"
         tmp_cache_file = self.cache_file + ".tmp"
         tmp_cache_key_file = self.cache_key_file + ".tmp"
         tmp_cache_key_file = self.cache_key_file + ".tmp"
         with self.lock:  # Ensure thread-safe access
         with self.lock:  # Ensure thread-safe access