Fix OpenAI API with new param (show_after), closes #6747 (#6749)

---------

Co-authored-by: oobabooga <oobabooga4@gmail.com>
This commit is contained in:
Kelvie Wong 2025-02-18 07:01:30 -08:00 committed by oobabooga
parent 7c883ef2f0
commit 769eee1ff3

View file

@ -412,7 +412,7 @@ def generate_chat_reply(text, state, regenerate=False, _continue=False, loading_
yield history
return
show_after = html.escape(state["show_after"]) if state["show_after"] else None
show_after = html.escape(state.get("show_after")) if state.get("show_after") else None
for history in chatbot_wrapper(text, state, regenerate=regenerate, _continue=_continue, loading_message=loading_message, for_ui=for_ui):
if show_after:
after = history["visible"][-1][1].partition(show_after)[2] or "*Is thinking...*"