chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,81 @@
|
||||
// Copyright (c) Microsoft. All rights reserved.
|
||||
|
||||
using System.ClientModel;
|
||||
using System.ClientModel.Primitives;
|
||||
using Azure.AI.OpenAI;
|
||||
using Microsoft.SemanticKernel;
|
||||
|
||||
#pragma warning disable CA5399 // HttpClient is created without enabling CheckCertificateRevocationList
|
||||
|
||||
namespace ChatCompletion;
|
||||
|
||||
/// <summary>
|
||||
/// This example shows a way of using a Custom HttpClient and HttpHandler with Azure OpenAI Connector to capture
|
||||
/// the request Uri and Headers for each request.
|
||||
/// </summary>
|
||||
public sealed class AzureOpenAI_CustomClient(ITestOutputHelper output) : BaseTest(output)
|
||||
{
|
||||
[Fact]
|
||||
public async Task UsingCustomHttpClientWithAzureOpenAI()
|
||||
{
|
||||
Assert.NotNull(TestConfiguration.AzureOpenAI.Endpoint);
|
||||
Assert.NotNull(TestConfiguration.AzureOpenAI.ChatDeploymentName);
|
||||
Assert.NotNull(TestConfiguration.AzureOpenAI.ApiKey);
|
||||
|
||||
Console.WriteLine($"======== Azure Open AI - {nameof(UsingCustomHttpClientWithAzureOpenAI)} ========");
|
||||
|
||||
// Create an HttpClient and include your custom header(s)
|
||||
using var myCustomHttpHandler = new MyCustomClientHttpHandler(Output);
|
||||
using var myCustomClient = new HttpClient(handler: myCustomHttpHandler);
|
||||
myCustomClient.DefaultRequestHeaders.Add("My-Custom-Header", "My Custom Value");
|
||||
|
||||
// Configure AzureOpenAIClient to use the customized HttpClient
|
||||
var clientOptions = new AzureOpenAIClientOptions
|
||||
{
|
||||
Transport = new HttpClientPipelineTransport(myCustomClient),
|
||||
NetworkTimeout = TimeSpan.FromSeconds(30),
|
||||
RetryPolicy = new ClientRetryPolicy()
|
||||
};
|
||||
|
||||
var customClient = new AzureOpenAIClient(new Uri(TestConfiguration.AzureOpenAI.Endpoint), new ApiKeyCredential(TestConfiguration.AzureOpenAI.ApiKey), clientOptions);
|
||||
|
||||
var kernel = Kernel.CreateBuilder()
|
||||
.AddAzureOpenAIChatCompletion(TestConfiguration.AzureOpenAI.ChatDeploymentName, customClient)
|
||||
.Build();
|
||||
|
||||
// Load semantic plugin defined with prompt templates
|
||||
string folder = RepoFiles.SamplePluginsPath();
|
||||
|
||||
kernel.ImportPluginFromPromptDirectory(Path.Combine(folder, "FunPlugin"));
|
||||
|
||||
// Run
|
||||
var result = await kernel.InvokeAsync(
|
||||
kernel.Plugins["FunPlugin"]["Excuses"],
|
||||
new() { ["input"] = "I have no homework" }
|
||||
);
|
||||
Console.WriteLine(result.GetValue<string>());
|
||||
|
||||
myCustomClient.Dispose();
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Normally you would use a custom HttpClientHandler to add custom logic to your custom http client
|
||||
/// This uses the ITestOutputHelper to write the requested URI to the test output
|
||||
/// </summary>
|
||||
/// <param name="output">The <see cref="ITestOutputHelper"/> to write the requested URI to the test output </param>
|
||||
private sealed class MyCustomClientHttpHandler(ITestOutputHelper output) : HttpClientHandler
|
||||
{
|
||||
protected override async Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)
|
||||
{
|
||||
output.WriteLine($"Requested URI: {request.RequestUri}");
|
||||
|
||||
request.Headers.Where(h => h.Key != "Authorization")
|
||||
.ToList()
|
||||
.ForEach(h => output.WriteLine($"{h.Key}: {string.Join(", ", h.Value)}"));
|
||||
output.WriteLine("--------------------------------");
|
||||
|
||||
// Add custom logic here
|
||||
return await base.SendAsync(request, cancellationToken);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user