File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -205,7 +205,7 @@ def supported_models(cls) -> list[str]:
205205 """
206206
207207 return [
208- r'gemma-3 .*' ,
208+ r'gemma-.*' ,
209209 ]
210210
211211 @cached_property
Original file line number Diff line number Diff line change 1414
1515from google .adk import models
1616from google .adk .models .anthropic_llm import Claude
17+ from google .adk .models .gemma_llm import Gemma
1718from google .adk .models .google_llm import Gemini
1819from google .adk .models .lite_llm import LiteLlm
1920import pytest
@@ -56,6 +57,19 @@ def test_match_claude_family(model_name):
5657 assert models .LLMRegistry .resolve (model_name ) is Claude
5758
5859
60+ @pytest .mark .parametrize (
61+ 'model_name' ,
62+ [
63+ 'gemma-3-27b-it' ,
64+ 'gemma-4-27b-it' ,
65+ 'gemma-4-31b-it' ,
66+ ],
67+ )
68+ def test_match_gemma_family (model_name ):
69+ """Test that Gemma models are resolved correctly."""
70+ assert models .LLMRegistry .resolve (model_name ) is Gemma
71+
72+
5973@pytest .mark .parametrize (
6074 'model_name' ,
6175 [
You can’t perform that action at this time.
0 commit comments