From 5343f84b29be63122c9690e1b1195eeb0b11f3b2 Mon Sep 17 00:00:00 2001 From: Christian Leopoldseder Date: Tue, 14 Apr 2026 02:44:56 -0700 Subject: [PATCH] chore: GenAI SDK client(multimodal) - Remove `UpdateMultimodalDatasetConfig` and replace with `VertexBaseConfig`. PiperOrigin-RevId: 899463238 --- vertexai/_genai/datasets.py | 16 +++++++------- vertexai/_genai/types/__init__.py | 6 ------ vertexai/_genai/types/common.py | 35 ++----------------------------- 3 files changed, 10 insertions(+), 47 deletions(-) diff --git a/vertexai/_genai/datasets.py b/vertexai/_genai/datasets.py index e56e96175e..65adb4a19e 100644 --- a/vertexai/_genai/datasets.py +++ b/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, diff --git a/vertexai/_genai/types/__init__.py b/vertexai/_genai/types/__init__.py index 07cc75b8c1..dede09b420 100644 --- a/vertexai/_genai/types/__init__.py +++ b/vertexai/_genai/types/__init__.py @@ -1225,9 +1225,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 @@ -2106,9 +2103,6 @@ "ListMultimodalDatasetsResponse", "ListMultimodalDatasetsResponseDict", "ListMultimodalDatasetsResponseOrDict", - "UpdateMultimodalDatasetConfig", - "UpdateMultimodalDatasetConfigDict", - "UpdateMultimodalDatasetConfigOrDict", "SchemaPredictParamsGroundingConfigSourceEntry", "SchemaPredictParamsGroundingConfigSourceEntryDict", "SchemaPredictParamsGroundingConfigSourceEntryOrDict", diff --git a/vertexai/_genai/types/common.py b/vertexai/_genai/types/common.py index 04b2415ab0..b559d2f5f6 100644 --- a/vertexai/_genai/types/common.py +++ b/vertexai/_genai/types/common.py @@ -13798,35 +13798,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.""" @@ -13839,9 +13810,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): @@ -13862,7 +13831,7 @@ class _UpdateMultimodalDatasetParametersDict(TypedDict, total=False): encryption_spec: Optional[genai_types.EncryptionSpecDict] """""" - config: Optional[UpdateMultimodalDatasetConfigDict] + config: Optional[VertexBaseConfigDict] """"""