fix: GenAI SDK client(multimodal) - Change dataset "name" to refer to resource name instead of dataset ID.

PiperOrigin-RevId: 896997282

Author: cleop-google

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

@@ -23,7 +23,7 @@
     "gs://google-cloud-aiplatform/schema/dataset/metadata/multimodal_1.0.0.yaml"
 )
 BIGQUERY_TABLE_NAME = "vertex-sdk-dev.multimodal_dataset.test-table"
-DATASET = "8810841321427173376"
+DATASET = "projects/vertex-sdk-dev/locations/us-central1/datasets/8810841321427173376"
 
 
 def test_assemble_dataset(client):

vertexai/_genai/datasets.py

@@ -274,9 +274,9 @@ def _assemble_multimodal_dataset(
             request_dict = _AssembleDatasetParameters_to_vertex(parameter_model)
             request_url_dict = request_dict.get("_url")
             if request_url_dict:
-                path = "datasets/{name}:assemble".format_map(request_url_dict)
+                path = "{name}:assemble".format_map(request_url_dict)
             else:
-                path = "datasets/{name}:assemble"
+                path = "{name}:assemble"
 
         query_params = request_dict.get("_query")
         if query_params:
@@ -364,9 +364,9 @@ def _assess_multimodal_dataset(
             request_dict = _AssessDatasetParameters_to_vertex(parameter_model)
             request_url_dict = request_dict.get("_url")
             if request_url_dict:
-                path = "datasets/{name}:assess".format_map(request_url_dict)
+                path = "{name}:assess".format_map(request_url_dict)
             else:
-                path = "datasets/{name}:assess"
+                path = "{name}:assess"
 
         query_params = request_dict.get("_query")
         if query_params:
@@ -513,9 +513,9 @@ def _delete_multimodal_dataset(
             )
             request_url_dict = request_dict.get("_url")
             if request_url_dict:
-                path = "datasets/{name}".format_map(request_url_dict)
+                path = "{name}".format_map(request_url_dict)
             else:
-                path = "datasets/{name}"
+                path = "{name}"
 
         query_params = request_dict.get("_query")
         if query_params:
@@ -583,9 +583,9 @@ def _get_multimodal_dataset(
             request_dict = _GetMultimodalDatasetParameters_to_vertex(parameter_model)
             request_url_dict = request_dict.get("_url")
             if request_url_dict:
-                path = "datasets/{name}".format_map(request_url_dict)
+                path = "{name}".format_map(request_url_dict)
             else:
-                path = "datasets/{name}"
+                path = "{name}"
 
         query_params = request_dict.get("_query")
         if query_params:
@@ -805,9 +805,9 @@ def _update_multimodal_dataset(
             request_dict = _UpdateMultimodalDatasetParameters_to_vertex(parameter_model)
             request_url_dict = request_dict.get("_url")
             if request_url_dict:
-                path = "datasets/{name}".format_map(request_url_dict)
+                path = "{name}".format_map(request_url_dict)
             else:
-                path = "datasets/{name}"
+                path = "{name}"
 
         query_params = request_dict.get("_query")
         if query_params:
@@ -1475,9 +1475,9 @@ async def _assemble_multimodal_dataset(
             request_dict = _AssembleDatasetParameters_to_vertex(parameter_model)
             request_url_dict = request_dict.get("_url")
             if request_url_dict:
-                path = "datasets/{name}:assemble".format_map(request_url_dict)
+                path = "{name}:assemble".format_map(request_url_dict)
             else:
-                path = "datasets/{name}:assemble"
+                path = "{name}:assemble"
 
         query_params = request_dict.get("_query")
         if query_params:
@@ -1567,9 +1567,9 @@ async def _assess_multimodal_dataset(
             request_dict = _AssessDatasetParameters_to_vertex(parameter_model)
             request_url_dict = request_dict.get("_url")
             if request_url_dict:
-                path = "datasets/{name}:assess".format_map(request_url_dict)
+                path = "{name}:assess".format_map(request_url_dict)
             else:
-                path = "datasets/{name}:assess"
+                path = "{name}:assess"
 
         query_params = request_dict.get("_query")
         if query_params:
@@ -1720,9 +1720,9 @@ async def _delete_multimodal_dataset(
             )
             request_url_dict = request_dict.get("_url")
             if request_url_dict:
-                path = "datasets/{name}".format_map(request_url_dict)
+                path = "{name}".format_map(request_url_dict)
             else:
-                path = "datasets/{name}"
+                path = "{name}"
 
         query_params = request_dict.get("_query")
         if query_params:
@@ -1792,9 +1792,9 @@ async def _get_multimodal_dataset(
             request_dict = _GetMultimodalDatasetParameters_to_vertex(parameter_model)
             request_url_dict = request_dict.get("_url")
             if request_url_dict:
-                path = "datasets/{name}".format_map(request_url_dict)
+                path = "{name}".format_map(request_url_dict)
             else:
-                path = "datasets/{name}"
+                path = "{name}"
 
         query_params = request_dict.get("_query")
         if query_params:
@@ -2020,9 +2020,9 @@ async def _update_multimodal_dataset(
             request_dict = _UpdateMultimodalDatasetParameters_to_vertex(parameter_model)
             request_url_dict = request_dict.get("_url")
             if request_url_dict:
-                path = "datasets/{name}".format_map(request_url_dict)
+                path = "{name}".format_map(request_url_dict)
             else:
-                path = "datasets/{name}"
+                path = "{name}"
 
         query_params = request_dict.get("_query")
         if query_params: