From 352cdb769183dd900f003d9b4489e89997ffb8d6 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Sat, 18 May 2024 23:09:07 +0200 Subject: [PATCH] bugfix --- lollms/media.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lollms/media.py b/lollms/media.py index 12441c4..00d7901 100644 --- a/lollms/media.py +++ b/lollms/media.py @@ -364,7 +364,7 @@ class AudioRecorder: if result["text"]!="": # TODO : send the output # self.transcription_signal.new_user_transcription.emit(filename, result["text"]) - self.discussion.add_message(MSG_TYPE.MSG_TYPE_FULL, SENDER_TYPES.SENDER_TYPES_USER, "user",result["text"]) + self.discussion.add_message(MSG_TYPE.MSG_TYPE_FULL.value, SENDER_TYPES.SENDER_TYPES_USER.value, "user",result["text"]) discussion = self.discussion.format_discussion(self.context_size) full_context = '!@>system:' + self.personality.personality_conditioning +"\n" + discussion+"\n!@>lollms:" ASCIIColors.red(" ---------------- Discussion ---------------------") @@ -381,7 +381,7 @@ class AudioRecorder: else: lollms_text = self.lc.generate(full_context+"!@>lollms: "+ lollms_text + "\n!@>functions outputs:\n"+ "\n".join(responses) +"!@>lollms:") lollms_text = self.fix_string_for_xtts(lollms_text) - self.discussion.add_message(MSG_TYPE.MSG_TYPE_FULL, SENDER_TYPES.SENDER_TYPES_AI, "lollms",lollms_text) + self.discussion.add_message(MSG_TYPE.MSG_TYPE_FULL.value, SENDER_TYPES.SENDER_TYPES_AI.value, "lollms",lollms_text) ASCIIColors.red(" -------------- LOLLMS answer -------------------") ASCIIColors.yellow(lollms_text) ASCIIColors.red(" -------------------------------------------------")