Continue with metadata

This commit is contained in:
Mykeehu 2025-06-04 11:56:39 +02:00 committed by GitHub
parent dddb887b60
commit 72264b5bac
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -699,6 +699,14 @@ def chatbot_wrapper(text, state, regenerate=False, _continue=False, loading_mess
output['internal'][-1] = [text, updated_internal] output['internal'][-1] = [text, updated_internal]
output['visible'][-1] = [visible_text, updated_visible] output['visible'][-1] = [visible_text, updated_visible]
row_idx = len(output['internal']) - 1
key = f"assistant_{row_idx}"
current_idx = output['metadata'][key]['current_version_index']
output['metadata'][key]['versions'][current_idx].update({
'content': output['internal'][row_idx][1],
'visible_content': output['visible'][row_idx][1]
})
if is_stream: if is_stream:
yield output yield output
elif not (j == 0 and visible_reply.strip() == ''): elif not (j == 0 and visible_reply.strip() == ''):