diff --git a/dotnet/src/Microsoft.Agents.AI.Workflows/Specialized/AIAgentHostExecutor.cs b/dotnet/src/Microsoft.Agents.AI.Workflows/Specialized/AIAgentHostExecutor.cs
index cd20fc4336..a49801cb37 100644
--- a/dotnet/src/Microsoft.Agents.AI.Workflows/Specialized/AIAgentHostExecutor.cs
+++ b/dotnet/src/Microsoft.Agents.AI.Workflows/Specialized/AIAgentHostExecutor.cs
@@ -70,6 +70,8 @@ internal class AIAgentHostExecutor : ChatProtocolExecutor
     protected override ProtocolBuilder ConfigureProtocol(ProtocolBuilder protocolBuilder)
     {
         return this.ConfigureUserInputHandling(base.ConfigureProtocol(protocolBuilder))
+                   .YieldsOutput<AgentResponseUpdate>()
+                   .YieldsOutput<AgentResponse>()
                    .ConfigureRoutes(routeBuilder => routeBuilder.AddHandler<ResetChatSignal>(this.ResetChat));
     }