From 1ca026f151521747a904bd84b1cb5e733b75faa1 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Tue, 7 Jan 2025 10:50:20 +0100 Subject: [PATCH] added user custom header --- lollms/personality.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/lollms/personality.py b/lollms/personality.py index 2393fdf..4a4839c 100644 --- a/lollms/personality.py +++ b/lollms/personality.py @@ -5345,15 +5345,19 @@ fetch('/open_file', { @property def ai_full_header(self) -> str: """Get the start_header_id_template.""" - return f"{self.start_user_header_id_template}{self.personality.name}{self.end_user_header_id_template}" + return f"{self.start_ai_header_id_template}{self.personality.name}{self.end_ai_header_id_template}" def system_custom_header(self, ai_name) -> str: + """Get the start_header_id_template.""" + return f"{self.start_header_id_template}{ai_name}{self.end_user_header_id_template}" + + def user_custom_header(self, ai_name) -> str: """Get the start_header_id_template.""" return f"{self.start_user_header_id_template}{ai_name}{self.end_user_header_id_template}" def ai_custom_header(self, ai_name) -> str: """Get the start_header_id_template.""" - return f"{self.start_user_header_id_template}{ai_name}{self.end_user_header_id_template}" + return f"{self.start_ai_header_id_template}{ai_name}{self.end_ai_header_id_template}" class AIPersonalityInstaller: