From d04800c329c380697aad7ae5b04620463c04f21e Mon Sep 17 00:00:00 2001
From: Peter Tripp <peter@zed.dev>
Date: Fri, 31 Jan 2025 15:48:55 -0500
Subject: [PATCH] Add OpenAI o3-mini support (#24044)

Release Notes:

- Add support for OpenAI o3-mini
---
 crates/language_model/src/model/cloud_model.rs | 1 +
 crates/open_ai/src/open_ai.rs                  | 5 +++++
 2 files changed, 6 insertions(+)

diff --git a/crates/language_model/src/model/cloud_model.rs b/crates/language_model/src/model/cloud_model.rs
index c922c6e919..ec51f1f073 100644
--- a/crates/language_model/src/model/cloud_model.rs
+++ b/crates/language_model/src/model/cloud_model.rs
@@ -82,6 +82,7 @@ impl CloudModel {
                 | open_ai::Model::O1Mini
                 | open_ai::Model::O1Preview
                 | open_ai::Model::O1
+                | open_ai::Model::O3Mini
                 | open_ai::Model::Custom { .. } => {
                     LanguageModelAvailability::RequiresPlan(Plan::ZedPro)
                 }
diff --git a/crates/open_ai/src/open_ai.rs b/crates/open_ai/src/open_ai.rs
index e75f3820c8..f6ce348352 100644
--- a/crates/open_ai/src/open_ai.rs
+++ b/crates/open_ai/src/open_ai.rs
@@ -78,6 +78,8 @@ pub enum Model {
     O1Preview,
     #[serde(rename = "o1-mini", alias = "o1-mini")]
     O1Mini,
+    #[serde(rename = "o3-mini", alias = "o3-mini")]
+    O3Mini,
 
     #[serde(rename = "custom")]
     Custom {
@@ -115,6 +117,7 @@ impl Model {
             Self::O1 => "o1",
             Self::O1Preview => "o1-preview",
             Self::O1Mini => "o1-mini",
+            Self::O3Mini => "o3-mini",
             Self::Custom { name, .. } => name,
         }
     }
@@ -129,6 +132,7 @@ impl Model {
             Self::O1 => "o1",
             Self::O1Preview => "o1-preview",
             Self::O1Mini => "o1-mini",
+            Self::O3Mini => "o3-mini",
             Self::Custom {
                 name, display_name, ..
             } => display_name.as_ref().unwrap_or(name),
@@ -145,6 +149,7 @@ impl Model {
             Self::O1 => 200000,
             Self::O1Preview => 128000,
             Self::O1Mini => 128000,
+            Self::O3Mini => 200000,
             Self::Custom { max_tokens, .. } => *max_tokens,
         }
     }