diff --git a/Microsoft.Azure.Cosmos/src/ThinClientStoreClient.cs b/Microsoft.Azure.Cosmos/src/ThinClientStoreClient.cs index e6475325d7..1ddce45dd4 100644 --- a/Microsoft.Azure.Cosmos/src/ThinClientStoreClient.cs +++ b/Microsoft.Azure.Cosmos/src/ThinClientStoreClient.cs @@ -6,13 +6,13 @@ namespace Microsoft.Azure.Cosmos { using System; using System.Collections.Concurrent; - using System.Diagnostics.CodeAnalysis; using System.IO; using System.Net.Http; using System.Threading; using System.Threading.Tasks; using Microsoft.Azure.Cosmos.Core.Trace; using Microsoft.Azure.Cosmos.Routing; + using Microsoft.Azure.Cosmos.Tracing; using Microsoft.Azure.Documents; using Newtonsoft.Json; using static Microsoft.Azure.Cosmos.ThinClientTransportSerializer; @@ -91,6 +91,11 @@ private async ValueTask PrepareRequestForProxyAsync( BufferProviderWrapper bufferProviderWrapper = this.bufferProviderWrapperPool.Get(); try { + ContainerProperties collection = await clientCollectionCache.ResolveCollectionAsync( + request, + CancellationToken.None, + NoOpTrace.Singleton); + request.ResourceId = collection.ResourceId; requestMessage.Headers.TryAddWithoutValidation( ThinClientConstants.ProxyOperationType, request.OperationType.ToOperationTypeString());