Refactor and fix embedding functionality

Author: jochenkirstaetter

src/Mscc.GenerativeAI.Web/GenerativeAIOptions.cs

@@ -1,6 +1,4 @@
-using System;
-
-namespace Mscc.GenerativeAI.Web
+namespace Mscc.GenerativeAI.Web
 {
     public interface IGenerativeAIOptions
     {

src/Mscc.GenerativeAI/Types/Content.cs

@@ -7,6 +7,8 @@ namespace Mscc.GenerativeAI
     [DebuggerDisplay($"{{{nameof(GetDebuggerDisplay)}(),nq}}")]
     public class Content
     {
+        private List<Part>? partTypes;
+        
         [JsonIgnore]
         public List<IPart>? Parts { get; set; }
         public string? Role { get; set; }
@@ -18,43 +20,40 @@ public virtual List<Part>? PartTypes
             get
             {
                 SynchronizeParts();
-                return PartTypes;
-            }
-            set
-            {
-                PartTypes = value;
+                return partTypes;
             }
+            set => partTypes = value;
         }
 
         private void SynchronizeParts()
         {
+            partTypes = null;
             if (Parts == null) return;
 
-            PartTypes = new List<Part>();
+            partTypes = new List<Part>();
             foreach (var part in Parts)
             {
                 if (part is TextData text)
                 {
-                    PartTypes.Add(new Part { TextData = text });
+                    partTypes.Add(new Part { TextData = text });
                 }
                 if (part is InlineData inline)
                 {
-                    PartTypes.Add(new Part { InlineData = inline });
+                    partTypes.Add(new Part { InlineData = inline });
                 }
                 if (part is FileData file)
                 {
-                    PartTypes.Add(new Part { FileData = file });
+                    partTypes.Add(new Part { FileData = file });
                 }
                 if (part is FunctionResponse response)
                 {
-                    PartTypes.Add(new Part { FunctionResponse = response });
+                    partTypes.Add(new Part { FunctionResponse = response });
                 }
                 if (part is FunctionCall call)
                 {
-                    PartTypes.Add(new Part { FunctionCall = call });
+                    partTypes.Add(new Part { FunctionCall = call });
                 }
             }
-
         }
 
         private string GetDebuggerDisplay()

src/Mscc.GenerativeAI/Types/EmbedContentRequest.cs

@@ -2,7 +2,6 @@
 using System.Text.Json.Serialization;
 #endif
 using System.Collections.Generic;
-using System.Linq;
 
 namespace Mscc.GenerativeAI
 {

src/Mscc.GenerativeAI/Types/EmbedContentResponse.cs

@@ -2,7 +2,6 @@
 using System.Text.Json.Serialization;
 #endif
 using System.Collections.Generic;
-using System.Linq;
 
 namespace Mscc.GenerativeAI
 {

src/Mscc.GenerativeAI/Types/Embedding.cs

@@ -2,7 +2,6 @@
 using System.Text.Json.Serialization;
 #endif
 using System.Collections.Generic;
-using System.Linq;
 
 namespace Mscc.GenerativeAI
 {

tests/GoogleAi_Embedding_Should.cs

@@ -1,7 +1,5 @@
 using FluentAssertions;
 using Mscc.GenerativeAI;
-using System.Collections.Generic;
-using System.Linq;
 using Xunit;
 using Xunit.Abstractions;
 
@@ -25,6 +23,7 @@ public void Initialize_Model()
         {
             // Arrange
             var expected = Model.Embedding;
+            
             // Act
             var model = new GenerativeModel(apiKey: fixture.ApiKey, model: this.model);
 
@@ -47,7 +46,10 @@ public async void Embed_Content()
             response.Should().NotBeNull();
             response.Embedding.Should().NotBeNull();
             response.Embedding.Values.Should().NotBeNull().And.HaveCountGreaterThanOrEqualTo(1);
-            response.ForEach(x => output.WriteLine(x));
+            response.Embedding.Values.ForEach(x =>
+            {
+                output.WriteLine(x.ToString());
+            });
         }
     }
 }