From 72264b5bacc600d6f49a991c0a8a48271ea66c85 Mon Sep 17 00:00:00 2001 From: Mykeehu Date: Wed, 4 Jun 2025 11:56:39 +0200 Subject: [PATCH] Continue with metadata --- modules/chat.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/modules/chat.py b/modules/chat.py index 2db72f36..408555b4 100644 --- a/modules/chat.py +++ b/modules/chat.py @@ -698,6 +698,14 @@ def chatbot_wrapper(text, state, regenerate=False, _continue=False, loading_mess output['internal'][-1] = [text, updated_internal] 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: yield output