Skip to content

Commit b6f478a

Browse files
authored
Fixed test_agent_types after Gemini 1.5 deprecation (#1043)
1 parent c4dfbb4 commit b6f478a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_agents.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -328,7 +328,7 @@ async def test_get_directory_index_w_no_citations(
328328

329329
@pytest.mark.flaky(reruns=2, only_rerun=["AssertionError", "httpx.RemoteProtocolError"])
330330
@pytest.mark.parametrize("agent_type", [FAKE_AGENT_TYPE, ToolSelector, SimpleAgent])
331-
@pytest.mark.parametrize("llm_name", ["gpt-4o", "gemini/gemini-1.5-flash"])
331+
@pytest.mark.parametrize("llm_name", ["gpt-4o", "gemini/gemini-2.0-flash-lite"])
332332
@pytest.mark.asyncio
333333
async def test_agent_types(
334334
agent_test_settings: Settings,

0 commit comments

Comments
 (0)