diff --git a/packages/agent/src/agent.ts b/packages/agent/src/agent.ts
index b87c582d..dec35373 100644
--- a/packages/agent/src/agent.ts
+++ b/packages/agent/src/agent.ts
@@ -306,6 +306,7 @@ export class Agent {
 						break;
 
 					case "agent_end":
+						this._state.isStreaming = false;
 						this._state.streamMessage = null;
 						break;
 				}