Fix mismatched output in web server (lm-sys#3097)

itsharex · Feb 27, 2024 · d04ce64 · d04ce64
1 parent 4edbd9d
commit d04ce64
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fastchat/serve/gradio_web_server.py b/fastchat/serve/gradio_web_server.py
@@ -279,7 +279,7 @@ def add_text(state, model_selector, text, image, request: gr.Request):
 
     if len(text) <= 0:
         state.skip_next = True
-        return (state, state.to_gradio_chatbot(), "") + (no_change_btn,) * 5
+        return (state, state.to_gradio_chatbot(), "", None) + (no_change_btn,) * 5
 
     flagged = moderation_filter(text, [state.model_name])
     if flagged:
@@ -290,7 +290,7 @@ def add_text(state, model_selector, text, image, request: gr.Request):
     if (len(state.conv.messages) - state.conv.offset) // 2 >= CONVERSATION_TURN_LIMIT:
         logger.info(f"conversation turn limit. ip: {ip}. text: {text}")
         state.skip_next = True
-        return (state, state.to_gradio_chatbot(), CONVERSATION_LIMIT_MSG) + (
+        return (state, state.to_gradio_chatbot(), CONVERSATION_LIMIT_MSG, None) + (
             no_change_btn,
         ) * 5