samples(genai): Update Vertex AI Express Mode sample to gemini-2.0-flash-001 (GoogleCloudPlatform#13142)

holtskinner · web-flow · commit a1d11b2e2cc5 · 2025-02-06T10:39:56.000-06:00
* samples(genai): Update Vertex AI Express Mode sample to `gemini-2.0-flash-001`

* Update requirements and test
diff --git a/genai/express_mode/api_key_example.py b/genai/express_mode/api_key_example.py
@@ -23,7 +23,7 @@ def generate_content() -> str:
     client = genai.Client(vertexai=True, api_key=API_KEY)
 
     response = client.models.generate_content(
-        model="gemini-2.0-flash-exp",
+        model="gemini-2.0-flash-001",
         contents="""Explain bubble sort to me.""",
     )
 
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
`@@ -23,7 +23,7 @@ def generate_content() -> str:`
`23`	`23`	`client = genai.Client(vertexai=True, api_key=API_KEY)`
`24`	`24`
`25`	`25`	`response = client.models.generate_content(`
`26`		`- model="gemini-2.0-flash-exp",`
	`26`	`+ model="gemini-2.0-flash-001",`
`27`	`27`	`contents="""Explain bubble sort to me.""",`
`28`	`28`	`)`
`29`	`29`
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."`