Add support for SQL Server Vector Search

extensions/SQLServer/SQLServer/DependencyInjection.cs

@@ -14,10 +14,10 @@ namespace Microsoft.KernelMemory;
 public static partial class KernelMemoryBuilderExtensions
 {
  /// <summary>
- /// Kernel Memory Builder extension method to add SqlServer memory connector.
+ /// Kernel Memory Builder extension method to add SQL Server memory connector.
  /// </summary>
  /// <param name="builder">KM builder instance</param>
- /// <param name="config">SqlServer configuration</param>
+ /// <param name="config">SQL Server configuration</param>
  public static IKernelMemoryBuilder WithSqlServerMemoryDb(
  this IKernelMemoryBuilder builder,
  SqlServerConfig config)
@@ -27,15 +27,16 @@ public static IKernelMemoryBuilder WithSqlServerMemoryDb(
  }
 
  /// <summary>
- /// Kernel Memory Builder extension method to add SqlServer memory connector.
+ /// Kernel Memory Builder extension method to add SQL Server memory connector.
  /// </summary>
  /// <param name="builder">KM builder instance</param>
- /// <param name="connString">SqlServer connection string</param>
+ /// <param name="connString">SQL Server connection string</param>
+ /// <param name="useVectorSearch">A value that determines if we want to use Vector Search</param> 
  public static IKernelMemoryBuilder WithSqlServerMemoryDb(
  this IKernelMemoryBuilder builder,
- string connString)
+ string connString, bool useVectorSearch = false)
  {
- builder.Services.AddSqlServerAsMemoryDb(connString);
+ builder.Services.AddSqlServerAsMemoryDb(connString, useVectorSearch);
  return builder;
  }
 }
@@ -46,10 +47,10 @@ public static IKernelMemoryBuilder WithSqlServerMemoryDb(
 public static partial class DependencyInjection
 {
  /// <summary>
- /// Inject SqlServer as the default implementation of IMemoryDb
+ /// Inject SQL Server as the default implementation of IMemoryDb
  /// </summary>
  /// <param name="services">Service collection</param>
- /// <param name="config">Postgres configuration</param>
+ /// <param name="config">SQL Service configuration</param>
  public static IServiceCollection AddSqlServerAsMemoryDb(
  this IServiceCollection services,
  SqlServerConfig config)
@@ -60,15 +61,16 @@ public static IServiceCollection AddSqlServerAsMemoryDb(
  }
 
  /// <summary>
- /// Inject SqlServer as the default implementation of IMemoryDb
+ /// Inject SQL Server as the default implementation of IMemoryDb
  /// </summary>
  /// <param name="services">Service collection</param>
  /// <param name="connString">SQL Server connection string</param>
+ /// <param name="useVectorSearch">A value that determines if we want to use Vector Search</param>
  public static IServiceCollection AddSqlServerAsMemoryDb(
  this IServiceCollection services,
- string connString)
+ string connString, bool useVectorSearch = false)
  {
- var config = new SqlServerConfig { ConnectionString = connString };
+ var config = new SqlServerConfig { ConnectionString = connString, UseVectorSearch = useVectorSearch };
  return services.AddSqlServerAsMemoryDb(config);
  }
 }

extensions/SQLServer/SQLServer/SqlServerConfig.cs

@@ -61,4 +61,9 @@ public class SqlServerConfig
  /// The SQL Server tags table name.
  /// </summary>
  public string TagsTableName { get; set; } = DefaultTagsTableName;
+
+ /// <summary>
+ /// It tells if we should use the vector search or not.
+ /// </summary>
+ public bool UseVectorSearch { get; set; } = false;
 }