Apply input extensions after yielding *Is typing...*

This commit is contained in:
oobabooga 2023-06-24 11:07:11 -03:00
parent 3e80f2aceb
commit ec482f3dae

View File

@ -170,10 +170,11 @@ def chatbot_wrapper(text, history, state, regenerate=False, _continue=False, loa
if visible_text is None:
visible_text = text
text = apply_extensions('input', text)
# *Is typing...*
if loading_message:
yield {'visible': output['visible'] + [[visible_text, shared.processing_message]], 'internal': output['internal']}
text = apply_extensions('input', text)
else:
text, visible_text = output['internal'][-1][0], output['visible'][-1][0]
if regenerate: