Added personality to new message

ParisNeo · Dec 1, 2023 · bae2a71 · bae2a71
1 parent f38b468
commit bae2a71
Showing 1 changed file with 17 additions and 5 deletions.
diff --git a/lollms/personality.py b/lollms/personality.py
@@ -194,7 +194,7 @@ def __init__(
             self.personality_output_folder = lollms_paths.personal_outputs_path/self.name
             self.personality_output_folder.mkdir(parents=True, exist_ok=True)
 
-    def new_message(self, message_text:str, message_type:MSG_TYPE= MSG_TYPE.MSG_TYPE_FULL, metadata=[], callback: Callable[[str, int, dict, list], bool]=None):
+    def new_message(self, message_text:str, message_type:MSG_TYPE= MSG_TYPE.MSG_TYPE_FULL, metadata=[], callback: Callable[[str, int, dict, list, Any], bool]=None):
         """This sends step rogress to front end
 
         Args:
@@ -205,7 +205,7 @@ def new_message(self, message_text:str, message_type:MSG_TYPE= MSG_TYPE.MSG_TYPE
             callback = self.callback
 
         if callback:
-            callback(message_text, MSG_TYPE.MSG_TYPE_NEW_MESSAGE, parameters={'type':message_type.value,'metadata':metadata})
+            callback(message_text, MSG_TYPE.MSG_TYPE_NEW_MESSAGE, parameters={'type':message_type.value,'metadata':metadata}, personality=self)
 
     def full(self, full_text:str, callback: Callable[[str, MSG_TYPE, dict, list], bool]=None):
         """This sends full text to front end
@@ -1545,6 +1545,18 @@ def __init__(
         else:
             self.load_personality_config()
 
+    def mounted(self):
+        """
+        triggered when mounted
+        """
+        pass
+
+    def selected(self):
+        """
+        triggered when mounted
+        """
+        pass
+
     def execute_command(self, command: str, parameters:list=[]):
         """
         Recovers user commands and executes them. Each personality can define a set of commands that they can receive and execute
@@ -2237,7 +2249,7 @@ def info(self, info_text:str, callback: Callable[[str, MSG_TYPE, dict, list], bo
         if callback:
             callback(info_text, MSG_TYPE.MSG_TYPE_FULL)
 
-    def step_progress(self, step_text:str, progress:float, callback: Callable[[str, MSG_TYPE, dict, list], bool]=None):
+    def step_progress(self, step_text:str, progress:float, callback: Callable[[str, MSG_TYPE, dict, list, AIPersonality], bool]=None):
         """This sends step rogress to front end
 
         Args:
@@ -2250,7 +2262,7 @@ def step_progress(self, step_text:str, progress:float, callback: Callable[[str,
         if callback:
             callback(step_text, MSG_TYPE.MSG_TYPE_STEP_PROGRESS, {'progress':progress})
 
-    def new_message(self, message_text:str, message_type:MSG_TYPE= MSG_TYPE.MSG_TYPE_FULL, metadata=[], callback: Callable[[str, int, dict, list], bool]=None):
+    def new_message(self, message_text:str, message_type:MSG_TYPE= MSG_TYPE.MSG_TYPE_FULL, metadata=[], callback: Callable[[str, int, dict, list, AIPersonality], bool]=None):
         """This sends step rogress to front end
 
         Args:
@@ -2261,7 +2273,7 @@ def new_message(self, message_text:str, message_type:MSG_TYPE= MSG_TYPE.MSG_TYPE
             callback = self.callback
 
         if callback:
-            callback(message_text, MSG_TYPE.MSG_TYPE_NEW_MESSAGE, parameters={'type':message_type.value,'metadata':metadata})
+            callback(message_text, MSG_TYPE.MSG_TYPE_NEW_MESSAGE, parameters={'type':message_type.value,'metadata':metadata},personality = self.personality)
 
     def finished_message(self, message_text:str="", callback: Callable[[str, MSG_TYPE, dict, list], bool]=None):
         """This sends step rogress to front end