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);