From 61f2d6bae474cc215fab6b03c2b60fb2d9309d7f Mon Sep 17 00:00:00 2001
From: Saifeddine ALOUI <aloui.seifeddine@gmail.com>
Date: Sat, 2 Dec 2023 00:53:37 +0100
Subject: [PATCH] Upgraded to new version

---
 api/__init__.py        | 11 ++++++++---
 app.py                 |  4 +++-
 lollms_core            |  2 +-
 zoos/models_zoo        |  2 +-
 zoos/personalities_zoo |  2 +-
 5 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/api/__init__.py b/api/__init__.py
index b41d29d0..bced7b3e 100644
--- a/api/__init__.py
+++ b/api/__init__.py
@@ -981,6 +981,8 @@ class LoLLMsAPI(LollmsApplication):
                                                     selected_language=personality.split(":")[1] if ":" in personality else None,
                                                     installation_option=InstallOption.FORCE_INSTALL)
                         mounted_personalities.append(personality)
+                        if personality.processor:
+                            personality.processor.mounted()
                     except Exception as ex:
                         ASCIIColors.error(f"Couldn't load personality at {personality_path}")
                         trace_exception(ex)
@@ -994,6 +996,9 @@ class LoLLMsAPI(LollmsApplication):
                                                     run_scripts=True,
                                                     installation_option=InstallOption.FORCE_INSTALL)
                         mounted_personalities.append(personality)
+                        if personality.processor:
+                            personality.processor.mounted()
+
                         ASCIIColors.info("Reverted to default personality")
         if self.config["active_personality_id"]>=0 and self.config["active_personality_id"]<len(self.config["personalities"]):
             ASCIIColors.success(f'selected model : {self.config["personalities"][self.config["active_personality_id"]]}')
@@ -1553,10 +1558,10 @@ class LoLLMsAPI(LollmsApplication):
                         self, 
                         chunk:str, 
                         message_type:MSG_TYPE,
-
                         parameters:dict=None, 
                         metadata:list=None, 
-                        client_id:int=0
+                        client_id:int=0,
+                        personality:AIPersonality=None
                     ):
         """
         Processes a chunk of generated text
@@ -1592,7 +1597,7 @@ class LoLLMsAPI(LollmsApplication):
             self.start_time = datetime.now()
             self.new_message(
                                 client_id, 
-                                self.personality.name, 
+                                self.personality.name if personality is None else personality.name, 
                                 chunk if parameters["type"]!=MSG_TYPE.MSG_TYPE_UI.value else '', 
                                 metadata = [{
                                     "title":chunk,
diff --git a/app.py b/app.py
index 9bb4e92c..5bacba56 100644
--- a/app.py
+++ b/app.py
@@ -2209,7 +2209,9 @@ try:
             print(f"- Selecting active personality {id} ...",end="")
             if id<len(self.mounted_personalities):
                 self.config["active_personality_id"]=id
-                self.personality = self.mounted_personalities[self.config["active_personality_id"]]
+                self.personality:AIPersonality = self.mounted_personalities[self.config["active_personality_id"]]
+                if self.personality.processor:
+                    self.personality.processor.selected()
                 ASCIIColors.success("ok")
                 print(f"Selected {self.personality.name}")
                 if self.config.auto_save:
diff --git a/lollms_core b/lollms_core
index f38b468d..bae2a71f 160000
--- a/lollms_core
+++ b/lollms_core
@@ -1 +1 @@
-Subproject commit f38b468d1b5a03eddcd7a9940ac5d33b28a95b8f
+Subproject commit bae2a71fa62c4408c311770f2b98bd43f01622bc
diff --git a/zoos/models_zoo b/zoos/models_zoo
index 836732fd..1faa8a7c 160000
--- a/zoos/models_zoo
+++ b/zoos/models_zoo
@@ -1 +1 @@
-Subproject commit 836732fd0c2892fc00e6e98bd1452df4fed17efa
+Subproject commit 1faa8a7c1edbb79cd27a8b90cf599b6c70dff207
diff --git a/zoos/personalities_zoo b/zoos/personalities_zoo
index 7a98155c..67a62dc6 160000
--- a/zoos/personalities_zoo
+++ b/zoos/personalities_zoo
@@ -1 +1 @@
-Subproject commit 7a98155c77b1a7e8eacb08e8e1981dea2a04063d
+Subproject commit 67a62dc6bf1d4417da1f2fb2e1c6c9a09f7404f6