fix for 46109 (Azure#46401)

annelo-msft · web-flow · commit 2d93e3d6de33 · 2024-10-02T14:43:55.000-07:00
diff --git a/sdk/openai/Azure.AI.OpenAI/src/Custom/AzureTokenAuthenticationPolicy.cs b/sdk/openai/Azure.AI.OpenAI/src/Custom/AzureTokenAuthenticationPolicy.cs
@@ -34,7 +34,7 @@ public override void Process(PipelineMessage message, IReadOnlyList<PipelinePoli
                 TokenRequestContext tokenRequestContext = CreateRequestContext(message.Request);
                 _currentToken = _credential.GetToken(tokenRequestContext, cancellationToken: default);
             }
-            message?.Request?.Headers?.Add("Authorization", $"Bearer {_currentToken.Value.Token}");
+            message?.Request?.Headers?.Set("Authorization", $"Bearer {_currentToken.Value.Token}");
         }
         ProcessNext(message, pipeline, currentIndex);
         if (message?.Response?.Status == (int)HttpStatusCode.Unauthorized)
@@ -53,7 +53,7 @@ public override async ValueTask ProcessAsync(PipelineMessage message, IReadOnlyL
                 _currentToken
                     = await _credential.GetTokenAsync(tokenRequestContext, cancellationToken: default).ConfigureAwait(false);
             }
-            message?.Request?.Headers?.Add("Authorization", $"Bearer {_currentToken.Value.Token}");
+            message?.Request?.Headers?.Set("Authorization", $"Bearer {_currentToken.Value.Token}");
         }
         await ProcessNextAsync(message, pipeline, currentIndex).ConfigureAwait(false);
         if (message?.Response?.Status == (int)HttpStatusCode.Unauthorized)

Original file line number	Diff line number	Diff line change
`@@ -34,7 +34,7 @@ public override void Process(PipelineMessage message, IReadOnlyList<PipelinePoli`
`34`	`34`	`TokenRequestContext tokenRequestContext = CreateRequestContext(message.Request);`
`35`	`35`	`_currentToken = _credential.GetToken(tokenRequestContext, cancellationToken: default);`
`36`	`36`	`}`
`37`		`- message?.Request?.Headers?.Add("Authorization", $"Bearer {_currentToken.Value.Token}");`
	`37`	`+ message?.Request?.Headers?.Set("Authorization", $"Bearer {_currentToken.Value.Token}");`
`38`	`38`	`}`
`39`	`39`	`ProcessNext(message, pipeline, currentIndex);`
`40`	`40`	`if (message?.Response?.Status == (int)HttpStatusCode.Unauthorized)`
`@@ -53,7 +53,7 @@ public override async ValueTask ProcessAsync(PipelineMessage message, IReadOnlyL`
`53`	`53`	`_currentToken`
`54`	`54`	`= await _credential.GetTokenAsync(tokenRequestContext, cancellationToken: default).ConfigureAwait(false);`
`55`	`55`	`}`
`56`		`- message?.Request?.Headers?.Add("Authorization", $"Bearer {_currentToken.Value.Token}");`
	`56`	`+ message?.Request?.Headers?.Set("Authorization", $"Bearer {_currentToken.Value.Token}");`
`57`	`57`	`}`
`58`	`58`	`await ProcessNextAsync(message, pipeline, currentIndex).ConfigureAwait(false);`
`59`	`59`	`if (message?.Response?.Status == (int)HttpStatusCode.Unauthorized)`