diff --git a/src/utils/agents/streaming.py b/src/utils/agents/streaming.py index 37a094f1c..9d500aacf 100644 --- a/src/utils/agents/streaming.py +++ b/src/utils/agents/streaming.py @@ -516,6 +516,11 @@ def _( ) state.chunk_id += 1 state.turn_summary.next_chunk_id = state.chunk_id + + # Fix the missing refusal message issue in conversation/v2 + if not state.turn_summary.partial_tokens: + state.turn_summary.llm_response = final_text + return payload