RavenDB-23556 more renames

Lwiel · Jan 26, 2025 · 92ed83f · 92ed83f
1 parent a7d23b5
commit 92ed83f
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 25 deletions.
diff --git a/...ocuments/ETL/Providers/AI/AiExtensions.cs → ...L/Providers/AI/Extensions/AiExtensions.cs b/...ocuments/ETL/Providers/AI/AiExtensions.cs → ...L/Providers/AI/Extensions/AiExtensions.cs
@@ -1,9 +1,15 @@
 using System;
+using System.Diagnostics.CodeAnalysis;
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.SemanticKernel;
 using Microsoft.SemanticKernel.Connectors.Onnx;
+using Microsoft.SemanticKernel.Embeddings;
 using Raven.Client.Documents.Operations.ETL.AI;
+using Raven.Server.Documents.Indexes.VectorSearch;
+
 #pragma warning disable SKEXP0070
 
-namespace Raven.Server.Documents.ETL.Providers.AI;
+namespace Raven.Server.Documents.ETL.Providers.AI.Extensions;
 
 public static class AiExtensions
 {
@@ -37,4 +43,17 @@ public static EmbeddingPoolingMode ToEmbeddingPoolingMode(this OnnxEmbeddingPool
                 throw new ArgumentOutOfRangeException(nameof(poolingMode), poolingMode, null);
         }
     }
+
+    [Experimental("SKEXP0070")]
+    public static IKernelBuilder AddCustomBertOnnxTextEmbeddingGeneration(
+        this IKernelBuilder builder,
+        BertOnnxOptions options = null,
+        string serviceId = null)
+    {
+        builder.Services.AddKeyedSingleton<ITextEmbeddingGenerationService>(
+            serviceId,
+            GenerateEmbeddings.CreateTextEmbeddingGenerationService(options));
+
+        return builder;
+    }
 }
diff --git a/src/Raven.Server/Documents/ETL/Providers/AI/Extensions/KernelBuilderExtensions.cs b/src/Raven.Server/Documents/ETL/Providers/AI/Extensions/KernelBuilderExtensions.cs