diff --git a/prompt-babbler-service/Directory.Packages.props b/prompt-babbler-service/Directory.Packages.props
index 5447716..ede53dc 100644
--- a/prompt-babbler-service/Directory.Packages.props
+++ b/prompt-babbler-service/Directory.Packages.props
@@ -19,7 +19,7 @@
     <PackageVersion Include="Aspire.Microsoft.Azure.Cosmos" Version="13.2.0" />
     <PackageVersion Include="Aspire.Hosting.Azure.CosmosDB" Version="13.2.0" />
     <PackageVersion Include="Azure.AI.OpenAI" Version="2.1.0" />
-    <PackageVersion Include="Microsoft.Azure.Cosmos" Version="3.57.1" />
+    <PackageVersion Include="Microsoft.Azure.Cosmos" Version="3.58.0" />
     <PackageVersion Include="Microsoft.CognitiveServices.Speech" Version="1.48.2" />
     <PackageVersion Include="Microsoft.Agents.AI.OpenAI" Version="1.0.0-rc4" />
     <PackageVersion Include="Microsoft.Extensions.AI.OpenAI" Version="10.3.0" />
diff --git a/prompt-babbler-service/src/Orchestration/AppHost/PromptBabbler.AppHost.csproj b/prompt-babbler-service/src/Orchestration/AppHost/PromptBabbler.AppHost.csproj
index e1eccb1..73c22f8 100644
--- a/prompt-babbler-service/src/Orchestration/AppHost/PromptBabbler.AppHost.csproj
+++ b/prompt-babbler-service/src/Orchestration/AppHost/PromptBabbler.AppHost.csproj
@@ -10,6 +10,7 @@
     <PackageReference Include="Aspire.Hosting.JavaScript" />
   </ItemGroup>
   <ItemGroup>
+    <PackageReference Include="Microsoft.Azure.Cosmos" />
     <ProjectReference Include="..\..\Api\PromptBabbler.Api.csproj" />
   </ItemGroup>
 </Project>