chore: add support for MLDev cache creation tests in shared test suite.

FUTURE_COPYBARA_INTEGRATE_REVIEW=#1526 from googleapis:release-please--branches--main f06d3b9
PiperOrigin-RevId: 822286882

Author: speedstorm1

google/genai/tests/data/a-man-and-a-dog.png

@@ -13,23 +13,38 @@
 # limitations under the License.
 #
 
-"""Tests for caches.create() and caches.delete()."""
+"""Tests for caches.create(), caches.get(), and caches.delete()."""
 
 from .... import types as genai_types
 from ... import pytest_helper
 
 
-def create_delete(client, parameters):
-  cache = client.caches.create(
-      model=parameters.model,
-      config=parameters.config,
-  )
-  client.caches.delete(name=cache.name)
+def create_get_delete(client, parameters):
+  if client.vertexai:
+    cache = client.caches.create(
+        model=parameters.model,
+        config=parameters.config,
+    )
+  else:
+    # For MLDev, we must create the file before creating the cache.
+    file = client.files.upload(file="tests/data/a-man-and-a-dog.png")
+    config = genai_types.CreateCachedContentConfig(
+        contents=[
+            genai_types.Part.from_uri(
+                file_uri=file.uri,
+                mime_type=file.mime_type,
+            )
+        ]
+        * 5
+    )
+    cache = client.caches.create(model=parameters.model, config=config)
+  got_cache = client.caches.get(name=cache.name)
+  client.caches.delete(name=got_cache.name)
 
 
 test_table: list[pytest_helper.TestTableItem] = [
     pytest_helper.TestTableItem(
-        name="test_create_delete_vertex",
+        name="test_create_get_delete",
         parameters=genai_types._CreateCachedContentParameters(
             model="gemini-2.5-flash",
             config=genai_types.CreateCachedContentConfig(
@@ -39,14 +54,13 @@ def create_delete(client, parameters):
                 )] * 5
             ),
         ),
-        exception_if_mldev="unsupported",
     ),
 ]
 
 pytestmark = pytest_helper.setup(
     file=__file__,
     globals_for_file=globals(),
-    test_method="create_delete",
+    test_method="create_get_delete",
     test_table=test_table,
 )
 

google/genai/tests/shared/caches/test_create_delete.py renamed to google/genai/tests/shared/caches/test_create_get_delete.py

@@ -20,21 +20,35 @@
 
 
 def create_update_get(client, parameters):
-  cache = client.caches.create(
-      model=parameters.model,
-      config=parameters.config,
-  )
+  if client.vertexai:
+    cache = client.caches.create(
+        model=parameters.model,
+        config=parameters.config,
+    )
+  else:
+    # For MLDev, we must create the file before creating the cache.
+    file = client.files.upload(file="tests/data/a-man-and-a-dog.png")
+    config = genai_types.CreateCachedContentConfig(
+        contents=[
+            genai_types.Part.from_uri(
+                file_uri=file.uri,
+                mime_type=file.mime_type,
+            )
+        ]
+        * 5
+    )
+    cache = client.caches.create(model=parameters.model, config=config)
+
   updated_cache = client.caches.update(
       name=cache.name,
-      config=genai_types.UpdateCachedContentConfig(ttl="3600s"),
+      config=genai_types.UpdateCachedContentConfig(ttl="7200s"),
   )
-  got_cache = client.caches.get(name=updated_cache.name)
-  client.caches.delete(name=got_cache.name)
+  _ = client.caches.get(name=updated_cache.name)
 
 
 test_table: list[pytest_helper.TestTableItem] = [
     pytest_helper.TestTableItem(
-        name="test_create_update_get_vertex",
+        name="test_create_update_get",
         parameters=genai_types._CreateCachedContentParameters(
             model="gemini-2.5-flash",
             config=genai_types.CreateCachedContentConfig(
@@ -44,7 +58,6 @@ def create_update_get(client, parameters):
                 )] * 5
             ),
         ),
-        exception_if_mldev="unsupported",
     ),
 ]