From 64c59af2e2819a01c3f2d06d805b1e00c2381f00 Mon Sep 17 00:00:00 2001 From: Ariel Date: Fri, 28 Feb 2025 03:07:06 +0800 Subject: [PATCH] feat(model): add context window limit for OpenAI reasoning models (#1423) --- crates/goose/src/model.rs | 3 +++ 1 file changed, 3 insertions(+) diff --git a/crates/goose/src/model.rs b/crates/goose/src/model.rs index a35d22238b..5c4aa09685 100644 --- a/crates/goose/src/model.rs +++ b/crates/goose/src/model.rs @@ -59,6 +59,9 @@ impl ModelConfig { // OpenAI models, https://platform.openai.com/docs/models#models-overview name if name.contains("gpt-4o") => Some(128_000), name if name.contains("gpt-4-turbo") => Some(128_000), + 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), // Anthropic models, https://docs.anthropic.com/en/docs/about-claude/models name if name.contains("claude-3") => Some(200_000),