Update modules/chat.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
This commit is contained in:
oobabooga 2025-10-09 21:01:13 -03:00 committed by GitHub
parent 748007f6ee
commit 0d03813e98
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1196,7 +1196,7 @@ def find_all_histories_with_first_prompts(state):
if re.match(r'^[0-9]{8}-[0-9]{2}-[0-9]{2}-[0-9]{2}$', filename):
first_prompt = ""
if data and 'visible' in data and len(data['visible']) > 0:
if len(data['internal']) > 1 and data['internal'][0][0] == '<|BEGIN-VISIBLE-CHAT|>':
if len(data['internal']) > 0 and data['internal'][0][0] == '<|BEGIN-VISIBLE-CHAT|>':
if len(data['visible']) > 1:
first_prompt = html.unescape(data['visible'][1][0])
elif i == 0: