From 4a077e8de6264b588703233605a3303d9ddb8ba8 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 5 Jul 2023 17:54:56 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- .../jupyter-ai/jupyter_ai/chat_handlers/learn.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/packages/jupyter-ai/jupyter_ai/chat_handlers/learn.py b/packages/jupyter-ai/jupyter_ai/chat_handlers/learn.py index 42f591b27..b8296d393 100644 --- a/packages/jupyter-ai/jupyter_ai/chat_handlers/learn.py +++ b/packages/jupyter-ai/jupyter_ai/chat_handlers/learn.py @@ -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) @@ -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 @@ -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 @@ -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)