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

Added information about install

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-12-06 01:27:59 +01:00
parent 8388d197d2
commit dfb5470d8d
3 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
lollms/app.py
View File

@ -226,9 +226,9 @@ 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=True, client_id=None): def notify(self, content, is_success=True, duration=4, client_id=None, notification_type=0):
if self.notification_callback: if self.notification_callback:
return self.notification_callback(content, is_success, client_id) return self.notification_callback(content, is_success, duration, client_id, notification_type)
if is_success: if is_success:
ASCIIColors.yellow(content) ASCIIColors.yellow(content)
@ -240,13 +240,16 @@ class LollmsApplication:
binding = BindingBuilder().build_binding(self.config, self.lollms_paths, notification_callback=self.notify) binding = BindingBuilder().build_binding(self.config, self.lollms_paths, notification_callback=self.notify)
return binding return binding
except Exception as ex: except Exception as ex:
self.notify("Couldn't load binding", False)
self.notify("Trying to reinstall binding")
print(ex) print(ex)
print(f"Couldn't find binding. Please verify your configuration file at {self.lollms_paths.personal_configuration_path}/local_configs.yaml or use the next menu to select a valid binding") print(f"Couldn't find binding. Please verify your configuration file at {self.lollms_paths.personal_configuration_path}/local_configs.yaml or use the next menu to select a valid binding")
print(f"Trying to reinstall binding") print(f"Trying to reinstall binding")
try: try:
binding = BindingBuilder().build_binding(self.config, self.lollms_paths,installation_option=InstallOption.FORCE_INSTALL) binding = BindingBuilder().build_binding(self.config, self.lollms_paths,installation_option=InstallOption.FORCE_INSTALL)
except Exception as ex: except Exception as ex:
ASCIIColors.error("Couldn't reinstall model") self.notify("Couldn't reinstall binding", False)
ASCIIColors.error("Couldn't reinstall binding")
trace_exception(ex) trace_exception(ex)
return None return None
@ -258,6 +261,7 @@ class LollmsApplication:
if personality is not None: if personality is not None:
personality.model = model personality.model = model
except Exception as ex: except Exception as ex:
self.notify("Couldn't load model.", False)
ASCIIColors.error(f"Couldn't load model. Please verify your configuration file at {self.lollms_paths.personal_configuration_path} or use the next menu to select a valid model") ASCIIColors.error(f"Couldn't load model. Please verify your configuration file at {self.lollms_paths.personal_configuration_path} or use the next menu to select a valid model")
ASCIIColors.error(f"Binding returned this exception : {ex}") ASCIIColors.error(f"Binding returned this exception : {ex}")
trace_exception(ex) trace_exception(ex)

2
lollms/binding.py
View File

@ -103,7 +103,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=True): def notify(self, content:str, status:bool=True, duration=4, notification_type=0):
if self.notification_callback: if self.notification_callback:
self.notification_callback(content, status) self.notification_callback(content, status)

2
setup.py
View File

@ -26,7 +26,7 @@ def get_all_files(path):
setuptools.setup( setuptools.setup(
name="lollms", name="lollms",
version="6.6.0", version="6.7.0",
author="Saifeddine ALOUI", author="Saifeddine ALOUI",
author_email="aloui.saifeddine@gmail.com", author_email="aloui.saifeddine@gmail.com",
description="A python library for AI personality definition", description="A python library for AI personality definition",