Define visible_text before applying chat_input extensions

This commit is contained in:
oobabooga 2023-07-26 07:26:37 -07:00
parent 517d40cffe
commit d94ba6e68b

View File

@ -175,10 +175,8 @@ def chatbot_wrapper(text, state, regenerate=False, _continue=False, loading_mess
# Preparing the input
if not any((regenerate, _continue)):
text, visible_text = apply_extensions('chat_input', text, visible_text, state)
if visible_text is None:
visible_text = text
text, visible_text = apply_extensions('chat_input', text, visible_text, state)
text = apply_extensions('input', text, state)
# *Is typing...*