Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Jul 5, 2023
1 parent 1d1fd82 commit 4a077e8
Showing 1 changed file with 7 additions and 5 deletions.
12 changes: 7 additions & 5 deletions packages/jupyter-ai/jupyter_ai/chat_handlers/learn.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ async def _process_message(self, message: HumanChatMessage):
if args.list:
self.reply(self._build_list_response())
return

# Make sure the path exists.
if not len(args.path) == 1:
self.reply(f"{self.parser.format_usage()}", message)
Expand Down Expand Up @@ -165,7 +165,7 @@ async def delete_and_relearn(self):
if not self.metadata.dirs:
self.delete()
return

em_provider_cls, em_provider_args = self.get_embedding_provider()
curr_em_id = em_provider_cls.id + ":" + em_provider_args["model_id"]
prev_em_id = self.prev_em_id
Expand All @@ -176,8 +176,10 @@ async def delete_and_relearn(self):
# directly and break their instance.
if (prev_em_id is None) or (prev_em_id == curr_em_id):
return

self.log.info(f"Switching embedding provider from {prev_em_id} to {curr_em_id}.")

self.log.info(
f"Switching embedding provider from {prev_em_id} to {curr_em_id}."
)
message = f"""🔔 Hi there, it seems like you have updated the embeddings
model from `{prev_em_id}` to `{curr_em_id}`. I have to re-learn the
documents you had previously submitted for learning. Please wait to use
Expand Down Expand Up @@ -270,7 +272,7 @@ def get_embedding_provider(self):
em_provider_args = self.config_manager.get_em_provider_params()

return em_provider_cls, em_provider_args

def get_embedding_model(self):
em_provider_cls, em_provider_args = self.get_embedding_provider()
return em_provider_cls(**em_provider_args)

0 comments on commit 4a077e8

Please sign in to comment.