From 4c9b9f0d5da44ad182baf5035a8b45ba533db452 Mon Sep 17 00:00:00 2001
From: gnanam1990 <gnanasekaran.sekareee@gmail.com>
Date: Thu, 2 Apr 2026 01:32:30 +0530
Subject: [PATCH] fix: support Azure Cognitive Services and Azure OpenAI
 endpoints

Azure endpoints require two changes vs standard OpenAI:
1. Auth header: `api-key: {key}` instead of `Authorization: Bearer {key}`
2. URL path: `/openai/deployments/{model}/chat/completions?api-version={version}`
   instead of `/chat/completions`

Detection is automatic when OPENAI_BASE_URL contains
`cognitiveservices.azure.com` or `openai.azure.com`.

The api-version defaults to `2024-12-01-preview` and can be overridden
via the AZURE_OPENAI_API_VERSION env var.

Handles all common Azure base URL formats:
- https://{resource}.cognitiveservices.azure.com/
- https://{resource}.cognitiveservices.azure.com/openai/v1
- https://{resource}.openai.azure.com/openai/v1
- https://{resource}.cognitiveservices.azure.com/openai/deployments/{model}/v1

Fixes #79

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
---
 src/services/api/openaiShim.ts | 33 +++++++++++++++++++++++++++++++--
 1 file changed, 31 insertions(+), 2 deletions(-)

diff --git a/src/services/api/openaiShim.ts b/src/services/api/openaiShim.ts
index 4b70a63c..40b5cfaa 100644
--- a/src/services/api/openaiShim.ts
+++ b/src/services/api/openaiShim.ts
@@ -682,11 +682,40 @@ class OpenAIShimMessages {
     }
 
     const apiKey = process.env.OPENAI_API_KEY ?? ''
+    const isAzure = /cognitiveservices\.azure\.com|openai\.azure\.com/.test(request.baseUrl)
+
     if (apiKey) {
-      headers.Authorization = `Bearer ${apiKey}`
+      if (isAzure) {
+        // Azure uses api-key header instead of Bearer token
+        headers['api-key'] = apiKey
+      } else {
+        headers.Authorization = `Bearer ${apiKey}`
+      }
     }
 
-    const response = await fetch(`${request.baseUrl}/chat/completions`, {
+    // Build the chat completions URL
+    // Azure Cognitive Services / Azure OpenAI require a deployment-specific path
+    // and an api-version query parameter.
+    // Standard format: {base}/openai/deployments/{model}/chat/completions?api-version={version}
+    // Non-Azure: {base}/chat/completions
+    let chatCompletionsUrl: string
+    if (isAzure) {
+      const apiVersion = process.env.AZURE_OPENAI_API_VERSION ?? '2024-12-01-preview'
+      const deployment = request.resolvedModel ?? process.env.OPENAI_MODEL ?? 'gpt-4o'
+      // If base URL already contains /deployments/, use it as-is with api-version
+      if (/\/deployments\//i.test(request.baseUrl)) {
+        const base = request.baseUrl.replace(/\/+$/, '')
+        chatCompletionsUrl = `${base}/chat/completions?api-version=${apiVersion}`
+      } else {
+        // Strip trailing /v1 or /openai/v1 if present, then build Azure path
+        const base = request.baseUrl.replace(/\/(openai\/)?v1\/?$/, '').replace(/\/+$/, '')
+        chatCompletionsUrl = `${base}/openai/deployments/${deployment}/chat/completions?api-version=${apiVersion}`
+      }
+    } else {
+      chatCompletionsUrl = `${request.baseUrl}/chat/completions`
+    }
+
+    const response = await fetch(chatCompletionsUrl, {
       method: 'POST',
       headers,
       body: JSON.stringify(body),