From 058c275a7219810151486fb0aac78801d62c391d Mon Sep 17 00:00:00 2001
From: Yi Ding <yi.s.ding@gmail.com>
Date: Thu, 23 May 2024 09:27:04 -0700
Subject: [PATCH] New azure versions (#877)

---
 packages/core/src/llm/azure.ts | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/packages/core/src/llm/azure.ts b/packages/core/src/llm/azure.ts
index df214ea53..964187463 100644
--- a/packages/core/src/llm/azure.ts
+++ b/packages/core/src/llm/azure.ts
@@ -24,6 +24,10 @@ const ALL_AZURE_OPENAI_CHAT_MODELS = {
     contextWindow: 128000,
     openAIModel: "gpt-4-turbo",
   },
+  "gpt-4-turbo-2024-04-09": {
+    contextWindow: 128000,
+    openAIModel: "gpt-4-turbo",
+  },
   "gpt-4-vision-preview": {
     contextWindow: 128000,
     openAIModel: "gpt-4-vision-preview",
@@ -32,6 +36,10 @@ const ALL_AZURE_OPENAI_CHAT_MODELS = {
     contextWindow: 128000,
     openAIModel: "gpt-4-1106-preview",
   },
+  "gpt-4o-2024-05-13": {
+    contextWindow: 128000,
+    openAIModel: "gpt-4o-2024-05-13",
+  },
 };
 
 const ALL_AZURE_OPENAI_EMBEDDING_MODELS = {
@@ -63,8 +71,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
+  "2024-04-01-preview",
+  "2024-05-01-preview",
 ];
 
 const DEFAULT_API_VERSION = "2023-05-15";
-- 
GitLab