36 instantiations of Embedding
Aspire.Azure.AI.OpenAI.Tests (1)
AspireAzureOpenAIClientBuilderEmbeddingGeneratorExtensionsTests.cs (1)
224return Task.FromResult(new GeneratedEmbeddings<Embedding<float>>(inputs.Select(i => new Embedding<float>(floats))));
Aspire.OpenAI.Tests (1)
AspireOpenAIClientBuilderEmbeddingGeneratorExtensionsTests.cs (1)
271return Task.FromResult(new GeneratedEmbeddings<Embedding<float>>(inputs.Select(i => new Embedding<float>(floats))));
Microsoft.Extensions.AI.Abstractions.Tests (27)
Embeddings\EmbeddingGeneratorExtensionsTests.cs (2)
89Embedding<float> result = new(new float[] { 1f, 2f, 3f }); 110.Select(i => new Embedding<float>(Enumerable.Range(i, 4).Select(i => (float)i).ToArray()))
Embeddings\EmbeddingTests.cs (4)
22Embedding<float> e = new(floats) 43Embedding<Half> e = new(halfs); 57Embedding<float> e = new(floats); 70Embedding<double> e = new(floats);
Embeddings\GeneratedEmbeddingsTests.cs (21)
42Assert.False(instance.Contains(new Embedding<float>(new float[] { 1, 2, 3 }))); 45Assert.Equal(-1, instance.IndexOf(new Embedding<float>(new float[] { 1, 2, 3 }))); 86var e = new Embedding<float>(new float[] { 7, 8, 9 }); 117var e = new Embedding<float>(new float[] { 1, 2, 3 }); 129var e1 = new Embedding<float>(new float[] { 1, 2, 3 }); 130var e2 = new Embedding<float>(new float[] { 4, 5, 6 }); 144var e1 = new Embedding<float>(new float[] { 1, 2, 3 }); 145var e2 = new Embedding<float>(new float[] { 4, 5, 6 }); 160var e1 = new Embedding<float>(new float[] { 1, 2, 3 }); 161var e2 = new Embedding<float>(new float[] { 4, 5, 6 }); 183var e1 = new Embedding<float>(new float[] { 1, 2, 3 }); 184var e2 = new Embedding<float>(new float[] { 4, 5, 6 }); 202var e1 = new Embedding<float>(new float[] { 1, 2, 3 }); 203var e2 = new Embedding<float>(new float[] { 4, 5, 6 }); 208var e3 = new Embedding<float>(new float[] { 7, 8, 9 }); 220var e1 = new Embedding<float>(new float[] { 1, 2, 3 }); 221var e2 = new Embedding<float>(new float[] { 4, 5, 6 }); 226var e3 = new Embedding<float>(new float[] { 7, 8, 9 }); 238var e1 = new Embedding<float>(new float[] { 1, 2, 3 }); 239var e2 = new Embedding<float>(new float[] { 4, 5, 6 }); 251var expectedGeneratedEmbeddings = new GeneratedEmbeddings<Embedding<float>>([new Embedding<float>(new float[] { 1, 2, 3 })]);
Microsoft.Extensions.AI.AzureAIInference (1)
AzureAIInferenceEmbeddingGenerator.cs (1)
97new Embedding<float>(ParseBase64Floats(e.Embedding))
Microsoft.Extensions.AI.Integration.Tests (1)
QuantizationEmbeddingGenerator.cs (1)
83return new(result)
Microsoft.Extensions.AI.OpenAI (1)
OpenAIEmbeddingGenerator.cs (1)
107new Embedding<float>(e.ToFloats())
Microsoft.Extensions.AI.Tests (4)
Embeddings\DistributedCachingEmbeddingGeneratorTest.cs (1)
17private readonly Embedding<float> _expectedEmbedding = new(new float[] { 1.0f, 2.0f, 3.0f })
Embeddings\LoggingEmbeddingGeneratorTests.cs (1)
58return Task.FromResult(new GeneratedEmbeddings<Embedding<float>>([new Embedding<float>(new float[] { 1f, 2f, 3f })]));
Embeddings\OpenTelemetryEmbeddingGeneratorTests.cs (1)
36return new GeneratedEmbeddings<Embedding<float>>([new Embedding<float>(new float[] { 1, 2, 3 })])
Embeddings\UseDelegateEmbeddingGeneratorTests.cs (1)
53e.Add(new Embedding<float>(default));
229 references to Embedding
Aspire.Azure.AI.OpenAI.Tests (18)
AspireAzureOpenAIClientBuilderEmbeddingGeneratorExtensionsTests.cs (18)
36host.Services.GetRequiredKeyedService<IEmbeddingGenerator<string, Embedding<float>>>("openai_embeddinggenerator") : 37host.Services.GetRequiredService<IEmbeddingGenerator<string, Embedding<float>>>(); 66host.Services.GetRequiredKeyedService<IEmbeddingGenerator<string, Embedding<float>>>("openai_embeddinggenerator") : 67host.Services.GetRequiredService<IEmbeddingGenerator<string, Embedding<float>>>(); 94host.Services.GetRequiredKeyedService<IEmbeddingGenerator<string, Embedding<float>>>("openai_embeddinggenerator") : 95host.Services.GetRequiredService<IEmbeddingGenerator<string, Embedding<float>>>(); 125host.Services.GetRequiredKeyedService<IEmbeddingGenerator<string, Embedding<float>>>("openai_embeddinggenerator") : 126host.Services.GetRequiredService<IEmbeddingGenerator<string, Embedding<float>>>(); 156host.Services.GetRequiredKeyedService<IEmbeddingGenerator<string, Embedding<float>>>("openai_embeddinggenerator") : 157host.Services.GetRequiredService<IEmbeddingGenerator<string, Embedding<float>>>(); 187host.Services.GetRequiredKeyedService<IEmbeddingGenerator<string, Embedding<float>>>("openai_embeddinggenerator") : 188host.Services.GetRequiredService<IEmbeddingGenerator<string, Embedding<float>>>(); 190Assert.Equal(disableOpenTelemetry, generator.GetService<OpenTelemetryEmbeddingGenerator<string, Embedding<float>>>() is null); 214host.Services.GetRequiredKeyedService<IEmbeddingGenerator<string, Embedding<float>>>("openai_embeddinggenerator") : 215host.Services.GetRequiredService<IEmbeddingGenerator<string, Embedding<float>>>(); 221private Task<GeneratedEmbeddings<Embedding<float>>> TestMiddleware(IEnumerable<string> inputs, EmbeddingGenerationOptions? options, IEmbeddingGenerator<string, Embedding<float>> nextAsync, CancellationToken cancellationToken) 224return Task.FromResult(new GeneratedEmbeddings<Embedding<float>>(inputs.Select(i => new Embedding<float>(floats))));
Aspire.OpenAI (5)
AspireOpenAIClientBuilderEmbeddingGeneratorExtensions.cs (5)
23public static EmbeddingGeneratorBuilder<string, Embedding<float>> AddEmbeddingGenerator( 40public static EmbeddingGeneratorBuilder<string, Embedding<float>> AddKeyedEmbeddingGenerator( 58private static IEmbeddingGenerator<string, Embedding<float>> CreateInnerEmbeddingGenerator( 76return new OpenTelemetryEmbeddingGenerator<string, Embedding<float>>( 78loggerFactory?.CreateLogger(typeof(OpenTelemetryEmbeddingGenerator<string, Embedding<float>>)));
Aspire.OpenAI.Tests (20)
AspireOpenAIClientBuilderEmbeddingGeneratorExtensionsTests.cs (20)
39host.Services.GetRequiredKeyedService<IEmbeddingGenerator<string, Embedding<float>>>("openai_embeddinggenerator") : 40host.Services.GetRequiredService<IEmbeddingGenerator<string, Embedding<float>>>(); 69host.Services.GetRequiredKeyedService<IEmbeddingGenerator<string, Embedding<float>>>("openai_embeddinggenerator") : 70host.Services.GetRequiredService<IEmbeddingGenerator<string, Embedding<float>>>(); 97host.Services.GetRequiredKeyedService<IEmbeddingGenerator<string, Embedding<float>>>("openai_embeddinggenerator") : 98host.Services.GetRequiredService<IEmbeddingGenerator<string, Embedding<float>>>(); 128host.Services.GetRequiredKeyedService<IEmbeddingGenerator<string, Embedding<float>>>("openai_embeddinggenerator") : 129host.Services.GetRequiredService<IEmbeddingGenerator<string, Embedding<float>>>(); 159host.Services.GetRequiredKeyedService<IEmbeddingGenerator<string, Embedding<float>>>("openai_embeddinggenerator") : 160host.Services.GetRequiredService<IEmbeddingGenerator<string, Embedding<float>>>(); 190host.Services.GetRequiredKeyedService<IEmbeddingGenerator<string, Embedding<float>>>("openai_embeddinggenerator") : 191host.Services.GetRequiredService<IEmbeddingGenerator<string, Embedding<float>>>(); 193Assert.Equal(disableOpenTelemetry, generator.GetService<OpenTelemetryEmbeddingGenerator<string, Embedding<float>>>() is null); 217host.Services.GetRequiredKeyedService<IEmbeddingGenerator<string, Embedding<float>>>("openai_embeddinggenerator") : 218host.Services.GetRequiredService<IEmbeddingGenerator<string, Embedding<float>>>(); 250host.Services.GetRequiredKeyedService<IEmbeddingGenerator<string, Embedding<float>>>("openai_embeddinggenerator") : 251host.Services.GetRequiredService<IEmbeddingGenerator<string, Embedding<float>>>(); 268private Task<GeneratedEmbeddings<Embedding<float>>> TestMiddleware(IEnumerable<string> inputs, EmbeddingGenerationOptions? options, IEmbeddingGenerator<string, Embedding<float>> nextAsync, CancellationToken cancellationToken) 271return Task.FromResult(new GeneratedEmbeddings<Embedding<float>>(inputs.Select(i => new Embedding<float>(floats))));
Microsoft.Extensions.AI (1)
Embeddings\CachingEmbeddingGenerator.cs (1)
114/// <summary>Returns a previously cached <see cref="Embedding{TEmbedding}"/>, if available.</summary>
Microsoft.Extensions.AI.Abstractions (13)
Embeddings\Embedding.cs (4)
15[JsonDerivedType(typeof(Embedding<float>), typeDiscriminator: "floats")] 16[JsonDerivedType(typeof(Embedding<double>), typeDiscriminator: "doubles")] 17[JsonDerivedType(typeof(Embedding<byte>), typeDiscriminator: "bytes")] 18[JsonDerivedType(typeof(Embedding<sbyte>), typeDiscriminator: "sbytes")]
Embeddings\Embedding{T}.cs (1)
13/// <summary>Initializes a new instance of the <see cref="Embedding{T}"/> class with the embedding vector.</summary>
Embeddings\EmbeddingGeneratorExtensions.cs (4)
103/// resulting <see cref="Embedding{T}"/>'s <see cref="Embedding{T}.Vector"/> property. 106this IEmbeddingGenerator<TInput, Embedding<TEmbeddingElement>> generator, 111var embedding = await GenerateEmbeddingAsync(generator, value, options, cancellationToken).ConfigureAwait(false);
Utilities\AIJsonUtilities.Defaults.cs (4)
103[JsonSerializable(typeof(Embedding<byte>))] 104[JsonSerializable(typeof(Embedding<int>))] 108[JsonSerializable(typeof(Embedding<float>))] 109[JsonSerializable(typeof(Embedding<double>))]
Microsoft.Extensions.AI.Abstractions.Tests (63)
Embeddings\DelegatingEmbeddingGeneratorTests.cs (7)
27var expectedResult = new TaskCompletionSource<GeneratedEmbeddings<Embedding<float>>>(); 28var expectedEmbedding = new GeneratedEmbeddings<Embedding<float>>([new(new float[] { 1.0f, 2.0f, 3.0f })]); 67var service = delegating.GetService<DelegatingEmbeddingGenerator<string, Embedding<float>>>(); 87var service = delegating.GetService<IEmbeddingGenerator<string, Embedding<float>>>(expectedKey); 115private sealed class NoOpDelegatingEmbeddingGenerator(IEmbeddingGenerator<string, Embedding<float>> innerGenerator) : 116DelegatingEmbeddingGenerator<string, Embedding<float>>(innerGenerator);
Embeddings\EmbeddingGeneratorExtensionsTests.cs (6)
31using IEmbeddingGenerator<string, Embedding<float>> generator = new TestEmbeddingGenerator 40if (serviceType == typeof(IEmbeddingGenerator<string, Embedding<float>>)) 89Embedding<float> result = new(new float[] { 1f, 2f, 3f }); 94Task.FromResult<GeneratedEmbeddings<Embedding<float>>>([result]) 108Embedding<float>[] embeddings = Enumerable 116Task.FromResult<GeneratedEmbeddings<Embedding<float>>>(new(embeddings))
Embeddings\EmbeddingTests.cs (10)
22Embedding<float> e = new(floats) 43Embedding<Half> e = new(halfs); 48Embedding<Half> result = Assert.IsType<Embedding<Half>>(JsonSerializer.Deserialize(json, TestJsonSerializerContext.Default.Embedding)); 57Embedding<float> e = new(floats); 62Embedding<float> result = Assert.IsType<Embedding<float>>(JsonSerializer.Deserialize(json, TestJsonSerializerContext.Default.Embedding)); 70Embedding<double> e = new(floats); 75Embedding<double> result = Assert.IsType<Embedding<double>>(JsonSerializer.Deserialize(json, TestJsonSerializerContext.Default.Embedding));
Embeddings\GeneratedEmbeddingsTests.cs (39)
20Assert.Throws<ArgumentNullException>("embeddings", () => new GeneratedEmbeddings<Embedding<float>>(null!)); 21Assert.Throws<ArgumentOutOfRangeException>("capacity", () => new GeneratedEmbeddings<Embedding<float>>(-1)); 27GeneratedEmbeddings<Embedding<float>>[] instances = 39Assert.False(((ICollection<Embedding<float>>)instance).IsReadOnly); 48instance.CopyTo(Array.Empty<Embedding<float>>(), 0); 58List<Embedding<float>> embeddings = 64var generatedEmbeddings = new GeneratedEmbeddings<Embedding<float>>(embeddings); 86var e = new Embedding<float>(new float[] { 7, 8, 9 }); 95GeneratedEmbeddings<Embedding<float>> embeddings = []; 116GeneratedEmbeddings<Embedding<float>> embeddings = []; 117var e = new Embedding<float>(new float[] { 1, 2, 3 }); 127GeneratedEmbeddings<Embedding<float>> embeddings = []; 129var e1 = new Embedding<float>(new float[] { 1, 2, 3 }); 130var e2 = new Embedding<float>(new float[] { 4, 5, 6 }); 142GeneratedEmbeddings<Embedding<float>> embeddings = []; 144var e1 = new Embedding<float>(new float[] { 1, 2, 3 }); 145var e2 = new Embedding<float>(new float[] { 4, 5, 6 }); 158GeneratedEmbeddings<Embedding<float>> embeddings = []; 160var e1 = new Embedding<float>(new float[] { 1, 2, 3 }); 161var e2 = new Embedding<float>(new float[] { 4, 5, 6 }); 181GeneratedEmbeddings<Embedding<float>> embeddings = []; 183var e1 = new Embedding<float>(new float[] { 1, 2, 3 }); 184var e2 = new Embedding<float>(new float[] { 4, 5, 6 }); 200GeneratedEmbeddings<Embedding<float>> embeddings = []; 202var e1 = new Embedding<float>(new float[] { 1, 2, 3 }); 203var e2 = new Embedding<float>(new float[] { 4, 5, 6 }); 208var e3 = new Embedding<float>(new float[] { 7, 8, 9 }); 218GeneratedEmbeddings<Embedding<float>> embeddings = []; 220var e1 = new Embedding<float>(new float[] { 1, 2, 3 }); 221var e2 = new Embedding<float>(new float[] { 4, 5, 6 }); 226var e3 = new Embedding<float>(new float[] { 7, 8, 9 }); 236GeneratedEmbeddings<Embedding<float>> embeddings = []; 238var e1 = new Embedding<float>(new float[] { 1, 2, 3 }); 239var e2 = new Embedding<float>(new float[] { 4, 5, 6 }); 251var expectedGeneratedEmbeddings = new GeneratedEmbeddings<Embedding<float>>([new Embedding<float>(new float[] { 1, 2, 3 })]); 253using IEmbeddingGenerator<object, Embedding<float>> acceptsObject = new TestEmbeddingGenerator<object, Embedding<float>> 258IEmbeddingGenerator<string, Embedding<float>> acceptsString = acceptsObject;
TestEmbeddingGenerator.cs (1)
41public sealed class TestEmbeddingGenerator : TestEmbeddingGenerator<string, Embedding<float>>;
Microsoft.Extensions.AI.AzureAIInference (4)
AzureAIInferenceEmbeddingGenerator.cs (3)
26IEmbeddingGenerator<string, Embedding<float>> 89public async Task<GeneratedEmbeddings<Embedding<float>>> GenerateAsync( 96GeneratedEmbeddings<Embedding<float>> result = new(embeddings.Data.Select(e =>
AzureAIInferenceExtensions.cs (1)
24public static IEmbeddingGenerator<string, Embedding<float>> AsEmbeddingGenerator(
Microsoft.Extensions.AI.AzureAIInference.Tests (11)
AzureAIInferenceEmbeddingGeneratorIntegrationTests.cs (1)
8protected override IEmbeddingGenerator<string, Embedding<float>>? CreateEmbeddingGenerator() =>
AzureAIInferenceEmbeddingGeneratorTests.cs (10)
51IEmbeddingGenerator<string, Embedding<float>> embeddingGenerator = client.AsEmbeddingGenerator(model); 64Assert.Same(embeddingGenerator, embeddingGenerator.GetService<IEmbeddingGenerator<string, Embedding<float>>>()); 67using IEmbeddingGenerator<string, Embedding<float>> pipeline = embeddingGenerator 73Assert.NotNull(pipeline.GetService<DistributedCachingEmbeddingGenerator<string, Embedding<float>>>()); 74Assert.NotNull(pipeline.GetService<CachingEmbeddingGenerator<string, Embedding<float>>>()); 75Assert.NotNull(pipeline.GetService<OpenTelemetryEmbeddingGenerator<string, Embedding<float>>>()); 78Assert.IsType<OpenTelemetryEmbeddingGenerator<string, Embedding<float>>>(pipeline.GetService<IEmbeddingGenerator<string, Embedding<float>>>()); 113using IEmbeddingGenerator<string, Embedding<float>> generator = new EmbeddingsClient(new("http://somewhere"), new AzureKeyCredential("key"), new() 129foreach (Embedding<float> e in response)
Microsoft.Extensions.AI.Integration.Tests (14)
CallCountingEmbeddingGenerator.cs (3)
13internal sealed class CallCountingEmbeddingGenerator(IEmbeddingGenerator<string, Embedding<float>> innerGenerator) 14: DelegatingEmbeddingGenerator<string, Embedding<float>>(innerGenerator) 20public override Task<GeneratedEmbeddings<Embedding<float>>> GenerateAsync(
EmbeddingGeneratorIntegrationTests.cs (3)
27private readonly IEmbeddingGenerator<string, Embedding<float>>? _embeddingGenerator; 40protected abstract IEmbeddingGenerator<string, Embedding<float>>? CreateEmbeddingGenerator(); 176using IEmbeddingGenerator<string, Embedding<Half>> generator =
QuantizationEmbeddingGenerator.cs (8)
18, IEmbeddingGenerator<string, Embedding<Half>> 21private readonly IEmbeddingGenerator<string, Embedding<float>> _floatService; 23public QuantizationEmbeddingGenerator(IEmbeddingGenerator<string, Embedding<float>> floatService) 45private static BinaryEmbedding QuantizeToBinary(Embedding<float> embedding) 67async Task<GeneratedEmbeddings<Embedding<Half>>> IEmbeddingGenerator<string, Embedding<Half>>.GenerateAsync( 78private static Embedding<Half> QuantizeToHalf(Embedding<float> embedding)
Microsoft.Extensions.AI.Ollama (2)
OllamaEmbeddingGenerator.cs (2)
18public sealed class OllamaEmbeddingGenerator : IEmbeddingGenerator<string, Embedding<float>> 85public async Task<GeneratedEmbeddings<Embedding<float>>> GenerateAsync(
Microsoft.Extensions.AI.Ollama.Tests (11)
OllamaEmbeddingGeneratorIntegrationTests.cs (1)
13protected override IEmbeddingGenerator<string, Embedding<float>>? CreateEmbeddingGenerator() =>
OllamaEmbeddingGeneratorTests.cs (10)
30Assert.Same(generator, generator.GetService<IEmbeddingGenerator<string, Embedding<float>>>()); 32using IEmbeddingGenerator<string, Embedding<float>> pipeline = generator 38Assert.NotNull(pipeline.GetService<DistributedCachingEmbeddingGenerator<string, Embedding<float>>>()); 39Assert.NotNull(pipeline.GetService<CachingEmbeddingGenerator<string, Embedding<float>>>()); 40Assert.NotNull(pipeline.GetService<OpenTelemetryEmbeddingGenerator<string, Embedding<float>>>()); 43Assert.IsType<OpenTelemetryEmbeddingGenerator<string, Embedding<float>>>(pipeline.GetService<IEmbeddingGenerator<string, Embedding<float>>>()); 52using IEmbeddingGenerator<string, Embedding<float>> generator = new OllamaEmbeddingGenerator(endpoint, model); 81using IEmbeddingGenerator<string, Embedding<float>> generator = new OllamaEmbeddingGenerator("http://localhost:11434", "all-minilm", httpClient); 94foreach (Embedding<float> e in response)
Microsoft.Extensions.AI.OpenAI (4)
OpenAIClientExtensions.cs (2)
45public static IEmbeddingGenerator<string, Embedding<float>> AsEmbeddingGenerator(this OpenAIClient openAIClient, string modelId, int? dimensions = null) => 52public static IEmbeddingGenerator<string, Embedding<float>> AsEmbeddingGenerator(this EmbeddingClient embeddingClient, int? dimensions = null) =>
OpenAIEmbeddingGenerator.cs (2)
20public sealed class OpenAIEmbeddingGenerator : IEmbeddingGenerator<string, Embedding<float>> 100public async Task<GeneratedEmbeddings<Embedding<float>>> GenerateAsync(IEnumerable<string> values, EmbeddingGenerationOptions? options = null, CancellationToken cancellationToken = default)
Microsoft.Extensions.AI.OpenAI.Tests (20)
OpenAIEmbeddingGeneratorIntegrationTests.cs (1)
8protected override IEmbeddingGenerator<string, Embedding<float>>? CreateEmbeddingGenerator() =>
OpenAIEmbeddingGeneratorTests.cs (19)
57IEmbeddingGenerator<string, Embedding<float>> embeddingGenerator = client.AsEmbeddingGenerator(model); 73IEmbeddingGenerator<string, Embedding<float>> embeddingGenerator = openAIClient.AsEmbeddingGenerator("model"); 75Assert.Same(embeddingGenerator, embeddingGenerator.GetService<IEmbeddingGenerator<string, Embedding<float>>>()); 82using IEmbeddingGenerator<string, Embedding<float>> pipeline = embeddingGenerator 88Assert.NotNull(pipeline.GetService<DistributedCachingEmbeddingGenerator<string, Embedding<float>>>()); 89Assert.NotNull(pipeline.GetService<CachingEmbeddingGenerator<string, Embedding<float>>>()); 90Assert.NotNull(pipeline.GetService<OpenTelemetryEmbeddingGenerator<string, Embedding<float>>>()); 93Assert.IsType<OpenTelemetryEmbeddingGenerator<string, Embedding<float>>>(pipeline.GetService<IEmbeddingGenerator<string, Embedding<float>>>()); 100IEmbeddingGenerator<string, Embedding<float>> embeddingGenerator = openAIClient.AsEmbeddingGenerator(); 102Assert.Same(embeddingGenerator, embeddingGenerator.GetService<IEmbeddingGenerator<string, Embedding<float>>>()); 105using IEmbeddingGenerator<string, Embedding<float>> pipeline = embeddingGenerator 111Assert.NotNull(pipeline.GetService<DistributedCachingEmbeddingGenerator<string, Embedding<float>>>()); 112Assert.NotNull(pipeline.GetService<CachingEmbeddingGenerator<string, Embedding<float>>>()); 113Assert.NotNull(pipeline.GetService<OpenTelemetryEmbeddingGenerator<string, Embedding<float>>>()); 116Assert.IsType<OpenTelemetryEmbeddingGenerator<string, Embedding<float>>>(pipeline.GetService<IEmbeddingGenerator<string, Embedding<float>>>()); 151using IEmbeddingGenerator<string, Embedding<float>> generator = new OpenAIClient(new ApiKeyCredential("apikey"), new OpenAIClientOptions 167foreach (Embedding<float> e in response)
Microsoft.Extensions.AI.Tests (43)
ChatCompletion\DependencyInjectionPatterns.cs (6)
172Assert.Equal(typeof(IEmbeddingGenerator<string, Embedding<float>>), sd.ServiceType); 194IEmbeddingGenerator<string, Embedding<float>>? g = sp.GetService<IEmbeddingGenerator<string, Embedding<float>>>(); 217Assert.Equal(typeof(IEmbeddingGenerator<string, Embedding<float>>), sd.ServiceType); 240IEmbeddingGenerator<string, Embedding<float>>? g = sp.GetKeyedService<IEmbeddingGenerator<string, Embedding<float>>>("key");
Embeddings\ConfigureOptionsEmbeddingGeneratorTests.cs (4)
16Assert.Throws<ArgumentNullException>("innerGenerator", () => new ConfigureOptionsEmbeddingGenerator<string, Embedding<float>>(null!, _ => { })); 17Assert.Throws<ArgumentNullException>("configure", () => new ConfigureOptionsEmbeddingGenerator<string, Embedding<float>>(new TestEmbeddingGenerator(), null!)); 35GeneratedEmbeddings<Embedding<float>> expectedEmbeddings = []; 38using IEmbeddingGenerator<string, Embedding<float>> innerGenerator = new TestEmbeddingGenerator
Embeddings\DistributedCachingEmbeddingGeneratorTest.cs (3)
17private readonly Embedding<float> _expectedEmbedding = new(new float[] { 1.0f, 2.0f, 3.0f }) 37return Task.FromResult<GeneratedEmbeddings<Embedding<float>>>([_expectedEmbedding]); 40using var outer = new DistributedCachingEmbeddingGenerator<string, Embedding<float>>(testGenerator, _storage)
Embeddings\EmbeddingGeneratorBuilderTests.cs (9)
19var builder = new EmbeddingGeneratorBuilder<string, Embedding<float>>(services => 60Assert.Throws<ArgumentNullException>("innerGenerator", () => new EmbeddingGeneratorBuilder<string, Embedding<float>>((IEmbeddingGenerator<string, Embedding<float>>)null!)); 61Assert.Throws<ArgumentNullException>("innerGenerator", () => ((IEmbeddingGenerator<string, Embedding<float>>)null!).AsBuilder()); 68() => new EmbeddingGeneratorBuilder<string, Embedding<float>>((Func<IServiceProvider, IEmbeddingGenerator<string, Embedding<float>>>)null!)); 81private sealed class InnerServiceCapturingEmbeddingGenerator(string name, IEmbeddingGenerator<string, Embedding<float>> innerGenerator) : 82DelegatingEmbeddingGenerator<string, Embedding<float>>(innerGenerator) 87public new IEmbeddingGenerator<string, Embedding<float>> InnerGenerator => base.InnerGenerator;
Embeddings\LoggingEmbeddingGeneratorTests.cs (10)
19Assert.Throws<ArgumentNullException>("innerGenerator", () => new LoggingEmbeddingGenerator<string, Embedding<float>>(null!, NullLogger.Instance)); 20Assert.Throws<ArgumentNullException>("logger", () => new LoggingEmbeddingGenerator<string, Embedding<float>>(new TestEmbeddingGenerator(), null!)); 28Assert.Null(innerGenerator.AsBuilder().UseLogging(NullLoggerFactory.Instance).Build().GetService(typeof(LoggingEmbeddingGenerator<string, Embedding<float>>))); 29Assert.Same(innerGenerator, innerGenerator.AsBuilder().UseLogging(NullLoggerFactory.Instance).Build().GetService(typeof(IEmbeddingGenerator<string, Embedding<float>>))); 32Assert.NotNull(innerGenerator.AsBuilder().UseLogging(factory).Build().GetService(typeof(LoggingEmbeddingGenerator<string, Embedding<float>>))); 37Assert.NotNull(innerGenerator.AsBuilder().UseLogging().Build(services).GetService(typeof(LoggingEmbeddingGenerator<string, Embedding<float>>))); 38Assert.NotNull(innerGenerator.AsBuilder().UseLogging(null).Build(services).GetService(typeof(LoggingEmbeddingGenerator<string, Embedding<float>>))); 39Assert.Null(innerGenerator.AsBuilder().UseLogging(NullLoggerFactory.Instance).Build(services).GetService(typeof(LoggingEmbeddingGenerator<string, Embedding<float>>))); 54using IEmbeddingGenerator<string, Embedding<float>> innerGenerator = new TestEmbeddingGenerator 58return Task.FromResult(new GeneratedEmbeddings<Embedding<float>>([new Embedding<float>(new float[] { 1f, 2f, 3f })]));
Embeddings\OpenTelemetryEmbeddingGeneratorTests.cs (1)
36return new GeneratedEmbeddings<Embedding<float>>([new Embedding<float>(new float[] { 1, 2, 3 })])
Embeddings\UseDelegateEmbeddingGeneratorTests.cs (8)
18EmbeddingGeneratorBuilder<string, Embedding<float>> builder = new(generator); 21builder.Use((Func<IEnumerable<string>, EmbeddingGenerationOptions?, IEmbeddingGenerator<string, Embedding<float>>, CancellationToken, Task<GeneratedEmbeddings<Embedding<float>>>>)null!)); 27GeneratedEmbeddings<Embedding<float>> expectedEmbeddings = []; 33using IEmbeddingGenerator<string, Embedding<float>> innerGenerator = new TestEmbeddingGenerator 45using IEmbeddingGenerator<string, Embedding<float>> generator = new EmbeddingGeneratorBuilder<string, Embedding<float>>(innerGenerator) 60GeneratedEmbeddings<Embedding<float>> actual = await generator.GenerateAsync(expectedValues, expectedOptions, expectedCts.Token);
test\Libraries\Microsoft.Extensions.AI.Abstractions.Tests\TestEmbeddingGenerator.cs (1)
41public sealed class TestEmbeddingGenerator : TestEmbeddingGenerator<string, Embedding<float>>;
TestJsonSerializerContext.cs (1)
21[JsonSerializable(typeof(Embedding<float>))]