diff --git a/DocGpt.Roslyn/DocGpt.Roslyn.csproj b/DocGpt.Roslyn/DocGpt.Roslyn.csproj
index 37957d7..96d29e2 100644
--- a/DocGpt.Roslyn/DocGpt.Roslyn.csproj
+++ b/DocGpt.Roslyn/DocGpt.Roslyn.csproj
@@ -19,7 +19,7 @@
 	</ItemGroup>
 
 	<ItemGroup>
-		<PackageReference Include="Azure.AI.OpenAI" Version="1.0.0-beta.17" />
+		<PackageReference Include="Azure.AI.OpenAI" Version="2.1.0" />
 		<PackageReference Include="Microsoft.CodeAnalysis.CSharp.Workspaces" Version="4.11.0" />
 	</ItemGroup>
 
diff --git a/DocGpt.Vsix/DocGpt.Vsix.csproj b/DocGpt.Vsix/DocGpt.Vsix.csproj
index cec17cb..1f58466 100644
--- a/DocGpt.Vsix/DocGpt.Vsix.csproj
+++ b/DocGpt.Vsix/DocGpt.Vsix.csproj
@@ -92,7 +92,7 @@
   </ItemGroup>
   <ItemGroup>
     <PackageReference Include="Azure.AI.OpenAI">
-      <Version>1.0.0-beta.17</Version>
+      <Version>2.1.0</Version>
     </PackageReference>
     <PackageReference Include="Microsoft.VisualStudio.Shell.15.0">
       <Version>17.11.40262</Version>