feat: GenAI SDK client(multimodal) - Add set_read_config to MultimodalDataset.

PiperOrigin-RevId: 885611714

Author: cleop-google

tests/unit/vertexai/genai/test_datasets.py

@@ -0,0 +1,56 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+"""Tests for multimodal datasets."""
+
+from vertexai._genai import types
+
+
+class TestMultimodalDataset:
+    def test_set_read_config(self):
+        dataset = types.MultimodalDataset()
+
+        dataset.set_read_config(
+            read_config={
+                "assembled_request_column_name": "test_column",
+            },
+        )
+
+        assert isinstance(dataset, types.MultimodalDataset)
+        assert (
+            dataset.metadata.gemini_request_read_config.assembled_request_column_name
+            == "test_column"
+        )
+
+    def test_set_read_config_preserves_other_fields(self):
+        dataset = types.MultimodalDataset(
+            metadata={
+                "inputConfig": {
+                    "bigquerySource": {"uri": "bq://test_table"},
+                },
+            },
+        )
+
+        dataset.set_read_config(
+            read_config={
+                "assembled_request_column_name": "test_column",
+            },
+        )
+
+        assert isinstance(dataset, types.MultimodalDataset)
+        assert (
+            dataset.metadata.gemini_request_read_config.assembled_request_column_name
+            == "test_column"
+        )
+        assert dataset.metadata.input_config.bigquery_source.uri == "bq://test_table"

vertexai/_genai/types/common.py

@@ -13087,6 +13087,18 @@ class MultimodalDataset(_common.BaseModel):
         default=None, description="""The description of the multimodal dataset."""
     )
 
+    def set_read_config(
+        self,
+        *,
+        read_config: GeminiRequestReadConfigOrDict,
+    ) -> None:
+        if isinstance(read_config, dict):
+            read_config = GeminiRequestReadConfig(**read_config)
+
+        if self.metadata is None:
+            self.metadata = SchemaTablesDatasetMetadata()
+        self.metadata.gemini_request_read_config = read_config
+
 
 class MultimodalDatasetDict(TypedDict, total=False):
     """Represents a multimodal dataset."""