-
+
diff --git a/examples/ollama/CommunityToolkit.Aspire.Hosting.Ollama.Web/Components/Pages/Home.razor b/examples/ollama/CommunityToolkit.Aspire.Hosting.Ollama.Web/Components/Pages/Home.razor
index 8df343793..3abe4b3f3 100644
--- a/examples/ollama/CommunityToolkit.Aspire.Hosting.Ollama.Web/Components/Pages/Home.razor
+++ b/examples/ollama/CommunityToolkit.Aspire.Hosting.Ollama.Web/Components/Pages/Home.razor
@@ -14,13 +14,13 @@
diff --git a/examples/ollama/CommunityToolkit.Aspire.Hosting.Ollama.Web/Program.cs b/examples/ollama/CommunityToolkit.Aspire.Hosting.Ollama.Web/Program.cs
index 33e3f3d40..1f56018a6 100644
--- a/examples/ollama/CommunityToolkit.Aspire.Hosting.Ollama.Web/Program.cs
+++ b/examples/ollama/CommunityToolkit.Aspire.Hosting.Ollama.Web/Program.cs
@@ -12,8 +12,8 @@
builder.Services.AddOutputCache();
-builder.AddOllamaApiClient(ServiceKeys.Phi3).AddChatClient();
-builder.AddOllamaApiClient(ServiceKeys.Llama).AddChatClient();
+builder.AddKeyedOllamaApiClient(ServiceKeys.Phi3).AddKeyedChatClient();
+builder.AddKeyedOllamaApiClient(ServiceKeys.Llama).AddKeyedChatClient();
var app = builder.Build();
diff --git a/tests/CommunityToolkit.Aspire.OllamaSharp.Tests/OllamaApiClientTests.cs b/tests/CommunityToolkit.Aspire.OllamaSharp.Tests/OllamaApiClientTests.cs
index 775945666..bcb143e91 100644
--- a/tests/CommunityToolkit.Aspire.OllamaSharp.Tests/OllamaApiClientTests.cs
+++ b/tests/CommunityToolkit.Aspire.OllamaSharp.Tests/OllamaApiClientTests.cs
@@ -140,8 +140,8 @@ public void RegisteringChatClientAndEmbeddingGeneratorReturnsCorrectModelForServ
var chatClient = host.Services.GetRequiredService();
var embeddingGenerator = host.Services.GetRequiredService>>();
- Assert.Equal("Chat", chatClient.GetService()?.ModelId);
- Assert.Equal("Embedding", embeddingGenerator.GetService()?.ModelId);
+ Assert.Equal("Chat", chatClient.GetService()?.DefaultModelId);
+ Assert.Equal("Embedding", embeddingGenerator.GetService()?.DefaultModelId);
}
[Fact]