From 3568ca8d5e058dc5d9792995cca949fcec13e124 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Sat, 25 Nov 2023 00:04:15 +0100 Subject: [PATCH] Upgraded notification --- lollms/app.py | 2 +- lollms/binding.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/lollms/app.py b/lollms/app.py index ee97835..c7327fb 100644 --- a/lollms/app.py +++ b/lollms/app.py @@ -149,7 +149,7 @@ class LollmsApplication: generated_text = self.personality.model.generate(full_discussion, n_predict=n_predict, callback=callback) return generated_text - def notify(self, content, is_success, client_id=None): + def notify(self, content, is_success=True, client_id=None): if self.notification_callback: return self.notification_callback(content, is_success, client_id) diff --git a/lollms/binding.py b/lollms/binding.py index 906d382..84e53e8 100644 --- a/lollms/binding.py +++ b/lollms/binding.py @@ -92,7 +92,7 @@ class LLMBinding: for models_folder in self.models_folders: models_folder.mkdir(parents=True, exist_ok=True) - def notify(self, content:str, status:bool): + def notify(self, content:str, status:bool=True): if self.notification_callback: self.notification_callback(content, status)