From ec731210200ab53dc26bf56e6616d4533f64c165 Mon Sep 17 00:00:00 2001 From: Mykeehu Date: Tue, 10 Jun 2025 05:17:05 +0200 Subject: [PATCH] Fix continue/start reply with when using translation extensions (#6944) --------- Co-authored-by: oobabooga --- modules/chat.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/modules/chat.py b/modules/chat.py index 25a0607b..88ba33bb 100644 --- a/modules/chat.py +++ b/modules/chat.py @@ -760,7 +760,18 @@ def chatbot_wrapper(text, state, regenerate=False, _continue=False, loading_mess if is_stream: yield output - output['visible'][-1][1] = apply_extensions('output', output['visible'][-1][1], state, is_chat=True) + if _continue: + # Reprocess the entire internal text for extensions (like translation) + full_internal = output['internal'][-1][1] + if state['mode'] in ['chat', 'chat-instruct']: + full_visible = re.sub("(||{{user}})", state['name1'], full_internal) + else: + full_visible = full_internal + + full_visible = html.escape(full_visible) + output['visible'][-1][1] = apply_extensions('output', full_visible, state, is_chat=True) + else: + output['visible'][-1][1] = apply_extensions('output', output['visible'][-1][1], state, is_chat=True) # Final sync for version metadata (in case streaming was disabled) if regenerate: