mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix clear history

Browse source
This commit is contained in:
oobabooga 2023-02-26 23:58:04 -03:00
parent 7a776ccf87
commit f24b6e78a3
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
modules/chat.py
View file

@ -216,11 +216,16 @@ def clear_html():
def clear_chat_log(name1, name2):
if shared.character != 'None':
found = False
for i in range(len(shared.history['internal'])):
if '<|BEGIN-VISIBLE-CHAT|>' in shared.history['internal'][i][0]:
shared.history['visible'] = [['', apply_extensions(shared.history['internal'][i][1], "output")]]
shared.history['internal'] = shared.history['internal'][:i+1]
found = True
break
if not found:
shared.history['visible'] = []
shared.history['internal'] = []
else:
shared.history['internal'] = []
shared.history['visible'] = []