переехали на вектурную бд

2026-07-22 22:36:12 +04:00
parent 17a64809cb
commit b6829ce753
5 changed files with 130 additions and 23 deletions
--- a/CodeBase/Repositories/QdrantRepository.cs
+++ b/CodeBase/Repositories/QdrantRepository.cs
@@ -0,0 +1,89 @@
+using CodeBase.Models;
+using Qdrant.Client;
+using Qdrant.Client.Grpc;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading.Tasks;
+
+public class QdrantRepository
+{
+    private readonly QdrantClient _client;
+    private const string CollectionName = "CodeBaseCollection";
+
+    public QdrantRepository(QdrantClient client)
+    {
+        _client = client;
+    }
+
+    // ==========================================
+    // 1. СОХРАНЕНИЕ ВЕКТОРОВ С ПРИВЯЗКОЙ К ПРОЕКТУ
+    // ==========================================
+    public async Task SaveChunksAsync(string projectName, List<CodeChunk> chunks)
+    {
+        var points = new List<PointStruct>();
+
+        foreach (var chunk in chunks)
+        {
+            if (chunk.Vector == null) continue;
+
+            // Формируем точку для БД
+            var point = new PointStruct
+            {
+                Id = Guid.NewGuid(), // Уникальный ID чанка
+                Vectors = chunk.Vector, // Тот самый массив float[]
+
+                // Вот здесь мы навешиваем все ярлыки!
+                Payload =
+                {
+                    ["ProjectName"] = projectName,
+                    ["FilePath"] = chunk.FilePath,
+                    ["MethodName"] = chunk.MethodName,
+                    ["Content"] = chunk.Content // Сам текст кода тоже храним прямо в БД
+                }
+            };
+            points.Add(point);
+        }
+
+        // Отправляем пачку векторов в базу
+        await _client.UpsertAsync(CollectionName, points);
+    }
+
+    // ==========================================
+    // 2. ПОИСК ТОЛЬКО В РАМКАХ ОДНОГО ПРОЕКТА
+    // ==========================================
+    public async Task<List<CodeChunk>> SearchAsync(string projectName, float[] queryVector, int topK = 15)
+    {
+        // Создаем фильтр: искать только там, где ProjectName совпадает с запрошенным
+        var projectFilter = new Filter
+        {
+            Must = {
+                new Condition
+                {
+                    Field = new FieldCondition
+                    {
+                        Key = "ProjectName",
+                        Match = new Match { Keyword = projectName }
+                    }
+                }
+            }
+        };
+
+        // База делает всю тяжелую математику за нас!
+        var searchResults = await _client.SearchAsync(
+            collectionName: CollectionName,
+            vector: queryVector,
+            limit: (ulong)topK,
+            filter: projectFilter, // Передаем наш фильтр
+            payloadSelector: true      // Просим БД вернуть Payload вместе с результатами
+        );
+
+        // Превращаем ответы базы обратно в наши C#-объекты CodeChunk
+        return searchResults.Select(hit => new CodeChunk
+        {
+            FilePath = hit.Payload["FilePath"].StringValue,
+            MethodName = hit.Payload["MethodName"].StringValue,
+            Content = hit.Payload["Content"].StringValue
+        }).ToList();
+    }
+}