35 lines
1.3 KiB
C#
35 lines
1.3 KiB
C#
using BotSharp.Abstraction.Plugins;
|
|
using BotSharp.Abstraction.Settings;
|
|
using BotSharp.Plugin.HuggingFace.Providers;
|
|
using BotSharp.Plugin.HuggingFace.Services;
|
|
using BotSharp.Plugin.HuggingFace.Settings;
|
|
using Refit;
|
|
|
|
namespace BotSharp.Plugin.HuggingFace;
|
|
|
|
public class HuggingFacePlugin : IBotSharpPlugin
|
|
{
|
|
public string Id => "4495c3d6-49bd-464e-92fb-c99049854cb7";
|
|
public string Name => "Hugging Face";
|
|
public string Description => "The Home of Machine Learning - Create, discover and collaborate on ML better.";
|
|
public string IconUrl => "https://upload.wikimedia.org/wikipedia/he/e/ee/Hugging_Face_logo.png";
|
|
public void RegisterDI(IServiceCollection services, IConfiguration config)
|
|
{
|
|
var settings = new HuggingFaceSettings();
|
|
config.Bind("HuggingFace", settings);
|
|
services.AddScoped(provider =>
|
|
{
|
|
var settingService = provider.GetRequiredService<ISettingService>();
|
|
return settingService.Bind<HuggingFaceSettings>("HuggingFace");
|
|
});
|
|
|
|
services
|
|
.AddRefitClient<IInferenceApi>()
|
|
.AddHttpMessageHandler<AuthHeaderHandler>()
|
|
.ConfigureHttpClient(c => c.BaseAddress = new Uri(settings.EndPoint));
|
|
|
|
services.AddTransient<AuthHeaderHandler>();
|
|
services.AddScoped<IChatCompletion, ChatCompletionProvider>();
|
|
}
|
|
}
|