From a2de7d48be7c9e662b01f5bd5e7b68b93ec46b67 Mon Sep 17 00:00:00 2001
From: SuYao <sy20010504@gmail.com>
Date: Wed, 26 Nov 2025 15:43:32 +0800
Subject: [PATCH] fix: update Azure provider handling in AI SDK integration
 (#11465)

---
 src/renderer/src/aiCore/provider/factory.ts | 8 ++++++--
 src/renderer/src/aiCore/utils/options.ts    | 4 +++-
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/renderer/src/aiCore/provider/factory.ts b/src/renderer/src/aiCore/provider/factory.ts
index 876f3acd97..9760839389 100644
--- a/src/renderer/src/aiCore/provider/factory.ts
+++ b/src/renderer/src/aiCore/provider/factory.ts
@@ -60,8 +60,12 @@ function tryResolveProviderId(identifier: string): ProviderId | null {
 export function getAiSdkProviderId(provider: Provider): string {
   // 1. 尝试解析provider.id
   const resolvedFromId = tryResolveProviderId(provider.id)
-  if (isAzureOpenAIProvider(provider) && isAzureResponsesEndpoint(provider)) {
-    return 'azure-responses'
+  if (isAzureOpenAIProvider(provider)) {
+    if (isAzureResponsesEndpoint(provider)) {
+      return 'azure-responses'
+    } else {
+      return 'azure'
+    }
   }
   if (resolvedFromId) {
     return resolvedFromId
diff --git a/src/renderer/src/aiCore/utils/options.ts b/src/renderer/src/aiCore/utils/options.ts
index f420908ba6..f1fc61aacd 100644
--- a/src/renderer/src/aiCore/utils/options.ts
+++ b/src/renderer/src/aiCore/utils/options.ts
@@ -249,7 +249,9 @@ export function buildProviderOptions(
       'google-vertex': 'google',
       'google-vertex-anthropic': 'anthropic',
       'azure-anthropic': 'anthropic',
-      'ai-gateway': 'gateway'
+      'ai-gateway': 'gateway',
+      azure: 'openai',
+      'azure-responses': 'openai'
     }[rawProviderId] || rawProviderId
 
   if (rawProviderKey === 'cherryin') {