From ecdc37c8bfa77b5f7e13896e2220370a11cc5784 Mon Sep 17 00:00:00 2001
From: Rabi Mishra <ramishra@redhat.com>
Date: Tue, 27 Jan 2026 05:08:58 +0530
Subject: [PATCH] fix: update Gemini context limits (#6536)

Signed-off-by: rabi <ramishra@redhat.com>
---
 crates/goose/src/model.rs | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/crates/goose/src/model.rs b/crates/goose/src/model.rs
index bc9b73c9b1..4adfa7d6ab 100644
--- a/crates/goose/src/model.rs
+++ b/crates/goose/src/model.rs
@@ -61,9 +61,12 @@ static MODEL_SPECIFIC_LIMITS: Lazy<Vec<(&'static str, usize)>> = Lazy::new(|| {
         // anthropic - all 200k
         ("claude", 200_000),
         // google
-        ("gemini-1.5-flash", 1_000_000),
+        ("gemini-1.5-flash", 1_048_576),
         ("gemini-1", 128_000),
-        ("gemini-2", 1_000_000),
+        ("gemini-2", 1_048_576),
+        ("gemini-3-pro-image", 65_536),
+        ("gemini-3-pro", 1_048_576),
+        ("gemini-3-flash", 1_048_576),
         ("gemma-3-27b", 128_000),
         ("gemma-3-12b", 128_000),
         ("gemma-3-4b", 128_000),