BotSharp/src/Plugins/BotSharp.Plugin.AzureOpenAI/AzureOpenAiPlugin.cs

35 lines
1.2 KiB
C#
Raw Normal View History

2023-06-17 02:42:35 +00:00
using BotSharp.Abstraction.MLTasks;
using BotSharp.Abstraction.Plugins;
2023-07-28 03:12:12 +00:00
using BotSharp.Abstraction.Utilities;
2023-10-16 20:07:07 +00:00
using BotSharp.Plugin.AzureOpenAI.Hooks;
2023-06-17 13:32:39 +00:00
using BotSharp.Plugin.AzureOpenAI.Providers;
2023-06-19 18:32:49 +00:00
using BotSharp.Plugin.AzureOpenAI.Settings;
2023-06-17 02:42:35 +00:00
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
2023-07-28 03:12:12 +00:00
using System;
2023-06-17 02:42:35 +00:00
namespace BotSharp.Platform.AzureAi;
2023-10-11 04:11:06 +00:00
/// <summary>
/// Azure OpenAI Service
/// </summary>
2023-06-17 02:42:35 +00:00
public class AzureOpenAiPlugin : IBotSharpPlugin
{
2023-10-12 11:36:36 +00:00
public string Name => "Azure OpenAI";
public string Description => "Azure OpenAI Service";
2023-06-17 02:42:35 +00:00
public void RegisterDI(IServiceCollection services, IConfiguration config)
{
var settings = new AzureOpenAiSettings();
config.Bind("AzureOpenAi", settings);
2023-07-28 03:12:12 +00:00
services.AddSingleton(x =>
{
Console.WriteLine($"Loaded AzureOpenAi settings: {settings.DeploymentModel} ({settings.Endpoint}) {settings.ApiKey.SubstringMax(4)}");
return settings;
});
2023-06-17 02:42:35 +00:00
services.AddScoped<ITextCompletion, TextCompletionProvider>();
2023-06-19 18:32:49 +00:00
services.AddScoped<IChatCompletion, ChatCompletionProvider>();
2023-10-16 20:07:07 +00:00
services.AddScoped<IContentGeneratingHook, TokenStatsConversationHook>();
2023-06-17 02:42:35 +00:00
}
}