rename

Author: soenneker

src/KernelPoolGeminiExtension.cs

@@ -14,7 +14,7 @@ namespace Soenneker.SemanticKernel.Pool.Gemini;
 /// <summary>
 /// Provides Gemini-specific registration extensions for KernelPoolManager, enabling integration with local LLMs via Semantic Kernel.
 /// </summary>
-public static class KernelPoolGeminiExtension
+public static class SemanticKernelPoolGeminiExtension
 {
     /// <summary>
     /// Registers a Gemini model in the kernel pool with optional rate and token limits.
@@ -31,7 +31,7 @@ public static class KernelPoolGeminiExtension
     /// <param name="tokensPerDay">Optional maximum number of tokens allowed per day.</param>
     /// <param name="cancellationToken">A cancellation token that can be used to cancel the operation.</param>
     /// <returns>A <see cref="ValueTask"/> representing the asynchronous registration operation.</returns>
-    public static ValueTask RegisterGemini(this IKernelPoolManager pool, string key, string modelId, string apiKey, string endpoint, IHttpClientCache httpClientCache, int? rps,
+    public static ValueTask RegisterGemini(this ISemanticKernelPool pool, string key, string modelId, string apiKey, string endpoint, IHttpClientCache httpClientCache, int? rps,
         int? rpm, int? rpd, int? tokensPerDay = null, CancellationToken cancellationToken = default)
     {
         var options = new SemanticKernelOptions
@@ -47,12 +47,11 @@ public static ValueTask RegisterGemini(this IKernelPoolManager pool, string key,
             {
                 HttpClient httpClient = await httpClientCache.Get($"gemini:{modelId}", () => new HttpClientOptions
                 {
-                    Timeout = TimeSpan.FromSeconds(600),
-                    BaseAddress = opts.Endpoint
-                }, cancellationToken);
+                    Timeout = TimeSpan.FromSeconds(300)
+                }, cancellationToken).NoSync();
 
 #pragma warning disable SKEXP0070
-                return Kernel.CreateBuilder().AddGoogleAIGeminiChatCompletion(modelId: opts.ModelId!, apiKey, httpClient: httpClient);
+                return Kernel.CreateBuilder().AddGoogleAIGeminiChatCompletion(modelId: opts.ModelId!, opts.ApiKey!, httpClient: httpClient);
 #pragma warning restore SKEXP0070
             }
         };
@@ -68,7 +67,7 @@ public static ValueTask RegisterGemini(this IKernelPoolManager pool, string key,
     /// <param name="httpClientCache">The HTTP client cache to remove the associated client from.</param>
     /// <param name="cancellationToken">A cancellation token that can be used to cancel the operation.</param>
     /// <returns>A <see cref="ValueTask"/> representing the asynchronous unregistration operation.</returns>
-    public static async ValueTask UnregisterGemini(this IKernelPoolManager pool, string key, IHttpClientCache httpClientCache, CancellationToken cancellationToken = default)
+    public static async ValueTask UnregisterGemini(this ISemanticKernelPool pool, string key, IHttpClientCache httpClientCache, CancellationToken cancellationToken = default)
     {
         await pool.Unregister(key, cancellationToken).NoSync();
         await httpClientCache.Remove($"gemini:{key}", cancellationToken).NoSync();

src/Soenneker.SemanticKernel.Pool.Gemini.csproj

@@ -43,7 +43,7 @@
 
     <ItemGroup>
         <PackageReference Include="Microsoft.SemanticKernel.Connectors.Google" Version="1.51.0-alpha" />
-        <PackageReference Include="Soenneker.SemanticKernel.Pool" Version="3.0.8" />
+        <PackageReference Include="Soenneker.SemanticKernel.Pool" Version="3.0.10" />
         <PackageReference Include="Soenneker.Utils.HttpClientCache" Version="3.0.1692" />
     </ItemGroup>
 </Project>

test/Soenneker.SemanticKernel.Pool.Gemini.Tests/Soenneker.SemanticKernel.Pool.Gemini.Tests.csproj

@@ -25,7 +25,7 @@
             <PrivateAssets>all</PrivateAssets>
         </PackageReference>
         <PackageReference Include="Soenneker.Utils.Test" Version="3.0.283" />
-        <PackageReference Include="Soenneker.Tests.FixturedUnit" Version="3.0.1842" />
+        <PackageReference Include="Soenneker.Tests.FixturedUnit" Version="3.0.1843" />
         <PackageReference Include="xUnit.v3" Version="2.0.2" />
     </ItemGroup>