From 8193bad5b59af6eddcb2f9718646604736b4ed10 Mon Sep 17 00:00:00 2001 From: Lily Delalande <119957291+lily-de@users.noreply.github.com> Date: Fri, 25 Apr 2025 16:06:39 -0400 Subject: [PATCH] Add context limits for latest gemini and gpt models (#2359) --- crates/goose/src/model.rs | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/crates/goose/src/model.rs b/crates/goose/src/model.rs index e39f127c..6e976066 100644 --- a/crates/goose/src/model.rs +++ b/crates/goose/src/model.rs @@ -78,10 +78,16 @@ impl ModelConfig { name if name.contains("o1-mini") || name.contains("o1-preview") => Some(128_000), name if name.contains("o1") => Some(200_000), name if name.contains("o3-mini") => Some(200_000), + name if name.contains("gpt-4.1") => Some(1_000_000), + name if name.contains("gpt-4-1") => Some(1_000_000), // Anthropic models, https://docs.anthropic.com/en/docs/about-claude/models name if name.contains("claude-3") => Some(200_000), + // Google models, https://ai.google/get-started/our-models/ + name if name.contains("gemini-2.5") => Some(1_000_000), + name if name.contains("gemini-2-5") => Some(1_000_000), + // Meta Llama models, https://github.com/meta-llama/llama-models/tree/main?tab=readme-ov-file#llama-models-1 name if name.contains("llama3.2") => Some(128_000), name if name.contains("llama3.3") => Some(128_000),