GithubBackup/Foxel
1
0
Fork 0
mirror of https://github.com/DrizzleTime/Foxel.git synced 2026-09-04 23:29:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(VectorDb): simplify QdrantVectorDbService class

Browse Source
This commit is contained in:
ShiYu
2025-05-31 23:07:45 +08:00
parent 6433781f07
commit 70efa3a9b2
1 changed files with 7 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Services/VectorDb/QdrantVectorDbService.cs
+7 -16
View File
@@ -7,33 +7,24 @@ using Qdrant.Client;
namespace Foxel.Services.VectorDB; namespace Foxel.Services.VectorDB;
public class QdrantVectorDbService : IVectorDbService public class QdrantVectorDbService(IDbContextFactory<MyDbContext> contextFactory, IConfigService configService)
: IVectorDbService
{ {
private readonly IDbContextFactory<MyDbContext> _contextFactory;
private readonly IConfigService _configService;
private VectorStore? _vectorStore; private VectorStore? _vectorStore;
private string? _currentHost; private string? _currentHost;
private string? _currentApiKey; private string? _currentApiKey;
public QdrantVectorDbService(IDbContextFactory<MyDbContext> contextFactory, IConfigService configService)
{
_contextFactory = contextFactory;
_configService = configService;
}
private VectorStore GetVectorStore() private VectorStore GetVectorStore()
{ {
string host = _configService["VectorDb:QdrantHost"] ?? string host = configService["VectorDb:QdrantHost"];
"b63da3b8-c126-4546-95ab-176f907fb1ef.eu-central-1-0.aws.cloud.qdrant.io";
string apiKey = configService["VectorDb:QdrantApiKey"];
string apiKey = _configService["VectorDb:QdrantApiKey"] ??
"eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhY2Nlc3MiOiJtIn0.QzQN4cyo5mldCi9ohe0Aqap4fpTMuSEMGkXtkgBTNQI";
if (_vectorStore == null || _currentHost != host || _currentApiKey != apiKey) if (_vectorStore == null || _currentHost != host || _currentApiKey != apiKey)
{ {
var qdrantClient = new QdrantClient(host, https: true, apiKey: apiKey); var qdrantClient = new QdrantClient(host, https: true, apiKey: apiKey);
_vectorStore = new QdrantVectorStore(qdrantClient, true); _vectorStore = new QdrantVectorStore(qdrantClient, true);
_currentHost = host; _currentHost = host;
_currentApiKey = apiKey; _currentApiKey = apiKey;
} }
@@ -43,7 +34,7 @@ public class QdrantVectorDbService : IVectorDbService
public async Task BuildUserPictureVectorsAsync() public async Task BuildUserPictureVectorsAsync()
{ {
await using var dbContext = await _contextFactory.CreateDbContextAsync(); await using var dbContext = await contextFactory.CreateDbContextAsync();
var userPictures = dbContext.Pictures var userPictures = dbContext.Pictures
.Where(p => p.UserId != null && p.Embedding != null) .Where(p => p.UserId != null && p.Embedding != null)
.Select(p => new { p.Id, p.Name, p.Embedding, p.UserId }) .Select(p => new { p.Id, p.Name, p.Embedding, p.UserId })