oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-03-08 22:53:51 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Allow extensions to skip output post-processing

Browse source
This commit is contained in:
oobabooga 2026-03-06 01:19:36 -03:00
parent 2e7e966ef2
commit 8a9afcbec6
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
modules/chat.py
View file

@ -923,7 +923,9 @@ def chatbot_wrapper(text, state, regenerate=False, _continue=False, loading_mess
visible_reply = html.escape(visible_reply)
if shared.stop_everything:
output['visible'][-1][1] = apply_extensions('output', output['visible'][-1][1], state, is_chat=True)
if not state.get('_skip_output_extensions'):
output['visible'][-1][1] = apply_extensions('output', output['visible'][-1][1], state, is_chat=True)
yield output
return
@ -956,9 +958,11 @@ def chatbot_wrapper(text, state, regenerate=False, _continue=False, loading_mess
full_visible = full_internal
full_visible = html.escape(full_visible)
output['visible'][-1][1] = apply_extensions('output', full_visible, state, is_chat=True)
if not state.get('_skip_output_extensions'):
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)
if not state.get('_skip_output_extensions'):
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: