Merge pull request #15 from doggy8088/Fix_Testing_Code_Initialize_Model

Fix a bug in Initialize_Model() test
mscraftsman · Mar 27, 2024 · fbf4fea · fbf4fea
2 parents 22a5ab0 + b298b6d
commit fbf4fea
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/Mscc.GenerativeAI/GoogleAi_Embedding_Should.cs b/tests/Mscc.GenerativeAI/GoogleAi_Embedding_Should.cs
@@ -22,8 +22,8 @@ public GoogleAi_Embedding_Should(ITestOutputHelper output, ConfigurationFixture
         public void Initialize_Model()
         {
             // Arrange
-            var expected = Model.Embedding;
-            
+            var expected = Model.Embedding.SanitizeModelName();
+
             // Act
             var model = new GenerativeModel(apiKey: fixture.ApiKey, model: this.model);