From 6ff7576eb966b82f8b2bc48fe955e1035791b17f Mon Sep 17 00:00:00 2001
From: Parham Saidi <parham@parha.me>
Date: Thu, 23 May 2024 18:22:25 +0200
Subject: [PATCH] feature: added the latest gpt-4o to azure (#875)

---
 .changeset/fluffy-papayas-promise.md | 5 +++++
 packages/core/src/llm/azure.ts       | 3 +++
 2 files changed, 8 insertions(+)
 create mode 100644 .changeset/fluffy-papayas-promise.md

diff --git a/.changeset/fluffy-papayas-promise.md b/.changeset/fluffy-papayas-promise.md
new file mode 100644
index 000000000..0bd2f8cc7
--- /dev/null
+++ b/.changeset/fluffy-papayas-promise.md
@@ -0,0 +1,5 @@
+---
+"llamaindex": patch
+---
+
+Added GPT-4o for Azure
diff --git a/packages/core/src/llm/azure.ts b/packages/core/src/llm/azure.ts
index 81caf19a0..df214ea53 100644
--- a/packages/core/src/llm/azure.ts
+++ b/packages/core/src/llm/azure.ts
@@ -17,6 +17,7 @@ const ALL_AZURE_OPENAI_CHAT_MODELS = {
     contextWindow: 16384,
     openAIModel: "gpt-3.5-turbo-16k",
   },
+  "gpt-4o": { contextWindow: 128000, openAIModel: "gpt-4o" },
   "gpt-4": { contextWindow: 8192, openAIModel: "gpt-4" },
   "gpt-4-32k": { contextWindow: 32768, openAIModel: "gpt-4-32k" },
   "gpt-4-turbo": {
@@ -62,6 +63,8 @@ const ALL_AZURE_API_VERSIONS = [
   "2024-02-01",
   "2024-02-15-preview",
   "2024-03-01-preview",
+  "2024-04-01-preview", // Contains gpt-4o
+  "2024-05-13", // Gives 404 for gpt-4o at time of writting this
 ];
 
 const DEFAULT_API_VERSION = "2023-05-15";
-- 
GitLab