diff --git a/google/genai/_local_tokenizer_loader.py b/google/genai/_local_tokenizer_loader.py index 0f6edc3ad..ee6316032 100644 --- a/google/genai/_local_tokenizer_loader.py +++ b/google/genai/_local_tokenizer_loader.py @@ -45,6 +45,7 @@ "gemini-2.0-flash-001": "gemma3", "gemini-2.0-flash-lite-001": "gemma3", "gemini-3-pro-preview": "gemma3", + "gemini-3-flash-preview": "gemma3", } diff --git a/google/genai/tests/local_tokenizer/test_local_tokenizer.py b/google/genai/tests/local_tokenizer/test_local_tokenizer.py index ca337b5e8..ea425ddea 100644 --- a/google/genai/tests/local_tokenizer/test_local_tokenizer.py +++ b/google/genai/tests/local_tokenizer/test_local_tokenizer.py @@ -37,7 +37,7 @@ def setUp(self): self.mock_tokenizer = MagicMock() self.mock_get_sentencepiece.return_value = self.mock_tokenizer - self.tokenizer = local_tokenizer.LocalTokenizer(model_name='gemini-3-pro-preview') + self.tokenizer = local_tokenizer.LocalTokenizer(model_name='gemini-3-flash-preview') def tearDown(self): patch.stopall()