diff --git a/crates/copilot/src/copilot_chat.rs b/crates/copilot/src/copilot_chat.rs
index 14e1a4b2103416f7441369654248dba4ec18cb28..9bab354091d1c6106093719d1463d9a32cff99f0 100644
--- a/crates/copilot/src/copilot_chat.rs
+++ b/crates/copilot/src/copilot_chat.rs
@@ -91,12 +91,12 @@ impl Model {
 
     pub fn max_token_count(&self) -> usize {
         match self {
-            Self::Gpt4o => 64000,
-            Self::Gpt4 => 32768,
-            Self::Gpt3_5Turbo => 12288,
-            Self::O3Mini => 20000,
-            Self::O1 => 20000,
-            Self::Claude3_5Sonnet => 200_000,
+            Self::Gpt4o => 64_000,
+            Self::Gpt4 => 32_768,
+            Self::Gpt3_5Turbo => 12_288,
+            Self::O3Mini => 64_000,
+            Self::O1 => 20_000,
+            Self::Claude3_5Sonnet => 128_000,
             Model::Gemini20Flash => 128_000,
         }
     }
diff --git a/crates/open_ai/src/open_ai.rs b/crates/open_ai/src/open_ai.rs
index b200da8ce0a534faf877b9bd19fb6f3083ba7f3c..c62b1c2950deab2c12a3e1cac8bdfe422a07e7d2 100644
--- a/crates/open_ai/src/open_ai.rs
+++ b/crates/open_ai/src/open_ai.rs
@@ -142,15 +142,15 @@ impl Model {
 
     pub fn max_token_count(&self) -> usize {
         match self {
-            Self::ThreePointFiveTurbo => 16385,
-            Self::Four => 8192,
-            Self::FourTurbo => 128000,
-            Self::FourOmni => 128000,
-            Self::FourOmniMini => 128000,
-            Self::O1 => 200000,
-            Self::O1Preview => 128000,
-            Self::O1Mini => 128000,
-            Self::O3Mini => 200000,
+            Self::ThreePointFiveTurbo => 16_385,
+            Self::Four => 8_192,
+            Self::FourTurbo => 128_000,
+            Self::FourOmni => 128_000,
+            Self::FourOmniMini => 128_000,
+            Self::O1 => 200_000,
+            Self::O1Preview => 128_000,
+            Self::O1Mini => 128_000,
+            Self::O3Mini => 200_000,
             Self::Custom { max_tokens, .. } => *max_tokens,
         }
     }