From 6331b8360a1b05d86a94f0c4fc32fe4a4ceb47c4 Mon Sep 17 00:00:00 2001
From: sinedied <noda@free.fr>
Date: Wed, 15 Nov 2023 09:54:03 +0100
Subject: [PATCH] Fix authentication issue when using managed identity on Azure

---
 src/model.ts | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/model.ts b/src/model.ts
index 6567a064..cb909e7e 100644
--- a/src/model.ts
+++ b/src/model.ts
@@ -97,7 +97,14 @@ export function createOpenAILanguageModel(apiKey: string, model: string, endPoin
  * @returns An instance of `TypeChatLanguageModel`.
  */
 export function createAzureOpenAILanguageModel(apiKey: string, endPoint: string,): TypeChatLanguageModel {
-    return createAxiosLanguageModel(endPoint, { headers: { "api-key": apiKey } }, {});
+    return createAxiosLanguageModel(endPoint, {
+        headers: {
+            // Needed when using managed identity
+            Authorization: `Bearer ${apiKey}`,
+            // Needed when using regular API key
+            "api-key": apiKey
+        }
+    }, {});
 }
 
 /**