From 769eee1ff30d922cb15b89c9840d8147c077231f Mon Sep 17 00:00:00 2001 From: Kelvie Wong Date: Tue, 18 Feb 2025 07:01:30 -0800 Subject: [PATCH] Fix OpenAI API with new param (show_after), closes #6747 (#6749) --------- Co-authored-by: oobabooga --- modules/chat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/chat.py b/modules/chat.py index 2852aaf3..66ab8c74 100644 --- a/modules/chat.py +++ b/modules/chat.py @@ -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...*"