From 25560934a30bbde2854a871a9fd3e32956073f9e Mon Sep 17 00:00:00 2001
From: Saifeddine ALOUI <aloui.seifeddine@gmail.com>
Date: Thu, 6 Feb 2025 15:51:27 +0100
Subject: [PATCH] Update lollms_webui.py
---
lollms_webui.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/lollms_webui.py b/lollms_webui.py
index 958c7bbf..cba8735f 100644
--- a/lollms_webui.py
+++ b/lollms_webui.py
@@ -621,7 +621,7 @@ class LOLLMSWebUI(LOLLMSElfServer):
# Get the list of messages
messages = discussion.get_messages()
- discussion_messages = f"{self.system_full_header}Create a short title to this discussion\nYour response should only contain the title without any comments or thoughts.\n"
+ discussion_messages = f"{self.system_full_header}Create a short title to this discussion\n--- discussion ---\n"
discussion_title = f"\n{self.ai_custom_header('assistant')}"
available_space = (
@@ -667,6 +667,8 @@ class LOLLMSWebUI(LOLLMSElfServer):
for message_tokens in full_message_list:
discussion_messages += self.model.detokenize(message_tokens)
+ discussion_messages +"\n--- ---\n"
+ discussion_messages +f"\n{self.user_custom_header('user')}Your response should only contain the title without any comments or thoughts.\n"
discussion_messages += discussion_title
title = [""]
@@ -681,7 +683,7 @@ class LOLLMSWebUI(LOLLMSElfServer):
else:
return True
- self._generate(discussion_messages, 512, client_id, receive)
+ self._generate(discussion_messages, 1024, client_id, receive)
title[0] = self.personality.remove_thinking_blocks(title[0])
ASCIIColors.info(f"TITLE:{title[0]}")
return title[0]