ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-04-06 18:36:41 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

upgraded

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-04-21 00:56:11 +02:00
parent 2774830914
commit 10dacbbb47
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
lollms/databases/discussions_database.py
View File

@ -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