add embeddings

Author: jochenkirstaetter

src/Mscc.GenerativeAI/Types/EmbedContentRequest.cs

@@ -0,0 +1,40 @@
+#if NET472_OR_GREATER || NETSTANDARD2_0
+using System.Text.Json.Serialization;
+#endif
+using System.Collections.Generic;
+using System.Linq;
+
+namespace Mscc.GenerativeAI
+{
+    /// <summary>
+    /// 
+    /// </summary>
+    public class EmbedContentRequest
+    {
+        public string Model { get; } = $"models/{GenerativeAI.Model.Embedding}";
+
+        /// <summary>
+        /// 
+        /// </summary>
+        public Content Content { get; set; }
+        // [JsonPropertyName("generation_config")]
+        // public GenerationConfig? GenerationConfig { get; set; }
+        // [JsonPropertyName("safety_settings")]
+        // public List<SafetySetting>? SafetySettings { get; set; }
+        // public List<Tool>? Tools { get; set; }
+
+        public EmbedContentRequest(string prompt, GenerationConfig? generationConfig = null, List<SafetySetting>? safetySettings = null, List<Tool>? tools = null)
+        {
+            Content = new Content
+            {
+                Parts = new List<IPart> { new TextData
+                {
+                    Text = prompt
+                }}
+            };
+            // if (generationConfig != null) GenerationConfig = generationConfig;
+            // if (safetySettings != null) SafetySettings = safetySettings;
+            // if (tools != null) Tools = tools;
+        }
+    }
+}

src/Mscc.GenerativeAI/Types/EmbedContentResponse.cs

@@ -0,0 +1,20 @@
+#if NET472_OR_GREATER || NETSTANDARD2_0
+using System.Text.Json.Serialization;
+#endif
+using System.Collections.Generic;
+using System.Linq;
+
+namespace Mscc.GenerativeAI
+{
+    /// <summary>
+    /// 
+    /// </summary>
+    public class EmbedContentResponse
+    {
+        /// <summary>
+        /// 
+        /// </summary>
+        public Embedding? Embedding { get; set; }
+        public List<Embedding>? Embeddings { get; set; }
+    }
+}

src/Mscc.GenerativeAI/Types/Embedding.cs

@@ -0,0 +1,19 @@
+#if NET472_OR_GREATER || NETSTANDARD2_0
+using System.Text.Json.Serialization;
+#endif
+using System.Collections.Generic;
+using System.Linq;
+
+namespace Mscc.GenerativeAI
+{
+    /// <summary>
+    /// 
+    /// </summary>
+    public class Embedding
+    {
+        /// <summary>
+        /// 
+        /// </summary>
+        public List<float> Values { get; set; }
+    }
+}

tests/GoogleAi_Embedding_Should.cs

@@ -0,0 +1,53 @@
+using FluentAssertions;
+using Mscc.GenerativeAI;
+using System.Collections.Generic;
+using System.Linq;
+using Xunit;
+using Xunit.Abstractions;
+
+namespace Test.Mscc.GenerativeAI
+{
+    [Collection(nameof(ConfigurationFixture))]
+    public class GoogleAi_Embedding_Should
+    {
+        private readonly ITestOutputHelper output;
+        private readonly ConfigurationFixture fixture;
+        private readonly string model = Model.Embedding;
+
+        public GoogleAi_Embedding_Should(ITestOutputHelper output, ConfigurationFixture fixture)
+        {
+            this.output = output;
+            this.fixture = fixture;
+        }
+
+        [Fact]
+        public void Initialize_Model()
+        {
+            // Arrange
+            var expected = Model.Embedding;
+            // Act
+            var model = new GenerativeModel(apiKey: fixture.ApiKey, model: this.model);
+
+            // Assert
+            model.Should().NotBeNull();
+            model.Name().Should().Be(expected);
+        }
+
+        [Fact]
+        public async void Embed_Content()
+        {
+            // Arrange
+            var prompt = "Write a story about a magic backpack.";
+            var model = new GenerativeModel(apiKey: fixture.ApiKey, model: this.model);
+
+            // Act
+            var response = await model.EmbedContent(prompt);
+
+            // Assert
+            response.Should().NotBeNull();
+            response.Embedding.Should().NotBeNull();
+            response.Embedding.Values.Should().NotBeNull().And.HaveCountGreaterThanOrEqualTo(1);
+            response.ForEach(x => output.WriteLine(x));
+        }
+    }
+}