feat: GenAI SDK client(multimodal) - Allow fetching dataset with dataset_id in get_multimodal_dataset.

FUTURE_COPYBARA_INTEGRATE_REVIEW=#6572 from googleapis:release-please--branches--main 1448130
PiperOrigin-RevId: 899573600

Author: cleop-google

tests/unit/vertexai/genai/replays/test_get_multimodal_datasets.py

@@ -41,6 +41,15 @@ def test_get_dataset_from_public_method(client):
     assert dataset.display_name == "test-display-name"
 
 
+def test_get_dataset_by_id(client):
+    dataset = client.datasets.get_multimodal_dataset(
+        dataset_id="8810841321427173376",
+    )
+    assert isinstance(dataset, types.MultimodalDataset)
+    assert dataset.name == DATASET
+    assert dataset.display_name == "test-display-name"
+
+
 pytestmark = pytest_helper.setup(
     file=__file__,
     globals_for_file=globals(),
@@ -67,3 +76,13 @@ async def test_get_dataset_from_public_method_async(client):
     assert isinstance(dataset, types.MultimodalDataset)
     assert dataset.name == DATASET
     assert dataset.display_name == "test-display-name"
+
+
+@pytest.mark.asyncio
+async def test_get_dataset_by_id_async(client):
+    dataset = await client.aio.datasets.get_multimodal_dataset(
+        dataset_id="8810841321427173376",
+    )
+    assert isinstance(dataset, types.MultimodalDataset)
+    assert dataset.name == DATASET
+    assert dataset.display_name == "test-display-name"

vertexai/_genai/datasets.py

@@ -1123,15 +1123,19 @@ def update_multimodal_dataset(
     def get_multimodal_dataset(
         self,
         *,
-        name: str,
+        name: Optional[str] = None,
+        dataset_id: Optional[str] = None,
         config: Optional[types.VertexBaseConfigOrDict] = None,
     ) -> types.MultimodalDataset:
         """Gets a multimodal dataset.
 
         Args:
           name:
-            Required. name of a multimodal dataset. The name should be in
+            Optional. name of a multimodal dataset. The name should be in
             the format of "projects/{project}/locations/{location}/datasets/{dataset}".
+          dataset_id:
+            Optional. The ID of the dataset to get. If provided, the full name
+            will be constructed using the client's project and location.
           config:
             Optional. A configuration for getting the multimodal dataset. If not
             provided, the default configuration will be used.
@@ -1145,6 +1149,14 @@ def get_multimodal_dataset(
         elif not config:
             config = types.VertexBaseConfig()
 
+        if not name and not dataset_id:
+            raise ValueError("Either 'name' or 'dataset_id' must be provided.")
+        if name and dataset_id:
+            raise ValueError("Only one of 'name' or 'dataset_id' can be provided.")
+
+        if dataset_id:
+            name = f"projects/{self._api_client.project}/locations/{self._api_client.location}/datasets/{dataset_id}"
+
         return self._get_multimodal_dataset(config=config, name=name)
 
     def delete_multimodal_dataset(
@@ -2345,15 +2357,19 @@ async def update_multimodal_dataset(
     async def get_multimodal_dataset(
         self,
         *,
-        name: str,
+        name: Optional[str] = None,
+        dataset_id: Optional[str] = None,
         config: Optional[types.VertexBaseConfigOrDict] = None,
     ) -> types.MultimodalDataset:
         """Gets a multimodal dataset.
 
         Args:
           name:
-            Required. name of a multimodal dataset. The name should be in
+            Optional. name of a multimodal dataset. The name should be in
             the format of "projects/{project}/locations/{location}/datasets/{dataset}".
+          dataset_id:
+            Optional. The ID of the dataset to get. If provided, the full name
+            will be constructed using the client's project and location.
           config:
             Optional. A configuration for getting the multimodal dataset. If not
             provided, the default configuration will be used.
@@ -2367,6 +2383,14 @@ async def get_multimodal_dataset(
         elif not config:
             config = types.VertexBaseConfig()
 
+        if not name and not dataset_id:
+            raise ValueError("Either 'name' or 'dataset_id' must be provided.")
+        if name and dataset_id:
+            raise ValueError("Only one of 'name' or 'dataset_id' can be provided.")
+
+        if dataset_id:
+            name = f"projects/{self._api_client.project}/locations/{self._api_client.location}/datasets/{dataset_id}"
+
         return await self._get_multimodal_dataset(config=config, name=name)
 
     async def delete_multimodal_dataset(