ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-03-01 04:06:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

added user custom header

Browse Source
This commit is contained in:
Saifeddine ALOUI 2025-01-07 10:50:20 +01:00
parent 7859aa748f
commit 1ca026f151
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lollms/personality.py
View File

@ -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: