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

Save the chat history right after sending a message

Browse source
This commit is contained in:
oobabooga 2025-05-04 18:52:01 -07:00
parent 2da197bba4
commit d0211afb3c
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
modules/chat.py
View file

@ -483,6 +483,8 @@ def generate_chat_reply_wrapper(text, state, regenerate=False, _continue=False):
history = state['history']
for i, history in enumerate(generate_chat_reply(text, state, regenerate, _continue, loading_message=True, for_ui=True)):
yield chat_html_wrapper(history, state['name1'], state['name2'], state['mode'], state['chat_style'], state['character_menu']), history
if i == 0:
save_history(history, state['unique_id'], state['character_menu'], state['mode'])
save_history(history, state['unique_id'], state['character_menu'], state['mode'])