ExternalVendorCode/lollms-webui
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms-webui.git synced 2025-01-18 10:46:27 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed some bugs

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-06-11 00:53:26 +02:00
parent a998f1211c
commit 5ea2c4b70d
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
api/__init__.py
View File

@ -278,13 +278,13 @@ class ModelProcess:
try: try:
print(f" {personality}") print(f" {personality}")
personality_path = self.lollms_paths.personalities_zoo_path/f"{personality}" personality_path = self.lollms_paths.personalities_zoo_path/f"{personality}"
personality = AIPersonality(self.lollms_paths, personality_path, run_scripts=True) personality = AIPersonality(self.lollms_paths, personality_path, run_scripts=True, model=self.model)
self.mounted_personalities.append(personality) self.mounted_personalities.append(personality)
except Exception as ex: except Exception as ex:
print(f"Personality file not found or is corrupted ({personality_path}).\nPlease verify that the personality you have selected exists or select another personality. Some updates may lead to change in personality name or category, so check the personality selection in settings to be sure.") print(f"Personality file not found or is corrupted ({personality_path}).\nPlease verify that the personality you have selected exists or select another personality. Some updates may lead to change in personality name or category, so check the personality selection in settings to be sure.")
if self.config["debug"]: if self.config["debug"]:
print(ex) print(ex)
personality = AIPersonality(self.lollms_paths) personality = AIPersonality(self.lollms_paths, model=self.model)
failed_personalities.append(personality_path) failed_personalities.append(personality_path)
self._set_config_result['errors'].append(f"couldn't build personalities:{ex}") self._set_config_result['errors'].append(f"couldn't build personalities:{ex}")
@ -336,10 +336,11 @@ class ModelProcess:
if self.personality.processor_cfg is not None: if self.personality.processor_cfg is not None:
if "custom_workflow" in self.personality.processor_cfg: if "custom_workflow" in self.personality.processor_cfg:
if self.personality.processor_cfg["custom_workflow"]: if self.personality.processor_cfg["custom_workflow"]:
print("Running workflow") ASCIIColors.print(ASCIIColors.color_green,"Running workflow")
self.completion_signal.clear() self.completion_signal.clear()
self.start_signal.set() self.start_signal.set()
output = self.personality.processor.run_workflow(self._generate, command[1], command[0], self._callback)
output = self.personality.processor.run_workflow( command[1], command[0], self._callback)
self._callback(output, 0) self._callback(output, 0)
self.completion_signal.set() self.completion_signal.set()
self.start_signal.clear() self.start_signal.clear()
@ -587,7 +588,7 @@ class LoLLMsAPPI():
"message":"", "message":"",
"user_message_id": self.current_user_message_id, "user_message_id": self.current_user_message_id,
"ai_message_id": self.current_ai_message_id, "ai_message_id": self.current_ai_message_id,
}, room=request.sid }, room=self.current_room_id
) )
@socketio.on('generate_msg_from') @socketio.on('generate_msg_from')
@ -750,6 +751,8 @@ class LoLLMsAPPI():
""" """
if message_type == MSG_TYPE.MSG_TYPE_CHUNK: if message_type == MSG_TYPE.MSG_TYPE_CHUNK:
self.bot_says += chunk self.bot_says += chunk
if message_type == MSG_TYPE.MSG_TYPE_FULL:
self.bot_says = chunk
if message_type.value < 2: if message_type.value < 2:
self.socketio.emit('message', { self.socketio.emit('message', {
'data': self.bot_says, 'data': self.bot_says,