chore: update gemini models in unitests

Author: morganroux

tests/unittests/agents/test_agent_config.py

@@ -37,7 +37,7 @@
 def test_agent_config_discriminator_default_is_llm_agent(tmp_path: Path):
   yaml_content = """\
 name: search_agent
-model: gemini-2.0-flash
+model: gemini-2.5-flash
 description: a sample description
 instruction: a fake instruction
 tools:
@@ -67,7 +67,7 @@ def test_agent_config_discriminator_llm_agent(
   yaml_content = f"""\
 agent_class: {agent_class_value}
 name: search_agent
-model: gemini-2.0-flash
+model: gemini-2.5-flash
 description: a sample description
 instruction: a fake instruction
 tools:
@@ -186,7 +186,7 @@ def test_agent_config_discriminator_with_sub_agents(
   sub_agent_dir.mkdir()
   sub_agent_config = """\
 name: sub_agent_{index}
-model: gemini-2.0-flash
+model: gemini-2.5-flash
 description: a sub agent
 instruction: sub agent instruction
 """
@@ -230,7 +230,7 @@ def test_agent_config_discriminator_llm_agent_with_sub_agents(
   sub_agent_dir.mkdir()
   sub_agent_config = """\
 name: sub_agent_{index}
-model: gemini-2.0-flash
+model: gemini-2.5-flash
 description: a sub agent
 instruction: sub agent instruction
 """
@@ -243,7 +243,7 @@ def test_agent_config_discriminator_llm_agent_with_sub_agents(
   yaml_content = f"""\
 agent_class: {agent_class_value}
 name: main_agent
-model: gemini-2.0-flash
+model: gemini-2.5-flash
 description: main agent with sub agents
 instruction: main agent instruction
 sub_agents:
@@ -368,14 +368,14 @@ def test_resolve_agent_reference_resolves_relative_paths(
   child_config_path.write_text(dedent(f"""
           agent_class: LlmAgent
           name: {child_name}
-          model: gemini-2.0-flash
+          model: gemini-2.5-flash
           instruction: {instruction}
           """).lstrip())
 
   config_file.write_text(dedent(f"""
           agent_class: LlmAgent
           name: main_agent
-          model: gemini-2.0-flash
+          model: gemini-2.5-flash
           instruction: I am the main agent
           sub_agents:
             - config_path: {child_rel_path.as_posix()}

tests/unittests/agents/test_gemini_context_cache_manager.py

@@ -75,7 +75,7 @@ def create_llm_request(self, cache_metadata=None, contents_count=3):
     )
 
     return LlmRequest(
-        model="gemini-2.0-flash",
+        model="gemini-2.5-flash",
         contents=contents,
         config=types.GenerateContentConfig(
             system_instruction="Test instruction",
@@ -343,7 +343,7 @@ def test_generate_cache_fingerprint(self):
     # Test that tool_config and tools are included in fingerprint
     # Create request without tools/tool_config
     llm_request_no_tools = LlmRequest(
-        model="gemini-2.0-flash",
+        model="gemini-2.5-flash",
         contents=[types.Content(role="user", parts=[types.Part(text="Test")])],
         config=types.GenerateContentConfig(
             system_instruction="Test instruction"
@@ -363,7 +363,7 @@ def test_generate_cache_fingerprint_different_requests(self):
     llm_request1 = self.create_llm_request()
 
     llm_request2 = LlmRequest(
-        model="gemini-2.0-flash",
+        model="gemini-2.5-flash",
         contents=[
             types.Content(
                 role="user", parts=[types.Part(text="Different message")]
@@ -396,7 +396,7 @@ def test_generate_cache_fingerprint_tool_config_variations(self):
     )
 
     llm_request_none = LlmRequest(
-        model="gemini-2.0-flash",
+        model="gemini-2.5-flash",
         contents=[types.Content(role="user", parts=[types.Part(text="Test")])],
         config=types.GenerateContentConfig(
             system_instruction="Test instruction",
@@ -507,7 +507,7 @@ def test_edge_cases(self):
     """Test various edge cases."""
     # Test with None cache_config
     llm_request_no_config = LlmRequest(
-        model="gemini-2.0-flash",
+        model="gemini-2.5-flash",
         contents=[types.Content(role="user", parts=[types.Part(text="Test")])],
         config=types.GenerateContentConfig(system_instruction="Test"),
         cache_config=None,
@@ -522,7 +522,7 @@ def test_edge_cases(self):
 
     # Test with empty contents
     llm_request_empty = LlmRequest(
-        model="gemini-2.0-flash",
+        model="gemini-2.5-flash",
         contents=[],
         config=types.GenerateContentConfig(system_instruction="Test"),
         cache_config=self.cache_config,

tests/unittests/agents/test_llm_agent_error_messages.py

@@ -22,10 +22,10 @@ def test_agent_not_found_enhanced_error():
   """Verify enhanced error message for agent not found."""
   root_agent = LlmAgent(
       name='root',
-      model='gemini-2.0-flash',
+      model='gemini-2.5-flash',
       sub_agents=[
-          LlmAgent(name='agent_a', model='gemini-2.0-flash'),
-          LlmAgent(name='agent_b', model='gemini-2.0-flash'),
+          LlmAgent(name='agent_a', model='gemini-2.5-flash'),
+          LlmAgent(name='agent_b', model='gemini-2.5-flash'),
       ],
   )
 
@@ -47,13 +47,13 @@ def test_agent_tree_traversal():
   """Verify agent tree traversal helper works correctly."""
   root_agent = LlmAgent(
       name='orchestrator',
-      model='gemini-2.0-flash',
+      model='gemini-2.5-flash',
       sub_agents=[
           LlmAgent(
               name='parent_agent',
-              model='gemini-2.0-flash',
+              model='gemini-2.5-flash',
               sub_agents=[
-                  LlmAgent(name='child_agent', model='gemini-2.0-flash'),
+                  LlmAgent(name='child_agent', model='gemini-2.5-flash'),
               ],
           ),
       ],
@@ -72,11 +72,11 @@ def test_agent_not_found_shows_all_agents():
   """Verify error message shows all agents (no truncation)."""
   # Create 100 sub-agents
   sub_agents = [
-      LlmAgent(name=f'agent_{i}', model='gemini-2.0-flash') for i in range(100)
+      LlmAgent(name=f'agent_{i}', model='gemini-2.5-flash') for i in range(100)
   ]
 
   root_agent = LlmAgent(
-      name='root', model='gemini-2.0-flash', sub_agents=sub_agents
+      name='root', model='gemini-2.5-flash', sub_agents=sub_agents
   )
 
   with pytest.raises(ValueError) as exc_info:

tests/unittests/agents/test_llm_agent_fields.py

@@ -58,7 +58,7 @@ async def _create_readonly_context(
     ('default_model', 'expected_model_name', 'expected_model_type'),
     [
         (LlmAgent.DEFAULT_MODEL, LlmAgent.DEFAULT_MODEL, Gemini),
-        ('gemini-2.0-flash', 'gemini-2.0-flash', Gemini),
+        ('gemini-2.5-flash', 'gemini-2.5-flash', Gemini),
     ],
 )
 def test_canonical_model_default_fallback(
@@ -478,7 +478,7 @@ def _tool_2():
     'model_name',
     [
         'gemini-1.5-flash',
-        'gemini-2.0-flash-exp',
+        'gemini-2.5-flash-exp',
     ],
 )
 def test_agent_with_gemini_string_model(model_name):

tests/unittests/cli/utils/test_agent_loader.py

@@ -516,7 +516,7 @@ def test_load_agent_from_yaml_config(self):
       yaml_content = dedent("""
         agent_class: LlmAgent
         name: yaml_test_agent
-        model: gemini-2.0-flash
+        model: gemini-2.5-flash
         instruction: You are a test agent loaded from YAML configuration.
         description: A test agent created from YAML config
       """)
@@ -533,7 +533,7 @@ def test_load_agent_from_yaml_config(self):
       from google.adk.agents.llm_agent import LlmAgent
 
       if isinstance(agent, LlmAgent):
-        assert agent.model == "gemini-2.0-flash"
+        assert agent.model == "gemini-2.5-flash"
         # Handle instruction which can be string or InstructionProvider
         instruction_text = str(agent.instruction)
         assert "test agent loaded from YAML" in instruction_text
@@ -548,7 +548,7 @@ def test_yaml_agent_caching_returns_same_instance(self):
       yaml_content = dedent("""
         agent_class: LlmAgent
         name: cached_yaml_test_agent
-        model: gemini-2.0-flash
+        model: gemini-2.5-flash
         instruction: You are a cached test agent.
       """)
 
@@ -597,7 +597,7 @@ def test_yaml_agent_invalid_yaml_error(self):
       # Create invalid YAML content with wrong field name
       invalid_yaml_content = dedent("""
         not_exist_field: invalid_yaml_test_agent
-        model: gemini-2.0-flash
+        model: gemini-2.5-flash
         instruction: You are a test agent with invalid YAML
       """)
 
@@ -828,7 +828,7 @@ def test_load_special_agent_from_yaml_config(self):
       yaml_content = dedent("""
         agent_class: LlmAgent
         name: special_yaml_test_agent
-        model: gemini-2.0-flash
+        model: gemini-2.5-flash
         instruction: You are a special test agent loaded from YAML configuration.
         description: A special test agent created from YAML config
       """)
@@ -859,7 +859,7 @@ def test_load_special_agent_from_yaml_config(self):
         from google.adk.agents.llm_agent import LlmAgent
 
         if isinstance(agent, LlmAgent):
-          assert agent.model == "gemini-2.0-flash"
+          assert agent.model == "gemini-2.5-flash"
           # Handle instruction which can be string or InstructionProvider
           instruction_text = str(agent.instruction)
           assert "special test agent loaded from YAML" in instruction_text
@@ -885,7 +885,7 @@ def test_yaml_config_agents_dir_parameter(self):
       regular_yaml_content = dedent("""
         agent_class: LlmAgent
         name: regular_yaml_agent
-        model: gemini-2.0-flash
+        model: gemini-2.5-flash
         instruction: Regular agent from default directory.
       """)
       self.create_yaml_agent_structure(
@@ -896,7 +896,7 @@ def test_yaml_config_agents_dir_parameter(self):
       custom_yaml_content = dedent("""
         agent_class: LlmAgent
         name: custom_yaml_agent
-        model: gemini-2.0-flash
+        model: gemini-2.5-flash
         instruction: Custom agent from custom directory.
       """)
       self.create_yaml_agent_structure(

tests/unittests/cli/utils/test_cli_create.py

@@ -60,7 +60,7 @@ def test_generate_files_with_api_key(agent_folder: Path) -> None:
   cli_create._generate_files(
       str(agent_folder),
       google_api_key="dummy-key",
-      model="gemini-2.0-flash-001",
+      model="gemini-2.5-flash",
       type="code",
   )
 
@@ -77,7 +77,7 @@ def test_generate_files_with_gcp(agent_folder: Path) -> None:
       str(agent_folder),
       google_cloud_project="proj",
       google_cloud_region="us-central1",
-      model="gemini-2.0-flash-001",
+      model="gemini-2.5-flash",
       type="code",
   )
 
@@ -95,7 +95,7 @@ def test_generate_files_overwrite(agent_folder: Path) -> None:
   cli_create._generate_files(
       str(agent_folder),
       google_api_key="new-key",
-      model="gemini-2.0-flash-001",
+      model="gemini-2.5-flash",
       type="code",
   )
 
@@ -111,14 +111,14 @@ def test_generate_files_permission_error(
   )
   with pytest.raises(PermissionError):
     cli_create._generate_files(
-        str(agent_folder), model="gemini-2.0-flash-001", type="code"
+        str(agent_folder), model="gemini-2.5-flash", type="code"
     )
 
 
 def test_generate_files_no_params(agent_folder: Path) -> None:
   """No backend parameters → minimal .env file is generated."""
   cli_create._generate_files(
-      str(agent_folder), model="gemini-2.0-flash-001", type="code"
+      str(agent_folder), model="gemini-2.5-flash", type="code"
   )
 
   env_content = (agent_folder / ".env").read_text()
@@ -147,7 +147,7 @@ def test_run_cmd_overwrite_reject(
   with pytest.raises(click.Abort):
     cli_create.run_cmd(
         agent_name,
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash",
         google_api_key=None,
         google_cloud_project=None,
         google_cloud_region=None,
@@ -164,7 +164,7 @@ def test_run_cmd_invalid_app_name(
   with pytest.raises(click.BadParameter, match="Invalid app name"):
     cli_create.run_cmd(
         "my-agent",
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash",
         google_api_key=None,
         google_cloud_project=None,
         google_cloud_region=None,
@@ -182,7 +182,7 @@ def test_run_cmd_with_type_config(
 
   cli_create.run_cmd(
       agent_name,
-      model="gemini-2.0-flash-001",
+      model="gemini-2.5-flash",
       google_api_key="test-key",
       google_cloud_project=None,
       google_cloud_region=None,
@@ -200,7 +200,7 @@ def test_run_cmd_with_type_config(
   # Check YAML content
   yaml_content = yaml_file.read_text()
   assert "name: root_agent" in yaml_content
-  assert "model: gemini-2.0-flash-001" in yaml_content
+  assert "model: gemini-2.5-flash" in yaml_content
   assert "description: A helpful assistant for user questions." in yaml_content
 
   # Should create empty __init__.py

tests/unittests/code_executors/test_built_in_code_executor.py

@@ -27,7 +27,7 @@ def test_process_llm_request_gemini_2_model_config_none(
     built_in_executor: BuiltInCodeExecutor,
 ):
   """Tests processing when llm_request.config is None for Gemini 2."""
-  llm_request = LlmRequest(model="gemini-2.0-flash")
+  llm_request = LlmRequest(model="gemini-2.5-flash")
   built_in_executor.process_llm_request(llm_request)
   assert llm_request.config is not None
   assert llm_request.config.tools == [
@@ -72,7 +72,7 @@ def test_process_llm_request_gemini_2_model_with_existing_tools(
       ]
   )
   llm_request = LlmRequest(
-      model="gemini-2.0-flash-001",
+      model="gemini-2.5-flash",
       config=types.GenerateContentConfig(tools=[existing_tool]),
   )
   built_in_executor.process_llm_request(llm_request)