54 lines
1.3 KiB
C#
54 lines
1.3 KiB
C#
using CodeBase.Services;
|
|
using Microsoft.Build.Locator;
|
|
using Neo4j.Driver;
|
|
|
|
if (!MSBuildLocator.IsRegistered)
|
|
{
|
|
MSBuildLocator.RegisterDefaults();
|
|
}
|
|
|
|
var builder = WebApplication.CreateBuilder(args);
|
|
|
|
// Add services to the container.
|
|
builder.Services.AddSingleton<IDriver>(sp =>
|
|
GraphDatabase.Driver(
|
|
"bolt://localhost:7687",
|
|
AuthTokens.Basic("neo4j", "password123")
|
|
)
|
|
);
|
|
|
|
builder.Services.AddTransient<VectorizationService>();
|
|
builder.Services.AddTransient<LlmService>();
|
|
builder.Services.AddScoped<LlmPromptBuilder>();
|
|
builder.Services.AddTransient<ChunkService>();
|
|
builder.Services.AddTransient<CodeService>();
|
|
builder.Services.AddTransient<GraphRepository>();
|
|
|
|
builder.Services.AddControllers();
|
|
builder.Services.AddSwaggerGen();
|
|
|
|
var app = builder.Build();
|
|
|
|
// Configure the HTTP request pipeline.
|
|
if (app.Environment.IsDevelopment())
|
|
{
|
|
app.UseSwagger();
|
|
app.UseSwaggerUI();
|
|
}
|
|
|
|
app.UseHttpsRedirection();
|
|
|
|
app.UseAuthorization();
|
|
|
|
app.MapControllers();
|
|
|
|
using (var scope = app.Services.CreateScope())
|
|
{
|
|
var graphRepo = scope.ServiceProvider.GetRequiredService<GraphRepository>();
|
|
Console.WriteLine("Проверяем и создаем векторный индекс в Neo4j...");
|
|
await graphRepo.InitializeDbAsync();
|
|
Console.WriteLine("Индекс готов!");
|
|
}
|
|
|
|
app.Run();
|