Skip to content

Commit 5356f51

Browse files
committed
Update source
1 parent 6df8292 commit 5356f51

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

LLama.Unittest/LLamaEmbedderTests.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,8 +45,8 @@ private async Task CompareEmbeddings(string modelPath)
4545
var generator = (IEmbeddingGenerator<string, Embedding<float>>)embedder;
4646
Assert.NotNull(generator.GetService<EmbeddingGeneratorMetadata>());
4747
Assert.Equal(nameof(LLamaEmbedder), generator.GetService<EmbeddingGeneratorMetadata>()?.ProviderName);
48-
Assert.NotNull(generator.GetService<EmbeddingGeneratorMetadata>()?.ModelId);
49-
Assert.NotEmpty(generator.GetService<EmbeddingGeneratorMetadata>()?.ModelId!);
48+
Assert.NotNull(generator.GetService<EmbeddingGeneratorMetadata>()?.DefaultModelId);
49+
Assert.NotEmpty(generator.GetService<EmbeddingGeneratorMetadata>()?.DefaultModelId!);
5050
Assert.Same(embedder, generator.GetService<LLamaEmbedder>());
5151
Assert.Same(generator, generator.GetService<IEmbeddingGenerator<string, Embedding<float>>>());
5252
Assert.Null(generator.GetService<string>());

LLama/LLamaEmbedder.EmbeddingGenerator.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,8 @@ public partial class LLamaEmbedder
2222
{
2323
return _metadata ??= new(
2424
nameof(LLamaEmbedder),
25-
modelId: Context.NativeHandle.ModelHandle.ReadMetadata().TryGetValue("general.name", out var name) ? name : null,
26-
dimensions: EmbeddingSize);
25+
defaultModelId: Context.NativeHandle.ModelHandle.ReadMetadata().TryGetValue("general.name", out var name) ? name : null,
26+
defaultModelDimensions: EmbeddingSize);
2727
}
2828

2929
if (serviceType?.IsInstanceOfType(Context) is true)

0 commit comments

Comments
 (0)