From b77bc884e8acd5add16eb5af1d7a33ce3fd00374 Mon Sep 17 00:00:00 2001 From: "Lumberbot (aka Jack)" <39504233+meeseeksmachine@users.noreply.github.com> Date: Sat, 27 Apr 2024 00:01:28 +0200 Subject: [PATCH] Backport PR #753: Load persisted vector store by default (#756) Co-authored-by: david qiu --- packages/jupyter-ai/jupyter_ai/chat_handlers/learn.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/packages/jupyter-ai/jupyter_ai/chat_handlers/learn.py b/packages/jupyter-ai/jupyter_ai/chat_handlers/learn.py index d05917374..a7d0d2343 100644 --- 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: