diff --git a/eng/Versions.props b/eng/Versions.props
index a8584efb18..81595bd15c 100644
--- a/eng/Versions.props
+++ b/eng/Versions.props
@@ -46,7 +46,7 @@
     <MicrosoftDotNetInteractiveVersion>1.0.0-beta.24375.2</MicrosoftDotNetInteractiveVersion>
     <MicrosoftMLOnnxRuntimeVersion>1.18.1</MicrosoftMLOnnxRuntimeVersion>
     <MlNetMklDepsVersion>0.0.0.12</MlNetMklDepsVersion>
-    <MicrosoftExtensionsAIVersion>9.0.0-preview.9.24507.7</MicrosoftExtensionsAIVersion>
+    <MicrosoftExtensionsAIVersion>9.0.1-preview.1.24570.5</MicrosoftExtensionsAIVersion>
     <!-- runtime.native.System.Data.SqlClient.sni is not updated by dependency flow as it is not produced live anymore. -->
     <RuntimeNativeSystemDataSqlClientSniVersion>4.4.0</RuntimeNativeSystemDataSqlClientSniVersion>
     <!--
diff --git a/src/Microsoft.ML.GenAI.Core/CausalLMPipelineChatClient.cs b/src/Microsoft.ML.GenAI.Core/CausalLMPipelineChatClient.cs
index c0aa398ed0..7d1ffee34d 100644
--- a/src/Microsoft.ML.GenAI.Core/CausalLMPipelineChatClient.cs
+++ b/src/Microsoft.ML.GenAI.Core/CausalLMPipelineChatClient.cs
@@ -82,8 +82,7 @@ public virtual void Dispose()
     {
     }
 
-    public virtual TService? GetService<TService>(object? key = null) where TService : class
-    {
-        return null;
-    }
+    public virtual object? GetService(Type serviceType, object? serviceKey = null) =>
+        serviceKey is null && serviceType is not null && serviceType.IsAssignableFrom(GetType()) ? this :
+        null;
 }