BotSharp/src/Infrastructure/BotSharp.Core/BotSharpServiceCollectionExtensions.cs

76 lines
2.2 KiB
C#
Raw Normal View History

2023-06-03 02:07:30 +00:00
using BotSharp.Abstraction.Conversations;
2023-06-07 01:08:25 +00:00
using BotSharp.Abstraction.TextGeneratives;
2023-06-03 02:07:30 +00:00
using BotSharp.Core.Conversations;
2023-06-07 01:08:25 +00:00
using BotSharp.Core.Plugins.TextGeneratives.LLamaSharp;
2023-06-03 02:07:30 +00:00
using BotSharp.Core.Repository;
using EntityFrameworkCore.BootKit;
2023-05-27 01:58:31 +00:00
using Microsoft.AspNetCore.Builder;
2023-06-03 02:07:30 +00:00
using Microsoft.Extensions.Configuration;
2023-05-27 01:58:31 +00:00
using Microsoft.Extensions.DependencyInjection;
namespace BotSharp.Core;
public static class BotSharpServiceCollectionExtensions
{
2023-06-03 14:57:08 +00:00
public static IServiceCollection AddBotSharp(this IServiceCollection services, IConfiguration config)
2023-05-27 01:58:31 +00:00
{
2023-06-03 14:57:08 +00:00
services.AddSingleton<ISessionService, SessionService>();
2023-06-03 02:07:30 +00:00
services.AddSingleton<IConversationService, ConversationService>();
2023-06-03 14:57:08 +00:00
RegisterRepository(services, config);
2023-06-10 02:56:04 +00:00
RegisterPlugins(services, config);
2023-05-27 01:58:31 +00:00
return services;
}
public static void ConfigureBotSharp(this IServiceCollection services)
{
}
public static IApplicationBuilder UseBotSharp(this IApplicationBuilder app)
{
if (app == null)
{
throw new ArgumentNullException(nameof(app));
}
return app;
}
2023-06-03 02:07:30 +00:00
2023-06-03 14:57:08 +00:00
public static void RegisterRepository(IServiceCollection services, IConfiguration config)
2023-06-03 02:07:30 +00:00
{
var databaseSettings = new DatabaseSettings();
config.Bind("Database", databaseSettings);
services.AddSingleton((IServiceProvider x) =>
{
return databaseSettings;
});
var myDatabaseSettings = new MyDatabaseSettings();
config.Bind("Database", myDatabaseSettings);
services.AddSingleton((IServiceProvider x) =>
{
return databaseSettings;
});
services.AddScoped((IServiceProvider x) =>
{
return DataContextHelper.GetDbContext<MongoDbContext>(myDatabaseSettings, x);
});
2023-06-07 00:59:32 +00:00
services.AddSingleton(x =>
{
var settings = new LlamaSharpSettings();
config.Bind("LlamaSharp", settings);
return settings;
});
2023-06-07 01:08:25 +00:00
services.AddSingleton<IChatCompletionProvider, ChatCompletionProvider>();
2023-06-03 02:07:30 +00:00
}
2023-06-10 02:56:04 +00:00
public static void RegisterPlugins(IServiceCollection services, IConfiguration config)
{
}
2023-05-27 01:58:31 +00:00
}