Update personality.py

This commit is contained in:
Saifeddine ALOUI 2024-05-29 13:46:20 +02:00 committed by GitHub
parent 8cfe6d73c4
commit d47da14c49
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1626,7 +1626,11 @@ class AIPersonality:
Returns:
bool: True if any antiprompt is found in the text (ignoring case), False otherwise.
"""
anti_prompts = [self.app.config.discussion_prompt_separator]
start_header_id_template = self.config.start_header_id_template
end_header_id_template = self.config.end_header_id_template
separator_template = self.config.separator_template
anti_prompts = [start_header_id_template, self.app.config.discussion_prompt_separator, separator_template, end_header_id_template]
if self.app.config.separator_template!="\n":
anti_prompts.append(self.app.config.separator_template)