Попытки в мультипарсинг
This commit is contained in:
@@ -1,9 +1,10 @@
|
||||
using CodeBase.Models;
|
||||
using CodeBase.Repositories;
|
||||
|
||||
namespace CodeBase.Services
|
||||
{
|
||||
public class CodeService(ChunkService service,
|
||||
GraphRepository repository,
|
||||
ScipRepository repository,
|
||||
VectorizationService vectorizationService,
|
||||
LlmService llmService)
|
||||
{
|
||||
@@ -18,7 +19,7 @@ namespace CodeBase.Services
|
||||
|
||||
var chunks = await vectorizationService.EnrichChunksWithVectorsAsync(allChunks);
|
||||
|
||||
await repository.SaveChunksAsync(name, chunks);
|
||||
//await repository.SaveChunksAsync(name, chunks);
|
||||
|
||||
return chunks;
|
||||
}
|
||||
@@ -35,12 +36,12 @@ namespace CodeBase.Services
|
||||
}
|
||||
|
||||
// Главный метод поиска
|
||||
public async Task<List<GraphNodeContext>> SearchAsync(
|
||||
public async Task<List<RetrievedContext>> SearchAsync(
|
||||
float[] queryVector,
|
||||
string name,
|
||||
int topK = 3) // Возвращаем топ-3 результата
|
||||
{
|
||||
var results = await repository.SearchAsync(name, queryVector, topK);
|
||||
var results = await repository.FindSimilarNodesAsync(queryVector, name, topK);
|
||||
|
||||
return results;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user