30 lines
1,011 B
C#
30 lines
1,011 B
C#
|
using BotSharp.Abstraction.Plugins;
|
||
|
|
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 void RegisterDI(IServiceCollection services, IConfiguration config)
|
||
|
|
{
|
||
|
|
var settings = new HuggingFaceSettings();
|
||
|
|
config.Bind("HuggingFace", settings);
|
||
|
|
services.AddSingleton(x =>
|
||
|
|
{
|
||
|
|
Console.WriteLine($"Loaded HuggingFace settings: {settings.EndPoint} ({settings.Model}) {settings.Token.SubstringMax(4)}");
|
||
|
|
return settings;
|
||
|
|
});
|
||
|
|
|
||
|
|
services
|
||
|
|
.AddRefitClient<IInferenceApi>()
|
||
|
|
.AddHttpMessageHandler<AuthHeaderHandler>()
|
||
|
|
.ConfigureHttpClient(c => c.BaseAddress = new Uri(settings.EndPoint));
|
||
|
|
|
||
|
|
services.AddTransient<AuthHeaderHandler>();
|
||
|
|
services.AddScoped<IChatCompletion, ChatCompletionProvider>();
|
||
|
|
}
|
||
|
|
}
|