Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-29 10:59:32 +01:00
Code Issues Packages Projects Releases Wiki Activity

Don't export include the example dialogue in the export json

Browse Source
This commit is contained in:
oobabooga 2023-01-21 02:55:13 -03:00
parent 990ee54ddd
commit 27e2d932b0
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
server.py
View File

@ -327,10 +327,17 @@ if args.chat or args.cai_chat:
return generate_chat_html(history, name1, name2, character) return generate_chat_html(history, name1, name2, character)
def save_history(): def save_history():
_history = copy.deepcopy(history)
for i in range(len(_history)):
if '<|BEGIN-VISIBLE-CHAT|>' in history[i][0]:
_history[i][0] = _history[i][0].replace('<|BEGIN-VISIBLE-CHAT|>', '')
_history = _history[i:]
break
if not Path('logs').exists(): if not Path('logs').exists():
Path('logs').mkdir() Path('logs').mkdir()
with open(Path('logs/conversation.json'), 'w') as f: with open(Path('logs/conversation.json'), 'w') as f:
f.write(json.dumps({'data': history})) f.write(json.dumps({'data': _history}))
return Path('logs/conversation.json') return Path('logs/conversation.json')
def load_history(file): def load_history(file):