diff --git a/src/xAI.Protocol/GrokServiceCollectionExtensions.cs b/src/xAI.Protocol/ProtocolServiceCollectionExtensions.cs
similarity index 92%
rename from src/xAI.Protocol/GrokServiceCollectionExtensions.cs
rename to src/xAI.Protocol/ProtocolServiceCollectionExtensions.cs
index bfcc087..ccb4b00 100644
--- a/src/xAI.Protocol/GrokServiceCollectionExtensions.cs
+++ b/src/xAI.Protocol/ProtocolServiceCollectionExtensions.cs
@@ -4,19 +4,18 @@
namespace Microsoft.Extensions.DependencyInjection;
-/// Registration extensions for Grok gRPC clients.
+/// Registration extensions for xAI gRPC clients.
[EditorBrowsable(EditorBrowsableState.Never)]
-public static class GrokServiceCollectionExtensions
+public static class ProtocolServiceCollectionExtensions
{
///
- /// Registers Grok gRPC clients with the specified API key.
+ /// Registers xAI gRPC protocol clients with the specified API key.
///
- public static IServiceCollection AddGrokClient(this IServiceCollection services, string apiKey,
+ public static IServiceCollection AddxAIProtocol(this IServiceCollection services, string apiKey,
Action? configureClient = null,
Action? configureHttp = null)
{
var address = new Uri("https://api.x.ai/");
-
var builder = services.AddGrpcClient(options =>
{
options.Address = address;
diff --git a/src/xAI.Tests/SanityChecks.cs b/src/xAI.Tests/SanityChecks.cs
index a9c6e4c..9236b25 100644
--- a/src/xAI.Tests/SanityChecks.cs
+++ b/src/xAI.Tests/SanityChecks.cs
@@ -11,7 +11,7 @@ public class SanityChecks(ITestOutputHelper output)
public async Task ListModelsAsync()
{
var services = new ServiceCollection()
- .AddGrokClient(Environment.GetEnvironmentVariable("XAI_API_KEY")!)
+ .AddxAIProtocol(Environment.GetEnvironmentVariable("XAI_API_KEY")!)
.BuildServiceProvider();
var client = services.GetRequiredService();
@@ -28,7 +28,7 @@ public async Task ListModelsAsync()
public async Task ExecuteLocalFunctionWithWebSearch()
{
var services = new ServiceCollection()
- .AddGrokClient(Environment.GetEnvironmentVariable("XAI_API_KEY")!)
+ .AddxAIProtocol(Environment.GetEnvironmentVariable("XAI_API_KEY")!)
.BuildServiceProvider();
var client = services.GetRequiredService();