Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-20 10:35:10 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Minor bug fixes after e7e1f5901e

Browse source
This commit is contained in:
oobabooga 2024-06-25 11:49:33 -07:00
parent 83534798b2
commit 46ca15cb79
1 changed files with 8 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
modules/chat.py
View file

@ -539,17 +539,18 @@ def find_all_histories_with_first_prompts(state):
filename = path.stem
with open(path, 'r', encoding='utf-8') as f:
data = json.load(f)
first_prompt = ""
if 'visible' in data and len(data['visible']) > 0:
if data['internal'][0][0] == '<|BEGIN-VISIBLE-CHAT|>':
first_prompt = data['visible'][1][0] if len(data['visible']) > 1 else ''
else:
first_prompt = data['visible'][0][0]
first_prompt = html.unescape(first_prompt)
if len(data['visible']) > 1:
first_prompt = html.unescape(data['visible'][1][0])
elif i == 0:
first_prompt = "New chat"
else:
first_prompt = ''
first_prompt = html.unescape(data['visible'][0][0])
elif i == 0:
first_prompt = "New chat"
first_prompt = first_prompt.strip()
@ -598,7 +599,7 @@ def load_history_after_deletion(state, idx):
history = load_history(histories[idx][1], state['character_menu'], state['mode'])
else:
history = start_new_chat(state)
histories = find_all_histories_with_first_prompts
histories = find_all_histories_with_first_prompts(state)
return history, gr.update(choices=histories, value=histories[idx][1])