Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

openai: Add back O1-Preview (#23715)

Browse source 
Follow-up to: https://github.com/zed-industries/zed/pull/23425
This commit is contained in:
Peter Tripp 2025-01-27 09:44:12 -05:00 • committed by GitHub
parent 5bde053b0d
commit f6d286c7db
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 9 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

1
crates/language_model/src/model/cloud_model.rs
View file

@ -80,6 +80,7 @@ impl CloudModel {
| open_ai::Model::FourOmni | open_ai::Model::FourOmni
| open_ai::Model::FourOmniMini | open_ai::Model::FourOmniMini
| open_ai::Model::O1Mini | open_ai::Model::O1Mini
| open_ai::Model::O1Preview
| open_ai::Model::O1 | open_ai::Model::O1
| open_ai::Model::Custom { .. } => { | open_ai::Model::Custom { .. } => {
LanguageModelAvailability::RequiresPlan(Plan::ZedPro) LanguageModelAvailability::RequiresPlan(Plan::ZedPro)

10
crates/open_ai/src/open_ai.rs
View file

@ -72,8 +72,10 @@ pub enum Model {
FourOmni, FourOmni,
#[serde(rename = "gpt-4o-mini", alias = "gpt-4o-mini")] #[serde(rename = "gpt-4o-mini", alias = "gpt-4o-mini")]
FourOmniMini, FourOmniMini,
#[serde(rename = "o1", alias = "o1-preview")] #[serde(rename = "o1", alias = "o1")]
O1, O1,
#[serde(rename = "o1-preview", alias = "o1-preview")]
O1Preview,
#[serde(rename = "o1-mini", alias = "o1-mini")] #[serde(rename = "o1-mini", alias = "o1-mini")]
O1Mini, O1Mini,
@ -97,6 +99,7 @@ impl Model {
"gpt-4o" => Ok(Self::FourOmni), "gpt-4o" => Ok(Self::FourOmni),
"gpt-4o-mini" => Ok(Self::FourOmniMini), "gpt-4o-mini" => Ok(Self::FourOmniMini),
"o1" => Ok(Self::O1), "o1" => Ok(Self::O1),
"o1-preview" => Ok(Self::O1Preview),
"o1-mini" => Ok(Self::O1Mini), "o1-mini" => Ok(Self::O1Mini),
_ => Err(anyhow!("invalid model id")), _ => Err(anyhow!("invalid model id")),
} }
@ -110,6 +113,7 @@ impl Model {
Self::FourOmni => "gpt-4o", Self::FourOmni => "gpt-4o",
Self::FourOmniMini => "gpt-4o-mini", Self::FourOmniMini => "gpt-4o-mini",
Self::O1 => "o1", Self::O1 => "o1",
Self::O1Preview => "o1-preview",
Self::O1Mini => "o1-mini", Self::O1Mini => "o1-mini",
Self::Custom { name, .. } => name, Self::Custom { name, .. } => name,
} }
@ -123,6 +127,7 @@ impl Model {
Self::FourOmni => "gpt-4o", Self::FourOmni => "gpt-4o",
Self::FourOmniMini => "gpt-4o-mini", Self::FourOmniMini => "gpt-4o-mini",
Self::O1 => "o1", Self::O1 => "o1",
Self::O1Preview => "o1-preview",
Self::O1Mini => "o1-mini", Self::O1Mini => "o1-mini",
Self::Custom { Self::Custom {
name, display_name, .. name, display_name, ..
@ -137,7 +142,8 @@ impl Model {
Self::FourTurbo => 128000, Self::FourTurbo => 128000,
Self::FourOmni => 128000, Self::FourOmni => 128000,
Self::FourOmniMini => 128000, Self::FourOmniMini => 128000,
Self::O1 => 128000, Self::O1 => 200000,
Self::O1Preview => 128000,
Self::O1Mini => 128000, Self::O1Mini => 128000,
Self::Custom { max_tokens, .. } => *max_tokens, Self::Custom { max_tokens, .. } => *max_tokens,
} }