agent/agent-framework
2
0
Fork 0
mirror of https://github.com/microsoft/agent-framework.git synced 2026-06-16 21:04:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: ChatMessage Aggregation in AIAgentHostExecutor (#530)

Browse Source 
The previous fix for the aggregation in AIAgentHostExecutor has a bug wherein we only include the first update from a ChatMessage in the outgoing "collected" message.

The fix is to ensure we put the collected updates into the message before sending it out.
This commit is contained in:
Jacob Alber
2025-08-28 14:23:19 -04:00
committed by GitHub
Unverified
parent fe1941d25f
commit ea7cff16ce
2 changed files with 217 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dotnet/src/Microsoft.Agents.Workflows/Specialized/AIAgentHostExecutor.cs
+6 -3
View File
@@ -94,7 +94,7 @@ internal class AIAgentHostExecutor : Executor
bool emitEvents = token.EmitEvents.HasValue ? token.EmitEvents.Value : this._emitEvents;
IAsyncEnumerable<AgentRunResponseUpdate> agentStream = this._agent.RunStreamingAsync(this._pendingMessages, this.EnsureThread(context));
List<AgentRunResponseUpdate> updates = new();
List<AIContent> updates = new();
ChatMessage? currentStreamingMessage = null;
await foreach (AgentRunResponseUpdate update in agentStream.ConfigureAwait(false))
@@ -109,8 +109,6 @@ internal class AIAgentHostExecutor : Executor
// providing some mechanisms to help the user complete the request, or route it out of the
// workflow.
updates.Add(update);
if (currentStreamingMessage == null || currentStreamingMessage.MessageId != update.MessageId)
{
await PublishCurrentMessageAsync().ConfigureAwait(false);
@@ -123,6 +121,8 @@ internal class AIAgentHostExecutor : Executor
AdditionalProperties = update.AdditionalProperties
};
}
updates.AddRange(update.Contents);
}
await PublishCurrentMessageAsync().ConfigureAwait(false);
@@ -132,6 +132,9 @@ internal class AIAgentHostExecutor : Executor
{
if (currentStreamingMessage != null)
{
currentStreamingMessage.Contents = updates;
updates = [];
await context.SendMessageAsync(currentStreamingMessage).ConfigureAwait(false);
}