diff --git a/src/Libraries/Microsoft.Extensions.AI/Embeddings/AnonymousDelegatingEmbeddingGenerator.cs b/src/Libraries/Microsoft.Extensions.AI/Embeddings/AnonymousDelegatingEmbeddingGenerator.cs index 9dd838be42e..0f6c696bd0d 100644 --- a/src/Libraries/Microsoft.Extensions.AI/Embeddings/AnonymousDelegatingEmbeddingGenerator.cs +++ b/src/Libraries/Microsoft.Extensions.AI/Embeddings/AnonymousDelegatingEmbeddingGenerator.cs @@ -12,7 +12,7 @@ namespace Microsoft.Extensions.AI; /// A delegating embedding generator that wraps an inner generator with implementations provided by delegates. /// The type from which embeddings will be generated. /// The type of embeddings to generate. -public sealed class AnonymousDelegatingEmbeddingGenerator : DelegatingEmbeddingGenerator +internal sealed class AnonymousDelegatingEmbeddingGenerator : DelegatingEmbeddingGenerator where TEmbedding : Embedding { /// The delegate to use as the implementation of . diff --git a/test/Libraries/Microsoft.Extensions.AI.Tests/Embeddings/UseDelegateEmbeddingGeneratorTests.cs b/test/Libraries/Microsoft.Extensions.AI.Tests/Embeddings/UseDelegateEmbeddingGeneratorTests.cs index ab178727cb9..1109cbc581a 100644 --- a/test/Libraries/Microsoft.Extensions.AI.Tests/Embeddings/UseDelegateEmbeddingGeneratorTests.cs +++ b/test/Libraries/Microsoft.Extensions.AI.Tests/Embeddings/UseDelegateEmbeddingGeneratorTests.cs @@ -19,13 +19,6 @@ public void InvalidArgs_Throws() Assert.Throws("generateFunc", () => builder.Use((Func, EmbeddingGenerationOptions?, IEmbeddingGenerator>, CancellationToken, Task>>>)null!)); - - Assert.Throws("innerGenerator", () => - new AnonymousDelegatingEmbeddingGenerator>( - null!, (values, options, innerGenerator, cancellationToken) => Task.FromResult(new GeneratedEmbeddings>(Array.Empty>())))); - - Assert.Throws("generateFunc", () => - new AnonymousDelegatingEmbeddingGenerator>(generator, null!)); } [Fact]