feat: Add has_template_config to MultimodalDataset.

Frances Hubis Thoma · copybara-github · commit 97a6e719c410 · 2025-09-18T13:49:28.000-07:00
PiperOrigin-RevId: 808720000
diff --git a/google/cloud/aiplatform/preview/datasets.py b/google/cloud/aiplatform/preview/datasets.py
@@ -1288,6 +1288,11 @@ def attach_template_config(
         )
         return self
 
+    def has_template_config(self) -> bool:
+        """Returns true if the dataset has a template config attached."""
+        self._assert_gca_resource_is_available()
+        return _GEMINI_TEMPLATE_CONFIG_SOURCE_FIELD in self._gca_resource.metadata
+
     @property
     def template_config(self) -> Optional[GeminiTemplateConfig]:
         """Return a copy of the template config attached to this dataset."""
diff --git a/tests/unit/aiplatform/test_multimodal_datasets.py b/tests/unit/aiplatform/test_multimodal_datasets.py
@@ -749,6 +749,18 @@ def test_attach_template_config_with_prompt(
             timeout=None,
         )
 
+    @pytest.mark.usefixtures("get_dataset_mock")
+    def test_has_template_config(self, update_dataset_with_template_config_mock):
+        aiplatform.init(project=_TEST_PROJECT)
+        dataset = ummd.MultimodalDataset(dataset_name=_TEST_NAME)
+        template_config = ummd.GeminiTemplateConfig(
+            field_mapping={"question": "questionColumn"},
+        )
+        assert dataset.has_template_config() is False
+        # Attach a template config to the dataset.
+        dataset.attach_template_config(template_config=template_config)
+        assert dataset.has_template_config() is True
+
     @pytest.mark.usefixtures(
         "get_dataset_with_prompt_resource_mock", "prompts_get_mock"
     )
