ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-18 20:27:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Upgraded notification

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-11-25 00:04:15 +01:00
parent 8b7690d0d7
commit 3568ca8d5e
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lollms/app.py
View File

@ -149,7 +149,7 @@ class LollmsApplication:
generated_text = self.personality.model.generate(full_discussion, n_predict=n_predict, callback=callback) generated_text = self.personality.model.generate(full_discussion, n_predict=n_predict, callback=callback)
return generated_text 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: if self.notification_callback:
return self.notification_callback(content, is_success, client_id) return self.notification_callback(content, is_success, client_id)

2
lollms/binding.py
View File

@ -92,7 +92,7 @@ class LLMBinding:
for models_folder in self.models_folders: for models_folder in self.models_folders:
models_folder.mkdir(parents=True, exist_ok=True) 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: if self.notification_callback:
self.notification_callback(content, status) self.notification_callback(content, status)