diff --git a/dotnet/src/Microsoft.Agents.AI.Workflows.Declarative/AzureAgentProvider.cs b/dotnet/src/Microsoft.Agents.AI.Workflows.Declarative/AzureAgentProvider.cs index 8ad65c3523..ac44890c1c 100644 --- a/dotnet/src/Microsoft.Agents.AI.Workflows.Declarative/AzureAgentProvider.cs +++ b/dotnet/src/Microsoft.Agents.AI.Workflows.Declarative/AzureAgentProvider.cs @@ -48,21 +48,18 @@ public sealed class AzureAgentProvider(string projectEndpoint, TokenCredential p } /// - public override Task CreateMessageAsync(string conversationId, ChatMessage conversationMessage, CancellationToken cancellationToken = default) + public override async Task CreateMessageAsync(string conversationId, ChatMessage conversationMessage, CancellationToken cancellationToken = default) { - // TODO: Switch to asynchronous "CreateMessageAsync", when fix properly applied: - // BUG: https://github.com/Azure/azure-sdk-for-net/issues/52571 - // PR: https://github.com/Azure/azure-sdk-for-net/pull/52653 PersistentThreadMessage newMessage = - this.GetAgentsClient().Messages.CreateMessage( + await this.GetAgentsClient().Messages.CreateMessageAsync( conversationId, role: s_roleMap[conversationMessage.Role.Value.ToUpperInvariant()], contentBlocks: GetContent(), attachments: null, metadata: GetMetadata(), - cancellationToken); + cancellationToken).ConfigureAwait(false); - return Task.FromResult(ToChatMessage(newMessage)); + return ToChatMessage(newMessage); Dictionary? GetMetadata() {