feat(db): add vector db

foxminchan · May 12, 2024 · e22b2d4 · e22b2d4
1 parent 7c114ad
commit e22b2d4
Show file tree

Hide file tree

Showing 8 changed files with 33 additions and 8 deletions.
diff --git a/Directory.Packages.props b/Directory.Packages.props
@@ -41,6 +41,12 @@
 		<PackageVersion Include="Npgsql.EntityFrameworkCore.PostgreSQL" Version="8.0.4" />
 		<PackageVersion Include="EFCore.NamingConventions" Version="8.0.3" />
 		<PackageVersion Include="EntityFrameworkCore.Exceptions.PostgreSQL" Version="8.1.2" />
+		<!-- Vector -->
+		<PackageVersion Include="Pgvector" Version="0.2.0" />
+		<PackageVersion Include="Pgvector.EntityFrameworkCore" Version="0.2.0" />
+		<!-- Semantic Kernel -->
+		<PackageVersion Include="Microsoft.SemanticKernel" Version="1.11.1" />
+		<PackageVersion Include="Microsoft.SemanticKernel.Connectors.OpenAI" Version="1.11.1" />
 		<!-- OpenTelemetry -->
 		<PackageVersion Include="OpenTelemetry.Extensions.Hosting" Version="1.8.1" />
 		<PackageVersion Include="OpenTelemetry.Exporter.OpenTelemetryProtocol" Version="1.8.1" />

diff --git a/docker-compose.yaml b/docker-compose.yaml
@@ -7,7 +7,7 @@ x-default-logging: &logging
 
 services:
   rookie.db:
-    image: postgres:16.2-alpine3.19
+    image: ankane/pgvector:v0.5.1
     container_name: rookie.db
     restart: unless-stopped
     ports:

diff --git a/src/RookieShop.Domain/RookieShop.Domain.csproj b/src/RookieShop.Domain/RookieShop.Domain.csproj
@@ -5,6 +5,11 @@
 		<PackageReference Include="StronglyTypedId" />
 	</ItemGroup>
 
+	<ItemGroup>
+		<PackageReference Include="Pgvector" />
+		<PackageReference Include="Pgvector.EntityFrameworkCore" />
+	</ItemGroup>
+
 	<ItemGroup>
 		<PackageReference Include="Ardalis.SmartEnum" />
 		<PackageReference Include="Ardalis.GuardClauses" />

diff --git a/src/RookieShop.Infrastructure/RookieShop.Infrastructure.csproj b/src/RookieShop.Infrastructure/RookieShop.Infrastructure.csproj
@@ -27,6 +27,11 @@
 		<PackageReference Include="FluentEmail.Mailtrap" />
 	</ItemGroup>
 
+	<ItemGroup>
+		<PackageReference Include="Microsoft.SemanticKernel" />
+		<PackageReference Include="Microsoft.SemanticKernel.Connectors.OpenAI" />
+	</ItemGroup>
+
 	<ItemGroup>
 		<PackageReference Include="AspNetCore.HealthChecks.Redis" />
 		<PackageReference Include="AspNetCore.HealthChecks.NpgSql" />

diff --git a/src/RookieShop.Persistence/ApplicationDbContext.cs b/src/RookieShop.Persistence/ApplicationDbContext.cs
@@ -31,6 +31,7 @@ protected override void OnModelCreating(ModelBuilder modelBuilder)
         base.OnModelCreating(modelBuilder);
         modelBuilder.ConfigureSmartEnum();
         modelBuilder.HasPostgresExtension(UniqueType.Extension);
+        modelBuilder.HasPostgresExtension(VectorType.Extension);
         modelBuilder.ApplyConfigurationsFromAssembly(AssemblyReference.DbContextAssembly);
     }
 }
diff --git a/src/RookieShop.Persistence/Constants/VectorType.cs b/src/RookieShop.Persistence/Constants/VectorType.cs
@@ -0,0 +1,7 @@
+namespace RookieShop.Persistence.Constants;
+
+public static class VectorType
+{
+    public const string Extension = "vector";
+    public const string DataType = "vector(384)";
+}
diff --git a/src/RookieShop.Persistence/Extension.cs b/src/RookieShop.Persistence/Extension.cs
@@ -5,6 +5,7 @@
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Hosting;
 using Microsoft.Extensions.Logging;
+using Npgsql;
 using RookieShop.Domain.SharedKernel;
 using RookieShop.Persistence.Interceptors;
 
@@ -24,11 +25,15 @@ public static IHostApplicationBuilder AddPersistence(this IHostApplicationBuilde
 
         builder.Services.AddDbContextPool<ApplicationDbContext>((sp, options) =>
         {
-            options.UseNpgsql(connectionString, sqlOptions =>
+            var npgsqlDataSourceBuilder = new NpgsqlDataSourceBuilder(connectionString);
+
+            npgsqlDataSourceBuilder.UseVector();
+
+            options.UseNpgsql(npgsqlDataSourceBuilder.ConnectionString, sqlOptions =>
                 {
+                    sqlOptions.UseVector();
                     sqlOptions.MigrationsAssembly(AssemblyReference.DbContextAssembly.FullName);
-                    sqlOptions.EnableRetryOnFailure(15, TimeSpan.FromSeconds(30),
-                        null);
+                    sqlOptions.EnableRetryOnFailure(15, TimeSpan.FromSeconds(30), null);
                 })
                 .UseExceptionProcessor()
                 .EnableServiceProviderCaching()

diff --git a/src/RookieShop.Persistence/RookieShop.Persistence.csproj b/src/RookieShop.Persistence/RookieShop.Persistence.csproj
@@ -29,8 +29,4 @@
 		</PackageReference>
 	</ItemGroup>
 
-	<ItemGroup>
-		<Folder Include="Constants\" />
-	</ItemGroup>
-
 </Project>