diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py index 34eac5e1..72804dfc 100644 --- a/pr_agent/algo/__init__.py +++ b/pr_agent/algo/__init__.py @@ -60,6 +60,7 @@ MAX_TOKENS = { 'anthropic/claude-3-5-sonnet-20240620': 100000, 'anthropic/claude-3-5-sonnet-20241022': 100000, 'anthropic/claude-3-7-sonnet-20250219': 200000, + 'claude-3-7-sonnet-20250219': 200000, 'anthropic/claude-3-5-haiku-20241022': 100000, 'bedrock/anthropic.claude-instant-v1': 100000, 'bedrock/anthropic.claude-v2': 100000, @@ -110,3 +111,8 @@ SUPPORT_REASONING_EFFORT_MODELS = [ "o3-mini", "o3-mini-2025-01-31" ] + +CLAUDE_EXTENDED_THINKING_MODELS = [ + "anthropic/claude-3-7-sonnet-20250219", + "claude-3-7-sonnet-20250219" +]