Update requirements and test

holtskinner · holtskinner · commit 3adc49539a81 · 2025-02-06T09:53:40.000-06:00
diff --git a/genai/express_mode/api_key_example_test.py b/genai/express_mode/api_key_example_test.py
@@ -40,7 +40,7 @@ def test_api_key_example(mock_genai_client: MagicMock) -> None:
 
     mock_genai_client.assert_called_once_with(vertexai=True, api_key="YOUR_API_KEY")
     mock_genai_client.return_value.models.generate_content.assert_called_once_with(
-        model="gemini-2.0-flash-exp",
+        model="gemini-2.0-flash-001",
         contents="Explain bubble sort to me.",
     )
     assert response == "This is a mocked bubble sort explanation."
diff --git a/genai/express_mode/requirements.txt b/genai/express_mode/requirements.txt
@@ -1 +1 @@
-google-genai==0.6.0
+google-genai==1.0.0

Original file line number	Diff line number	Diff line change
`@@ -40,7 +40,7 @@ def test_api_key_example(mock_genai_client: MagicMock) -> None:`
`40`	`40`
`41`	`41`	`mock_genai_client.assert_called_once_with(vertexai=True, api_key="YOUR_API_KEY")`
`42`	`42`	`mock_genai_client.return_value.models.generate_content.assert_called_once_with(`
`43`		`- model="gemini-2.0-flash-exp",`
	`43`	`+ model="gemini-2.0-flash-001",`
`44`	`44`	`contents="Explain bubble sort to me.",`
`45`	`45`	`)`
`46`	`46`	`assert response == "This is a mocked bubble sort explanation."`