fix: GenAI SDK client(multimodal) - Fix "type mismatch" warnings in datasets module.

PiperOrigin-RevId: 897127993

Author: cleop-google

vertexai/_genai/datasets.py

@@ -1080,7 +1080,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.
 
@@ -1104,9 +1104,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,
@@ -1120,7 +1120,7 @@ def get_multimodal_dataset(
         self,
         *,
         name: str,
-        config: Optional[types.CreateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.VertexBaseConfigOrDict] = None,
     ) -> types.MultimodalDataset:
         """Gets a multimodal dataset.
 
@@ -1136,17 +1136,17 @@ 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)
 
     def delete_multimodal_dataset(
         self,
         *,
         name: str,
-        config: Optional[types.CreateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.VertexBaseConfigOrDict] = None,
     ) -> types.MultimodalDatasetOperation:
         """Deletes a multimodal dataset.
 
@@ -1162,9 +1162,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)
 
@@ -2299,7 +2299,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.
 
@@ -2319,9 +2319,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,
@@ -2335,7 +2335,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.
 
@@ -2351,17 +2351,17 @@ 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)
 
     async def delete_multimodal_dataset(
         self,
         *,
         name: str,
-        config: Optional[types.CreateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.VertexBaseConfigOrDict] = None,
     ) -> types.MultimodalDatasetOperation:
         """Deletes a multimodal dataset.
 
@@ -2377,9 +2377,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)