Backport PR #753: Load persisted vector store by default (#756)

Co-authored-by: david qiu <[email protected]>
jupyterlab · Apr 26, 2024 · b77bc88 · b77bc88
1 parent b7cb8d9
commit b77bc88
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: