From cc7332e36f397bb8a4a142a1670a219902d5f946 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Tue, 20 Aug 2024 13:25:14 +0200 Subject: [PATCH] enhanced answer code --- lollms/personality.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lollms/personality.py b/lollms/personality.py index ec44409..fae442b 100644 --- a/lollms/personality.py +++ b/lollms/personality.py @@ -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))