Opt-in memory collections' snapshopts

.env.example

@@ -15,3 +15,6 @@ CORE_PORT=1865
 
 # Log levels
 LOG_LEVEL=WARNING
+
+# Turn on memory collections' snapshots on embedder change with SAVE_MEMORY_SNAPSHOTS=true
+SAVE_MEMORY_SNAPSHOTS=false

core/cat/memory/vector_memory.py

@@ -129,9 +129,12 @@ def check_embedding_size(self):
             log.info(f'Collection "{self.collection_name}" has the same embedder')
         else:
             log.warning(f'Collection "{self.collection_name}" has different embedder')
-            # dump collection on disk before deleting
-            self.save_dump()
-            log.info(f'Dump "{self.collection_name}" completed')
+            # Memory snapshot saving can be turned off in the .env file with:
+            # SAVE_MEMORY_SNAPSHOTS=false
+            if os.getenv("SAVE_MEMORY_SNAPSHOTS") == "true":
+                # dump collection on disk before deleting
+                self.save_dump()
+                log.info(f'Dump "{self.collection_name}" completed')
 
             self.client.delete_collection(self.collection_name)
             log.warning(f'Collection "{self.collection_name}" deleted')

docker-compose.yml

@@ -19,6 +19,7 @@ services:
       - API_KEY=${API_KEY:-}
       - LOG_LEVEL=${LOG_LEVEL:-WARNING}
       - DEBUG=${DEBUG:-true}
+      - SAVE_MEMORY_SNAPSHOTS=${SAVE_MEMORY_SNAPSHOTS:-false}
     ports:
       - ${CORE_PORT:-1865}:80
     volumes: