ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-18 20:27:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

enhanced answer code

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-08-20 13:25:14 +02:00
parent e37aca6f63
commit cc7332e36f
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lollms/personality.py
View File

@ -3916,11 +3916,11 @@ class APScript(StateMachine):
return rounds_info
def answer(self, context_details, callback=None, send_full=True):
def answer(self, context_details, callback=None, send_full=True, custom_entries = ""):
if context_details["is_continue"]:
full_prompt = self.build_prompt_from_context_details(context_details, suppress= ["ai_prefix"])
full_prompt = self.build_prompt_from_context_details(context_details, custom_entries=custom_entries, suppress= ["ai_prefix"])
else:
full_prompt = self.build_prompt_from_context_details(context_details)
full_prompt = self.build_prompt_from_context_details(context_details, custom_entries=custom_entries)
out = self.fast_gen(full_prompt)
nb_tokens = len(self.personality.model.tokenize(out))