From 4b4017b2c8e0b6f733cb5fb163e2432e8910bc26 Mon Sep 17 00:00:00 2001
From: Christian Leopoldseder <cleop@google.com>
Date: Thu, 9 Apr 2026 08:32:13 -0700
Subject: [PATCH] fix: GenAI SDK client(multimodal) - Fix "type mismatch"
 warnings in `datasets` module.

PiperOrigin-RevId: 897127993
---
 vertexai/_genai/datasets.py | 36 ++++++++++++++++++------------------
 1 file changed, 18 insertions(+), 18 deletions(-)

diff --git a/vertexai/_genai/datasets.py b/vertexai/_genai/datasets.py
index 0d9900e47e..2b6b71334e 100644
--- a/vertexai/_genai/datasets.py
+++ b/vertexai/_genai/datasets.py
@@ -1085,7 +1085,7 @@ def update_multimodal_dataset(
         self,
         *,
         multimodal_dataset: types.MultimodalDatasetOrDict,
-        config: Optional[types.CreateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.UpdateMultimodalDatasetConfigOrDict] = None,
     ) -> types.MultimodalDataset:
         """Updates a multimodal dataset.
 
@@ -1109,9 +1109,9 @@ def update_multimodal_dataset(
         _datasets_utils.validate_multimodal_dataset_bigquery_uri(multimodal_dataset)
 
         if isinstance(config, dict):
-            config = types.CreateMultimodalDatasetConfig(**config)
+            config = types.UpdateMultimodalDatasetConfig(**config)
         elif not config:
-            config = types.CreateMultimodalDatasetConfig()
+            config = types.UpdateMultimodalDatasetConfig()
 
         return self._update_multimodal_dataset(
             config=config,
@@ -1125,7 +1125,7 @@ def get_multimodal_dataset(
         self,
         *,
         name: str,
-        config: Optional[types.CreateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.VertexBaseConfigOrDict] = None,
     ) -> types.MultimodalDataset:
         """Gets a multimodal dataset.
 
@@ -1141,9 +1141,9 @@ def get_multimodal_dataset(
           dataset.
         """
         if isinstance(config, dict):
-            config = types.CreateMultimodalDatasetConfig(**config)
+            config = types.VertexBaseConfig(**config)
         elif not config:
-            config = types.CreateMultimodalDatasetConfig()
+            config = types.VertexBaseConfig()
 
         return self._get_multimodal_dataset(config=config, name=name)
 
@@ -1151,7 +1151,7 @@ def delete_multimodal_dataset(
         self,
         *,
         name: str,
-        config: Optional[types.CreateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.VertexBaseConfigOrDict] = None,
     ) -> types.MultimodalDatasetOperation:
         """Deletes a multimodal dataset.
 
@@ -1167,9 +1167,9 @@ def delete_multimodal_dataset(
           multimodal dataset operation.
         """
         if isinstance(config, dict):
-            config = types.CreateMultimodalDatasetConfig(**config)
+            config = types.VertexBaseConfig(**config)
         elif not config:
-            config = types.CreateMultimodalDatasetConfig()
+            config = types.VertexBaseConfig()
 
         return self._delete_multimodal_dataset(config=config, name=name)
 
@@ -2309,7 +2309,7 @@ async def update_multimodal_dataset(
         self,
         *,
         multimodal_dataset: types.MultimodalDatasetOrDict,
-        config: Optional[types.CreateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.UpdateMultimodalDatasetConfigOrDict] = None,
     ) -> types.MultimodalDataset:
         """Updates a multimodal dataset.
 
@@ -2329,9 +2329,9 @@ async def update_multimodal_dataset(
         _datasets_utils.validate_multimodal_dataset_bigquery_uri(multimodal_dataset)
 
         if isinstance(config, dict):
-            config = types.CreateMultimodalDatasetConfig(**config)
+            config = types.UpdateMultimodalDatasetConfig(**config)
         elif not config:
-            config = types.CreateMultimodalDatasetConfig()
+            config = types.UpdateMultimodalDatasetConfig()
 
         return await self._update_multimodal_dataset(
             config=config,
@@ -2345,7 +2345,7 @@ async def get_multimodal_dataset(
         self,
         *,
         name: str,
-        config: Optional[types.CreateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.VertexBaseConfigOrDict] = None,
     ) -> types.MultimodalDataset:
         """Gets a multimodal dataset.
 
@@ -2361,9 +2361,9 @@ async def get_multimodal_dataset(
           dataset.
         """
         if isinstance(config, dict):
-            config = types.CreateMultimodalDatasetConfig(**config)
+            config = types.VertexBaseConfig(**config)
         elif not config:
-            config = types.CreateMultimodalDatasetConfig()
+            config = types.VertexBaseConfig()
 
         return await self._get_multimodal_dataset(config=config, name=name)
 
@@ -2371,7 +2371,7 @@ async def delete_multimodal_dataset(
         self,
         *,
         name: str,
-        config: Optional[types.CreateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.VertexBaseConfigOrDict] = None,
     ) -> types.MultimodalDatasetOperation:
         """Deletes a multimodal dataset.
 
@@ -2387,9 +2387,9 @@ async def delete_multimodal_dataset(
           multimodal dataset operation.
         """
         if isinstance(config, dict):
-            config = types.CreateMultimodalDatasetConfig(**config)
+            config = types.VertexBaseConfig(**config)
         elif not config:
-            config = types.CreateMultimodalDatasetConfig()
+            config = types.VertexBaseConfig()
 
         return await self._delete_multimodal_dataset(config=config, name=name)