fix: GenAI SDK client(multimodal) - Remove UpdateMultimodalDatasetConfig and replace with VertexBaseConfig.

PiperOrigin-RevId: 900128682

Author: cleop-google

vertexai/_genai/datasets.py

@@ -782,7 +782,7 @@ def _update_multimodal_dataset(
         metadata: Optional[types.SchemaTablesDatasetMetadataOrDict] = None,
         description: Optional[str] = None,
         encryption_spec: Optional[genai_types.EncryptionSpecOrDict] = None,
-        config: Optional[types.UpdateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.VertexBaseConfigOrDict] = None,
     ) -> types.MultimodalDataset:
         """
         Updates a multimodal dataset resource.
@@ -1084,7 +1084,7 @@ def update_multimodal_dataset(
         self,
         *,
         multimodal_dataset: types.MultimodalDatasetOrDict,
-        config: Optional[types.UpdateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.VertexBaseConfigOrDict] = None,
     ) -> types.MultimodalDataset:
         """Updates a multimodal dataset.
 
@@ -1108,9 +1108,9 @@ def update_multimodal_dataset(
         _datasets_utils.validate_multimodal_dataset_bigquery_uri(multimodal_dataset)
 
         if isinstance(config, dict):
-            config = types.UpdateMultimodalDatasetConfig(**config)
+            config = types.VertexBaseConfig(**config)
         elif not config:
-            config = types.UpdateMultimodalDatasetConfig()
+            config = types.VertexBaseConfig()
 
         return self._update_multimodal_dataset(
             config=config,
@@ -2004,7 +2004,7 @@ async def _update_multimodal_dataset(
         metadata: Optional[types.SchemaTablesDatasetMetadataOrDict] = None,
         description: Optional[str] = None,
         encryption_spec: Optional[genai_types.EncryptionSpecOrDict] = None,
-        config: Optional[types.UpdateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.VertexBaseConfigOrDict] = None,
     ) -> types.MultimodalDataset:
         """
         Updates a multimodal dataset resource.
@@ -2310,7 +2310,7 @@ async def update_multimodal_dataset(
         self,
         *,
         multimodal_dataset: types.MultimodalDatasetOrDict,
-        config: Optional[types.UpdateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.VertexBaseConfigOrDict] = None,
     ) -> types.MultimodalDataset:
         """Updates a multimodal dataset.
 
@@ -2330,9 +2330,9 @@ async def update_multimodal_dataset(
         _datasets_utils.validate_multimodal_dataset_bigquery_uri(multimodal_dataset)
 
         if isinstance(config, dict):
-            config = types.UpdateMultimodalDatasetConfig(**config)
+            config = types.VertexBaseConfig(**config)
         elif not config:
-            config = types.UpdateMultimodalDatasetConfig()
+            config = types.VertexBaseConfig()
 
         return await self._update_multimodal_dataset(
             config=config,

vertexai/_genai/types/__init__.py

@@ -1251,9 +1251,6 @@
 from .common import UpdateAgentEngineSessionConfig
 from .common import UpdateAgentEngineSessionConfigDict
 from .common import UpdateAgentEngineSessionConfigOrDict
-from .common import UpdateMultimodalDatasetConfig
-from .common import UpdateMultimodalDatasetConfigDict
-from .common import UpdateMultimodalDatasetConfigOrDict
 from .common import UpdatePromptConfig
 from .common import UpdatePromptConfigDict
 from .common import UpdatePromptConfigOrDict
@@ -2156,9 +2153,6 @@
     "ListMultimodalDatasetsResponse",
     "ListMultimodalDatasetsResponseDict",
     "ListMultimodalDatasetsResponseOrDict",
-    "UpdateMultimodalDatasetConfig",
-    "UpdateMultimodalDatasetConfigDict",
-    "UpdateMultimodalDatasetConfigOrDict",
     "SchemaPredictParamsGroundingConfigSourceEntry",
     "SchemaPredictParamsGroundingConfigSourceEntryDict",
     "SchemaPredictParamsGroundingConfigSourceEntryOrDict",

vertexai/_genai/types/common.py

@@ -14145,35 +14145,6 @@ class ListMultimodalDatasetsResponseDict(TypedDict, total=False):
 ]
 
 
-class UpdateMultimodalDatasetConfig(_common.BaseModel):
-    """Config for updating a multimodal dataset resource."""
-
-    http_options: Optional[genai_types.HttpOptions] = Field(
-        default=None, description="""Used to override HTTP request options."""
-    )
-    timeout: Optional[int] = Field(
-        default=90,
-        description="""The timeout for the update dataset request in seconds. If not set,
-      the default timeout is 90 seconds.""",
-    )
-
-
-class UpdateMultimodalDatasetConfigDict(TypedDict, total=False):
-    """Config for updating a multimodal dataset resource."""
-
-    http_options: Optional[genai_types.HttpOptionsDict]
-    """Used to override HTTP request options."""
-
-    timeout: Optional[int]
-    """The timeout for the update dataset request in seconds. If not set,
-      the default timeout is 90 seconds."""
-
-
-UpdateMultimodalDatasetConfigOrDict = Union[
-    UpdateMultimodalDatasetConfig, UpdateMultimodalDatasetConfigDict
-]
-
-
 class _UpdateMultimodalDatasetParameters(_common.BaseModel):
     """Parameters for updating a multimodal dataset resource."""
 
@@ -14186,9 +14157,7 @@ class _UpdateMultimodalDatasetParameters(_common.BaseModel):
     encryption_spec: Optional[genai_types.EncryptionSpec] = Field(
         default=None, description=""""""
     )
-    config: Optional[UpdateMultimodalDatasetConfig] = Field(
-        default=None, description=""""""
-    )
+    config: Optional[VertexBaseConfig] = Field(default=None, description="""""")
 
 
 class _UpdateMultimodalDatasetParametersDict(TypedDict, total=False):
@@ -14209,7 +14178,7 @@ class _UpdateMultimodalDatasetParametersDict(TypedDict, total=False):
     encryption_spec: Optional[genai_types.EncryptionSpecDict]
     """"""
 
-    config: Optional[UpdateMultimodalDatasetConfigDict]
+    config: Optional[VertexBaseConfigDict]
     """"""