Simplify method

microsoft · May 21, 2024 · 52a1e6b · 52a1e6b
1 parent 11d2926
commit 52a1e6b
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/service/Abstractions/KernelMemoryBuilderExtensions.cs b/service/Abstractions/KernelMemoryBuilderExtensions.cs
@@ -1,7 +1,6 @@
 // Copyright (c) Microsoft. All rights reserved.
 
 using System;
-using Microsoft.Extensions.DependencyInjection;
 using Microsoft.KernelMemory.AI;
 using Microsoft.KernelMemory.Configuration;
 using Microsoft.KernelMemory.DataFormats;
@@ -141,11 +140,7 @@ public static IKernelMemoryBuilder WithCustomEmbeddingGenerator(
     public static IKernelMemoryBuilder WithCustomEmbeddingGenerator<T>(
            this IKernelMemoryBuilder builder) where T : class, ITextEmbeddingGenerator
     {
-        builder.Services.AddSingleton<ITextEmbeddingGenerator>(serviceProvider =>
-        {
-            return ActivatorUtilities.CreateInstance<T>(serviceProvider);
-        });
-
+        builder.AddSingleton<ITextEmbeddingGenerator, T>();
         return builder;
     }