From c825a2f7436a55017f769ba2e7ac614a253709db Mon Sep 17 00:00:00 2001
From: Peter Goldstein <peter.goldstein@hearst.com>
Date: Tue, 6 Aug 2024 03:23:28 -0400
Subject: [PATCH] Add gpt-4o-mini to Azure. Add 2024-06-01 API version for
Azure (#1102)
---
.changeset/cuddly-kiwis-bathe.md | 5 +++++
packages/llamaindex/src/llm/azure.ts | 6 ++++++
2 files changed, 11 insertions(+)
create mode 100644 .changeset/cuddly-kiwis-bathe.md
diff --git a/.changeset/cuddly-kiwis-bathe.md b/.changeset/cuddly-kiwis-bathe.md
new file mode 100644
index 000000000..4fe10a636
--- /dev/null
+++ b/.changeset/cuddly-kiwis-bathe.md
@@ -0,0 +1,5 @@
+---
+"llamaindex": patch
+---
+
+Add gpt-4o-mini to Azure. Add 2024-06-01 API version for Azure
diff --git a/packages/llamaindex/src/llm/azure.ts b/packages/llamaindex/src/llm/azure.ts
index bfbfb339f..a91e9c3cc 100644
--- a/packages/llamaindex/src/llm/azure.ts
+++ b/packages/llamaindex/src/llm/azure.ts
@@ -18,6 +18,7 @@ const ALL_AZURE_OPENAI_CHAT_MODELS = {
openAIModel: "gpt-3.5-turbo-16k",
},
"gpt-4o": { contextWindow: 128000, openAIModel: "gpt-4o" },
+ "gpt-4o-mini": { contextWindow: 128000, openAIModel: "gpt-4o-mini" },
"gpt-4": { contextWindow: 8192, openAIModel: "gpt-4" },
"gpt-4-32k": { contextWindow: 32768, openAIModel: "gpt-4-32k" },
"gpt-4-turbo": {
@@ -40,6 +41,10 @@ const ALL_AZURE_OPENAI_CHAT_MODELS = {
contextWindow: 128000,
openAIModel: "gpt-4o-2024-05-13",
},
+ "gpt-4o-mini-2024-07-18": {
+ contextWindow: 128000,
+ openAIModel: "gpt-4o-mini-2024-07-18",
+ },
};
const ALL_AZURE_OPENAI_EMBEDDING_MODELS = {
@@ -73,6 +78,7 @@ const ALL_AZURE_API_VERSIONS = [
"2024-03-01-preview",
"2024-04-01-preview",
"2024-05-01-preview",
+ "2024-06-01",
];
const DEFAULT_API_VERSION = "2023-05-15";
--
GitLab