Fixed regeneration

This commit is contained in:
jeffser 2024-09-02 17:22:54 -06:00
parent c2c047d8b7
commit d3159ae6ea

View File

@ -334,7 +334,7 @@ class action_buttons(Gtk.Box):
del chat.messages[message_id] del chat.messages[message_id]
window.save_history(chat) window.save_history(chat)
if len(chat.messages) == 0: if len(chat.messages) == 0:
chat.show_welcome_screen(len(window.model_selector.get_model_list()) > 0) chat.show_welcome_screen(len(window.model_manager.get_model_list()) > 0)
def copy_message(self): def copy_message(self):
logger.debug("Copying message") logger.debug("Copying message")
@ -353,12 +353,12 @@ class action_buttons(Gtk.Box):
message_element.action_buttons = None message_element.action_buttons = None
history = window.convert_history_to_ollama(chat)[:list(chat.messages).index(message_element.message_id)] history = window.convert_history_to_ollama(chat)[:list(chat.messages).index(message_element.message_id)]
data = { data = {
"model": window.model_selector.get_model(), "model": window.model_manager.get_selected_model(),
"messages": history, "messages": history,
"options": {"temperature": window.model_tweaks["temperature"], "seed": window.model_tweaks["seed"]}, "options": {"temperature": window.ollama_instance.tweaks["temperature"], "seed": window.ollama_instance.tweaks["seed"]},
"keep_alive": f"{window.model_tweaks['keep_alive']}m" "keep_alive": f"{window.ollama_instance.tweaks['keep_alive']}m"
} }
thread = threading.Thread(target=window.run_message, args=(data, message_element)) thread = threading.Thread(target=window.run_message, args=(data, message_element, chat))
thread.start() thread.start()
else: else:
window.show_toast(_("Message cannot be regenerated while receiving a response"), window.main_overlay) window.show_toast(_("Message cannot be regenerated while receiving a response"), window.main_overlay)