增加stop逻辑

2026-03-22 13:07:17 +08:00 · 2025-03-14 14:06:14 +08:00
parent a7e3a122eb
commit b16394cabb
1 changed files with 2 additions and 0 deletions
--- a/gradio_ui/app.py
+++ b/gradio_ui/app.py
@@ -115,6 +115,8 @@ def process_input(user_input, state, vision_agent_state):
        messages=state["messages"],
        vision_agent = agent
    ):
+        if state["stop"]:
+            return
        state['chatbox_messages'] = []
        for message in state['messages']:
            # convert message["content"] to gradio chatbox format