chore: GenAI - Context Caching - system test model changed to preview model

Zhenyi Qi · copybara-github · commit 97b91dbf71e1 · 2024-06-17T15:21:30.000-07:00
PiperOrigin-RevId: 644149587
diff --git a/tests/system/vertexai/test_generative_models.py b/tests/system/vertexai/test_generative_models.py
@@ -125,13 +125,13 @@ def test_generate_content_with_cached_content_from_text(self):
             ],
         )
 
-        model = generative_models.GenerativeModel.from_cached_content(
+        model = preview_generative_models.GenerativeModel.from_cached_content(
             cached_content=cached_content
         )
 
         response = model.generate_content(
             "Why is sky blue?",
-            generation_config=generative_models.GenerationConfig(temperature=0),
+            generation_config=preview_generative_models.GenerationConfig(temperature=0),
         )
         try:
             assert response.text

Original file line number	Diff line number	Diff line change
`@@ -125,13 +125,13 @@ def test_generate_content_with_cached_content_from_text(self):`
`125`	`125`	`],`
`126`	`126`	`)`
`127`	`127`
`128`		`- model = generative_models.GenerativeModel.from_cached_content(`
	`128`	`+ model = preview_generative_models.GenerativeModel.from_cached_content(`
`129`	`129`	`cached_content=cached_content`
`130`	`130`	`)`
`131`	`131`
`132`	`132`	`response = model.generate_content(`
`133`	`133`	`"Why is sky blue?",`
`134`		`- generation_config=generative_models.GenerationConfig(temperature=0),`
	`134`	`+ generation_config=preview_generative_models.GenerationConfig(temperature=0),`
`135`	`135`	`)`
`136`	`136`	`try:`
`137`	`137`	`assert response.text`