From f080a41387cb332f2a2962c0fa1bf9099e2d581d Mon Sep 17 00:00:00 2001 From: Salman Mohammed Date: Tue, 27 May 2025 22:28:57 -0400 Subject: [PATCH] feat: add context window limit for claude4 (#2689) --- crates/goose-llm/src/model.rs | 1 + crates/goose/src/model.rs | 1 + 2 files changed, 2 insertions(+) diff --git a/crates/goose-llm/src/model.rs b/crates/goose-llm/src/model.rs index ca9ecdad..5ad89ab2 100644 --- a/crates/goose-llm/src/model.rs +++ b/crates/goose-llm/src/model.rs @@ -43,6 +43,7 @@ impl ModelConfig { // Anthropic models, https://docs.anthropic.com/en/docs/about-claude/models name if name.contains("claude-3") => Some(200_000), + name if name.contains("claude-4") => Some(200_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), diff --git a/crates/goose/src/model.rs b/crates/goose/src/model.rs index 883194c1..091745df 100644 --- a/crates/goose/src/model.rs +++ b/crates/goose/src/model.rs @@ -22,6 +22,7 @@ static MODEL_SPECIFIC_LIMITS: Lazy> = Lazy::new(|| // Anthropic models, https://docs.anthropic.com/en/docs/about-claude/models map.insert("claude-3", 200_000); + map.insert("claude-4", 200_000); // Google models, https://ai.google/get-started/our-models/ map.insert("gemini-2.5", 1_000_000);