Backport PR jupyterlab#753: Load persisted vector store by default

meeseeksmachine · Apr 26, 2024 · 9f1a0d7 · 9f1a0d7
1 parent b7cb8d9
commit 9f1a0d7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/packages/jupyter-ai/jupyter_ai/chat_handlers/learn.py b/packages/jupyter-ai/jupyter_ai/chat_handlers/learn.py
@@ -76,7 +76,10 @@ def _load(self):
                 return
 
             self.index = FAISS.load_local(
-                INDEX_SAVE_DIR, embeddings, index_name=self.index_name
+                INDEX_SAVE_DIR,
+                embeddings,
+                index_name=self.index_name,
+                allow_dangerous_deserialization=True,
             )
             self.load_metadata()
         except Exception as e: