diff --git a/packages/jupyter-ai/jupyter_ai/chat_handlers/learn.py b/packages/jupyter-ai/jupyter_ai/chat_handlers/learn.py index 38390a44c..f9348a21f 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: