From 58cf6b48f88d9ebd52ff27d9b091df473f139fe2 Mon Sep 17 00:00:00 2001
From: Yingjie He <yingjiehe@squareup.com>
Date: Mon, 19 May 2025 11:57:36 -0700
Subject: [PATCH] fix: update o3/mini and o4 mini context limit (#2590)

---
 crates/goose/src/model.rs | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/crates/goose/src/model.rs b/crates/goose/src/model.rs
index 75829d4b3f..883194c1c2 100644
--- a/crates/goose/src/model.rs
+++ b/crates/goose/src/model.rs
@@ -14,10 +14,9 @@ static MODEL_SPECIFIC_LIMITS: Lazy<HashMap<&'static str, usize>> = Lazy::new(||
     // OpenAI models, https://platform.openai.com/docs/models#models-overview
     map.insert("gpt-4o", 128_000);
     map.insert("gpt-4-turbo", 128_000);
-    map.insert("o1-mini", 128_000);
-    map.insert("o1-preview", 128_000);
-    map.insert("o1", 200_000);
+    map.insert("o3", 200_000);
     map.insert("o3-mini", 200_000);
+    map.insert("o4-mini", 200_000);
     map.insert("gpt-4.1", 1_000_000);
     map.insert("gpt-4-1", 1_000_000);