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()
{