Fixing bugs and ollama errors

Messaging_system/LLM.py

@@ -39,7 +39,7 @@ class LLM:
         :return:
         """
         openai_models = ["4o-mini", "gpt-4o", "gpt-4.1-nano"]
-        ollama_models = ["deepseek-r1:1.5b", "gemma3:4b", "deepseek-r1:7b"]
+        ollama_models = ["deepseek-r1:1.5b", "gemma3:4b", "deepseek-r1:7b", "gemma3:4b"]
 
         if self.Core.model in openai_models:
             self.model_type = "openai"
@@ -167,6 +167,8 @@ class LLM:
                             print(
                                 f"'header' or 'message' is more than specified characters in response on attempt {attempt + 1}. Retrying...")
                             continue
+                        else:
+                            return output
 
                 except json.JSONDecodeError:
                     print(f"Invalid JSON from LLM on attempt {attempt + 1}. Retrying...")

Messaging_system/Permes.py

@@ -84,7 +84,7 @@ class Permes:
             personalize_message.set_features_to_use(selected_source_features)
 
         if model is not None:
-            personalize_message.set_llm_model(selected_source_features)
+            personalize_message.set_llm_model(model)
 
         # if involve_recsys_result is not None:
         #     personalize_message.set_messaging_mode("recsys_result")

messaging_main_test.py

@@ -165,6 +165,7 @@ if __name__ == "__main__":
     permes = Permes()
 
     users_message = permes.create_personalize_messages(session=session,
+                                                model="gemma3:4b",
                                                 users=users,
                                                 brand=brand,
                                                 config_file=config_file,