diff --git a/src/libs/Predibase/Predibase.csproj b/src/libs/Predibase/Predibase.csproj
index a070d04..de2c6b5 100644
--- a/src/libs/Predibase/Predibase.csproj
+++ b/src/libs/Predibase/Predibase.csproj
@@ -10,7 +10,7 @@
   </PropertyGroup>
 
   <ItemGroup>
-    <PackageReference Include="Microsoft.Extensions.AI" Version="10.5.2" />
+    <PackageReference Include="Microsoft.Extensions.AI" Version="10.6.0" />
   </ItemGroup>
 
 </Project>
diff --git a/src/tests/IntegrationTests/Predibase.IntegrationTests.csproj b/src/tests/IntegrationTests/Predibase.IntegrationTests.csproj
index ba74798..74887a6 100644
--- a/src/tests/IntegrationTests/Predibase.IntegrationTests.csproj
+++ b/src/tests/IntegrationTests/Predibase.IntegrationTests.csproj
@@ -5,7 +5,7 @@
   </PropertyGroup>
 
   <ItemGroup>
-    <PackageReference Include="MSTest" Version="4.2.2" />
+    <PackageReference Include="MSTest" Version="4.2.3" />
     <PackageReference Include="GitHubActionsTestLogger" Version="3.0.4">
       <PrivateAssets>all</PrivateAssets>
       <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>