chore: Adding regex for claude-4 models to anthropic_llm and updating tests

avidelatm · copybara-github · commit 7fc09b2c642c · 2025-05-28T13:39:04.000-07:00
Copybara import of the project: -- 824b437 by Andres Videla <andres.videla@trademe.co.nz>: Adding regex for claude-4 models to anthropic_llm and updating tests -- 8fa2a2d by Wei Sun (Jack) <Jacksunwei@gmail.com>: Adds trailing comma. COPYBARA_INTEGRATE_REVIEW=google#934 from avidelatm:feature/add-support-for-claude-4-models-to-anthropic_llm 8d10bac PiperOrigin-RevId: 764396694
diff --git a/src/google/adk/models/anthropic_llm.py b/src/google/adk/models/anthropic_llm.py
@@ -208,7 +208,7 @@ class Claude(BaseLlm):
   @staticmethod
   @override
   def supported_models() -> list[str]:
-    return [r"claude-3-.*"]
+    return [r"claude-3-.*", r"claude-.*-4.*"]
 
   @override
   async def generate_content_async(
diff --git a/tests/unittests/models/test_models.py b/tests/unittests/models/test_models.py
@@ -46,6 +46,8 @@ def test_match_gemini_family(model_name):
         'claude-3-haiku@20240307',
         'claude-3-opus@20240229',
         'claude-3-sonnet@20240229',
+        'claude-sonnet-4@20250514',
+        'claude-opus-4@20250514',
     ],
 )
 def test_match_claude_family(model_name):

Original file line number	Diff line number	Diff line change
`@@ -46,6 +46,8 @@ def test_match_gemini_family(model_name):`
`46`	`46`	`'claude-3-haiku@20240307',`
`47`	`47`	`'claude-3-opus@20240229',`
`48`	`48`	`'claude-3-sonnet@20240229',`
	`49`	`+ 'claude-sonnet-4@20250514',`
	`50`	`+ 'claude-opus-4@20250514',`
`49`	`51`	`],`
`50`	`52`	`)`
`51`	`53`	`def test_match_claude_family(model_name):`