From 10dacbbb47665b1c8179d3800be3918a2e80710d Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Sun, 21 Apr 2024 00:56:11 +0200 Subject: [PATCH] upgraded --- lollms/databases/discussions_database.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/lollms/databases/discussions_database.py b/lollms/databases/discussions_database.py index 71b8746..661442e 100644 --- a/lollms/databases/discussions_database.py +++ b/lollms/databases/discussions_database.py @@ -576,7 +576,7 @@ class Message: (self.sender, self.content, self.metadata, self.ui, self.message_type, self.rank, self.parent_message_id, self.binding, self.model, self.personality, self.created_at, self.started_generating_at, self.finished_generating_at, nb_tokens, self.discussion_id) ) - def update(self, new_content, new_metadata=None, new_ui=None, nb_tokens=None, commit=True): + def update(self, new_content, new_metadata=None, new_ui=None, started_generating_at=None, nb_tokens=None, commit=True): self.finished_generating_at = datetime.now().strftime('%Y-%m-%d %H:%M:%S') text = f"UPDATE message SET content = ?" params = [new_content] @@ -587,6 +587,10 @@ class Message: text+=", ui = ?" params.append(new_ui) + if started_generating_at is not None: + text+=", started_generating_at = ?" + params.append(started_generating_at) + if nb_tokens is not None: text+=", nb_tokens = ?" params.append(nb_tokens) @@ -776,14 +780,14 @@ class Discussion: else: return False - def update_message(self, new_content, new_metadata=None, new_ui=None, nb_tokens=None): + def update_message(self, new_content, new_metadata=None, new_ui=None, started_generating_at=None, nb_tokens=None): """Updates the content of a message Args: message_id (int): The id of the message to be changed new_content (str): The nex message content """ - self.current_message.update(new_content, new_metadata, new_ui, nb_tokens) + self.current_message.update(new_content, new_metadata, new_ui, started_generating_at, nb_tokens) def edit_message(self, message_id, new_content, new_metadata=None, new_ui=None): """Edits the content of a message