Expose specific protocol clients from GrokClient

src/xAI/GrokClient.cs

@@ -1,6 +1,7 @@
 using System.Collections.Concurrent;
 using System.Net.Http.Headers;
 using Grpc.Net.Client;
+using xAI.Protocol;
 
 namespace xAI;
 
@@ -23,6 +24,27 @@ public sealed class GrokClient(string apiKey, GrokClientOptions options) : IDisp
     /// <summary>Gets the options used to configure the client.</summary>
     public GrokClientOptions Options { get; } = options;
 
+    /// <summary>Gets a new instance of <see cref="Auth.AuthClient"/> that reuses the client configuration details provided to the <see cref="GrokClient"/> instance.</summary>
+    public Auth.AuthClient GetAuthClient() => new(Channel);
+
+    /// <summary>Gets a new instance of <see cref="Chat.ChatClient"/> that reuses the client configuration details provided to the <see cref="GrokClient"/> instance.</summary>
+    public Chat.ChatClient GetChatClient() => new(Channel);
+
+    /// <summary>Gets a new instance of <see cref="Documents.DocumentsClient"/> that reuses the client configuration details provided to the <see cref="GrokClient"/> instance.</summary>
+    public Documents.DocumentsClient GetDocumentsClient() => new(Channel);
+
+    /// <summary>Gets a new instance of <see cref="Embedder.EmbedderClient"/> that reuses the client configuration details provided to the <see cref="GrokClient"/> instance.</summary>
+    public Embedder.EmbedderClient GetEmbedderClient() => new(Channel);
+
+    /// <summary>Gets a new instance of <see cref="Image.ImageClient"/> that reuses the client configuration details provided to the <see cref="GrokClient"/> instance.</summary>
+    public Image.ImageClient GetImageClient() => new(Channel);
+
+    /// <summary>Gets a new instance of <see cref="Models.ModelsClient"/> that reuses the client configuration details provided to the <see cref="GrokClient"/> instance.</summary>
+    public Models.ModelsClient GetModelsClient() => new(Channel);
+
+    /// <summary>Gets a new instance of <see cref="Tokenize.TokenizeClient"/> that reuses the client configuration details provided to the <see cref="GrokClient"/> instance.</summary>
+    public Tokenize.TokenizeClient GetTokenizeClient() => new(Channel);
+
     internal GrpcChannel Channel => channels.GetOrAdd((Endpoint, ApiKey), key =>
     {
         var handler = new AuthenticationHeaderHandler(ApiKey)

src/xAI/GrokClientExtensions.cs

@@ -1,5 +1,6 @@
 using System.ComponentModel;
 using Microsoft.Extensions.AI;
+using xAI.Protocol;
 
 namespace xAI;
 
@@ -10,4 +11,8 @@ public static class GrokClientExtensions
     /// <summary>Creates a new <see cref="IChatClient"/> from the specified <see cref="GrokClient"/> using the given model as the default.</summary>
     public static IChatClient AsIChatClient(this GrokClient client, string defaultModelId)
         => new GrokChatClient(client.Channel, client.Options, defaultModelId);
-}
+
+    /// <summary>Creates a new <see cref="IChatClient"/> from the specified <see cref="Chat.ChatClient"/> using the given model as the default.</summary>
+    public static IChatClient AsIChatClient(this Chat.ChatClient client, string defaultModelId)
+        => new GrokChatClient(client, defaultModelId);
+}