.Net: GoogleVertexAI - memory builder

dotnet/samples/KernelSyntaxExamples/Example14_SemanticMemory.cs

@@ -61,6 +61,12 @@ public async Task RunAsync()
             .WithMemoryStore(new VolatileMemoryStore())
             .Build();
 
+        // Uncomment the following line to use GoogleAI embeddings
+        // var memoryWithCustomDb = new MemoryBuilder()
+        //     .WithGoogleAITextEmbeddingGeneration(TestConfiguration.GoogleAI.EmbeddingModelId, TestConfiguration.GoogleAI.ApiKey)
+        //     .WithMemoryStore(new VolatileMemoryStore())
+        //     .Build();
+
         await RunExampleAsync(memoryWithCustomDb);
     }
 
@@ -167,7 +173,5 @@ private static Dictionary<string, string> SampleData()
         };
     }
 
-    public Example14_SemanticMemory(ITestOutputHelper output) : base(output)
-    {
-    }
+    public Example14_SemanticMemory(ITestOutputHelper output) : base(output) { }
 }

dotnet/src/Connectors/Connectors.GoogleVertexAI.UnitTests/Extensions/GoogleAIMemoryBuilderExtensionsTests.cs

@@ -0,0 +1,32 @@
+// Copyright (c) Microsoft. All rights reserved.
+
+using Microsoft.SemanticKernel;
+using Microsoft.SemanticKernel.Memory;
+using Moq;
+using Xunit;
+
+namespace SemanticKernel.Connectors.GoogleVertexAI.UnitTests.Extensions;
+
+/// <summary>
+/// Unit tests for <see cref="GoogleAIMemoryBuilderExtensions"/> class.
+/// </summary>
+public sealed class GoogleAIMemoryBuilderExtensionsTests
+{
+    private readonly Mock<IMemoryStore> _mockMemoryStore = new();
+
+    [Fact]
+    public void ShouldBuildMemoryWithGoogleAIEmbeddingGenerator()
+    {
+        // Arrange
+        var builder = new MemoryBuilder();
+
+        // Act
+        var memory = builder
+            .WithGoogleAITextEmbeddingGeneration("fake-model", "fake-apikey")
+            .WithMemoryStore(this._mockMemoryStore.Object)
+            .Build();
+
+        // Assert
+        Assert.NotNull(memory);
+    }
+}

dotnet/src/Connectors/Connectors.GoogleVertexAI.UnitTests/Extensions/VertexAIMemoryBuilderExtensionsTests.cs

@@ -0,0 +1,32 @@
+// Copyright (c) Microsoft. All rights reserved.
+
+using Microsoft.SemanticKernel;
+using Microsoft.SemanticKernel.Memory;
+using Moq;
+using Xunit;
+
+namespace SemanticKernel.Connectors.GoogleVertexAI.UnitTests.Extensions;
+
+/// <summary>
+/// Unit tests for <see cref="VertexAIMemoryBuilderExtensions"/> class.
+/// </summary>
+public sealed class VertexAIMemoryBuilderExtensionsTests
+{
+    private readonly Mock<IMemoryStore> _mockMemoryStore = new();
+
+    [Fact]
+    public void ShouldBuildMemoryWithVertexAIEmbeddingGenerator()
+    {
+        // Arrange
+        var builder = new MemoryBuilder();
+
+        // Act
+        var memory = builder
+            .WithVertexAITextEmbeddingGeneration("fake-model", "fake-bearer-key", "fake-location", "fake-project")
+            .WithMemoryStore(this._mockMemoryStore.Object)
+            .Build();
+
+        // Assert
+        Assert.NotNull(memory);
+    }
+}

dotnet/src/Connectors/Connectors.GoogleVertexAI/Extensions/GoogleAIMemoryBuilderExtensions.cs

@@ -0,0 +1,40 @@
+// Copyright (c) Microsoft. All rights reserved.
+
+using System.Net.Http;
+using Microsoft.SemanticKernel.Connectors.GoogleVertexAI;
+using Microsoft.SemanticKernel.Http;
+using Microsoft.SemanticKernel.Memory;
+
+namespace Microsoft.SemanticKernel;
+
+/// <summary>
+/// Provides extension methods for the <see cref="MemoryBuilder"/> class to configure GoogleAI connector.
+/// </summary>
+public static class GoogleAIMemoryBuilderExtensions
+{
+    /// <summary>
+    /// Add GoogleAI embeddings generation service to the memory builder.
+    /// </summary>
+    /// <param name="builder">The <see cref="MemoryBuilder"/> instance</param>
+    /// <param name="modelId">The model for text generation.</param>
+    /// <param name="apiKey">The API key for authentication Gemini API.</param>
+    /// <param name="httpClient">The optional custom HttpClient.</param>
+    /// <returns>The updated memory builder.</returns>
+    public static MemoryBuilder WithGoogleAITextEmbeddingGeneration(
+        this MemoryBuilder builder,
+        string modelId,
+        string apiKey,
+        HttpClient? httpClient = null)
+    {
+        Verify.NotNull(builder);
+        Verify.NotNull(modelId);
+        Verify.NotNull(apiKey);
+
+        return builder.WithTextEmbeddingGeneration((loggerFactory, builderHttpClient) =>
+            new GoogleAITextEmbeddingGenerationService(
+                model: modelId,
+                apiKey: apiKey,
+                httpClient: HttpClientProvider.GetHttpClient(httpClient ?? builderHttpClient),
+                loggerFactory: loggerFactory));
+    }
+}

dotnet/src/Connectors/Connectors.GoogleVertexAI/Extensions/VertexAIMemoryBuilderExtensions.cs

@@ -0,0 +1,48 @@
+// Copyright (c) Microsoft. All rights reserved.
+
+using System.Net.Http;
+using Microsoft.SemanticKernel.Connectors.GoogleVertexAI;
+using Microsoft.SemanticKernel.Http;
+using Microsoft.SemanticKernel.Memory;
+
+namespace Microsoft.SemanticKernel;
+
+/// <summary>
+/// Provides extension methods for the <see cref="MemoryBuilder"/> class to configure VertexAI connector.
+/// </summary>
+public static class VertexAIMemoryBuilderExtensions
+{
+    /// <summary>
+    /// Add VertexAI embeddings generation service to the memory builder.
+    /// </summary>
+    /// <param name="builder">The <see cref="MemoryBuilder"/> instance</param>
+    /// <param name="modelId">The model for text generation.</param>
+    /// <param name="bearerKey">The Bearer Key for authentication.</param>
+    /// <param name="location">The location to process the request</param>
+    /// <param name="projectId">Your project ID</param>
+    /// <param name="httpClient">The optional custom HttpClient.</param>
+    /// <returns>The updated memory builder.</returns>
+    public static MemoryBuilder WithVertexAITextEmbeddingGeneration(
+        this MemoryBuilder builder,
+        string modelId,
+        string bearerKey,
+        string location,
+        string projectId,
+        HttpClient? httpClient = null)
+    {
+        Verify.NotNull(builder);
+        Verify.NotNull(modelId);
+        Verify.NotNull(bearerKey);
+        Verify.NotNull(location);
+        Verify.NotNull(projectId);
+
+        return builder.WithTextEmbeddingGeneration((loggerFactory, builderHttpClient) =>
+            new VertexAITextEmbeddingGenerationService(
+                model: modelId,
+                bearerKey: bearerKey,
+                location: location,
+                projectId: projectId,
+                httpClient: HttpClientProvider.GetHttpClient(httpClient ?? builderHttpClient),
+                loggerFactory: loggerFactory));
+    }
+}