add gemini2 to more tests

Author: evalstate

tests/e2e/multimodal/test_multimodal_images.py

@@ -18,6 +18,7 @@
     [
         "gpt-4.1-mini",  # OpenAI model
         "sonnet",  # Anthropic model
+        "gemini2",
     ],
 )
 async def test_agent_with_image_prompt(fast_agent, model_name):
@@ -49,9 +50,9 @@ async def agent_function():
 @pytest.mark.parametrize(
     "model_name",
     [
-        "gpt-4.1-mini",  # OpenAI model
-        "sonnet",  # Anthropic model
-        #    "gemini2",
+        "gpt-4.1-mini",  # OpenAI
+        "sonnet",  # Anthropic
+        "gemini2",  # Google
     ],
 )
 async def test_agent_with_mcp_image(fast_agent, model_name):
@@ -88,7 +89,6 @@ async def agent_function():
     ],
 )
 async def test_agent_with_mcp_pdf(fast_agent, model_name):
-    """Test that the agent can process an image and respond appropriately."""
     fast = fast_agent
 
     # Define the agent
@@ -191,6 +191,7 @@ async def agent_function():
     "model_name",
     [
         "gpt-4.1-mini",  # OpenAI model
+        "gemini2",
     ],
 )
 async def test_agent_includes_tool_results_in_multipart_result_openai(fast_agent, model_name):

tests/e2e/smoke/test_e2e_smoke.py

@@ -24,10 +24,9 @@
         "gpt-4o-mini",  # OpenAI model
         "haiku35",  # Anthropic model
         "deepseek",
-        #        "generic.qwen2.5:latest",
         "generic.llama3.2:latest",
         "openrouter.google/gemini-2.0-flash-001",
-        "google.gemini-2.0-flash",
+        "gemini2",
     ],
 )
 async def test_basic_textual_prompting(fast_agent, model_name):
@@ -61,10 +60,10 @@ async def agent_function():
         "haiku35",  # Anthropic model
         "deepseek",
         "openrouter.google/gemini-2.0-flash-001",
+        "gemini2",
     ],
 )
 async def test_multiple_text_blocks_prompting(fast_agent, model_name):
-    """Test that the agent can process an image and respond appropriately."""
     fast = fast_agent
 
     # Define the agent
@@ -148,6 +147,7 @@ class WeatherForecast(BaseModel):
         "gpt-4.1",
         "gpt-4.1-nano",
         "gpt-4.1-mini",
+        "gemini2",
     ],
 )
 async def test_structured_weather_forecast_openai_structured_api(fast_agent, model_name):
@@ -261,7 +261,7 @@ async def agent_function():
         "gpt-4.1",
         "gpt-4.1-nano",
         "gpt-4.1-mini",
-        "google.gemini-2.0-flash",
+        "gemini2",
         "openrouter.google/gemini-2.0-flash-001",
     ],
 )
@@ -307,7 +307,7 @@ async def weather_forecast():
         "gpt-4.1",
         "gpt-4.1-nano",
         "gpt-4.1-mini",
-        "google.gemini-2.0-flash",
+        "gemini2",
         "openrouter.anthropic/claude-3.7-sonnet",
     ],
 )

tests/e2e/structured/test_structured_outputs.py

@@ -125,7 +125,7 @@ async def create_structured():
         # "haiku", -- anthropic do not support structured outputs this way
         "gpt-4.1-mini",
         "openrouter.google/gemini-2.0-flash-001",
-        "google.gemini-2.0-flash",
+        "gemini2",
     ],
 )
 async def test_structured_output_with_response_format_overriden(fast_agent, model_name):
@@ -157,7 +157,7 @@ async def create_structured():
 @pytest.mark.e2e
 @pytest.mark.parametrize(
     "model_name",
-    ["gpt-4.1-mini", "haiku", "google.gemini-2.0-flash"],
+    ["gpt-4.1-mini", "haiku", "gemini2"],
 )
 async def test_history_management_with_structured(fast_agent, model_name):
     """Test that the agent can generate structured response with response_format_specified."""