From 5e30e190b891c413eb7825ae695bd2b914a46663 Mon Sep 17 00:00:00 2001 From: Kenny Dizi Date: Mon, 3 Mar 2025 18:22:31 +0700 Subject: [PATCH] Define models that support extended thinking feature --- pr_agent/algo/__init__.py | 6 ++++++ 1 file changed, 6 insertions(+) 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" +]