From c8be61d0dd3dafc6ccb83ac7b436b7b0998d6624 Mon Sep 17 00:00:00 2001
From: Daniel Cazzulino <daniel@cazzulino.com>
Date: Thu, 1 Jan 2026 21:13:34 -0300
Subject: [PATCH] Rename extension method to add xAI protocol clients

Since they are not strictly Grok-specific.
---
 ...ensions.cs => ProtocolServiceCollectionExtensions.cs} | 9 ++++-----
 src/xAI.Tests/SanityChecks.cs                            | 4 ++--
 2 files changed, 6 insertions(+), 7 deletions(-)
 rename src/xAI.Protocol/{GrokServiceCollectionExtensions.cs => ProtocolServiceCollectionExtensions.cs} (92%)

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;
 
-/// <summary>Registration extensions for Grok gRPC clients.</summary>
+/// <summary>Registration extensions for xAI gRPC clients.</summary>
 [EditorBrowsable(EditorBrowsableState.Never)]
-public static class GrokServiceCollectionExtensions
+public static class ProtocolServiceCollectionExtensions
 {
     /// <summary>
-    /// Registers Grok gRPC clients with the specified API key.
+    /// Registers xAI gRPC protocol clients with the specified API key.
     /// </summary>
-    public static IServiceCollection AddGrokClient(this IServiceCollection services, string apiKey,
+    public static IServiceCollection AddxAIProtocol(this IServiceCollection services, string apiKey,
         Action<GrpcClientFactoryOptions>? configureClient = null,
         Action<IHttpClientBuilder>? configureHttp = null)
     {
         var address = new Uri("https://api.x.ai/");
-
         var builder = services.AddGrpcClient<Auth.AuthClient>(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<Models.ModelsClient>();
@@ -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<Chat.ChatClient>();