Skip to content

Commit

Permalink
Catch embedding model validation errors on extension init (jupyterlab…
Browse files Browse the repository at this point in the history
…#735)

* catch embedding model validation errors on extension init

* pre-commit
  • Loading branch information
dlqqq authored and Marchlak committed Oct 28, 2024
1 parent 29a47b0 commit 2c69e22
Showing 1 changed file with 16 additions and 10 deletions.
26 changes: 16 additions & 10 deletions packages/jupyter-ai/jupyter_ai/chat_handlers/learn.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,17 +67,23 @@ def __init__(self, *args, **kwargs):

def _load(self):
"""Loads the vector store."""
embeddings = self.get_embedding_model()
if not embeddings:
if self.index is not None:
return
if self.index is None:
try:
self.index = FAISS.load_local(
INDEX_SAVE_DIR, embeddings, index_name=self.index_name
)
self.load_metadata()
except Exception as e:
self.log.error("Could not load vector index from disk.")

try:
embeddings = self.get_embedding_model()
if not embeddings:
return

self.index = FAISS.load_local(
INDEX_SAVE_DIR, embeddings, index_name=self.index_name
)
self.load_metadata()
except Exception as e:
self.log.error(
"Could not load vector index from disk. Full exception details printed below."
)
self.log.error(e)

async def process_message(self, message: HumanChatMessage):
# If no embedding provider has been selected
Expand Down

0 comments on commit 2c69e22

Please sign in to comment.