simcop2387/text-generation-webui-mirror
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2025-06-07 14:17:09 -04:00
Code Issues Projects Releases Packages Wiki Activity

Avoid duplicates when adding message version

Browse source 
Moves current index to existing version if already exists
This commit is contained in:
Th-Underscore 2025-05-27 07:30:07 -04:00
parent fac0ac4418
commit 11424f7a8d
No known key found for this signature in database
GPG key ID: 8D0551EF8593B2F0
1 changed files with 14 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
modules/chat.py
View file

@ -414,16 +414,23 @@ def add_message_version(history, row_idx, is_current=True):
if "versions" not in history['metadata'][key]: if "versions" not in history['metadata'][key]:
history['metadata'][key]["versions"] = [] history['metadata'][key]["versions"] = []
# Add current message as a version index = None
history['metadata'][key]["versions"].append({ for index, version in enumerate(history['metadata'][key]["versions"]):
"content": history['internal'][row_idx][1], print(index, version['content'] == history['internal'][row_idx][1], version['visible_content'] == history['visible'][row_idx][1])
"visible_content": history['visible'][row_idx][1], if version['content'] == history['internal'][row_idx][1] and version['visible_content'] == history['visible'][row_idx][1]:
"timestamp": get_current_timestamp() break
})
if index is None:
# Add current message as a version
history['metadata'][key]["versions"].append({
"content": history['internal'][row_idx][1],
"visible_content": history['visible'][row_idx][1],
"timestamp": get_current_timestamp()
})
# Update index if this is the current version # Update index if this is the current version
if is_current: if is_current:
history['metadata'][key]["current_version_index"] = len(history['metadata'][key]["versions"]) - 1 history['metadata'][key]["current_version_index"] = index if index is not None else len(history['metadata'][key]["versions"]) - 1
def add_message_attachment(history, row_idx, file_path, is_user=True): def add_message_attachment(history, row_idx, file_path, is_user=True):