Попытки в мультипарсинг
This commit is contained in:
45
CodeBase/Repositories/DatabaseInitializer.cs
Normal file
45
CodeBase/Repositories/DatabaseInitializer.cs
Normal file
@@ -0,0 +1,45 @@
|
||||
using Neo4j.Driver;
|
||||
|
||||
namespace CodeBase.Repositories
|
||||
{
|
||||
public class DatabaseInitializer
|
||||
{
|
||||
private readonly IDriver _neo4jDriver;
|
||||
|
||||
public DatabaseInitializer(IDriver neo4jDriver)
|
||||
{
|
||||
_neo4jDriver = neo4jDriver;
|
||||
}
|
||||
|
||||
public async Task InitializeAsync()
|
||||
{
|
||||
try
|
||||
{
|
||||
await using var session = _neo4jDriver.AsyncSession();
|
||||
|
||||
// IF NOT EXISTS гарантирует, что запрос не упадет с ошибкой,
|
||||
// если индекс уже был создан при предыдущем запуске
|
||||
var query = @"
|
||||
CREATE VECTOR INDEX code_embeddings IF NOT EXISTS
|
||||
FOR (m:CodeEntity) ON (m.embedding)
|
||||
OPTIONS {
|
||||
indexConfig: {
|
||||
`vector.dimensions`: 768, // <-- УКАЖИ ТУТ РАЗМЕРНОСТЬ ТВОЕЙ МОДЕЛИ
|
||||
`vector.similarity_function`: 'cosine'
|
||||
}
|
||||
}";
|
||||
|
||||
await session.ExecuteWriteAsync(async tx =>
|
||||
{
|
||||
await tx.RunAsync(query);
|
||||
});
|
||||
|
||||
Console.WriteLine("[БД] Векторный индекс успешно инициализирован.");
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
Console.WriteLine($"[БД ОШИБКА] Ошибка при создании индекса: {ex.Message}");
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user