feat: Add List FeatureMonitor function to FeatureGroup in Vertex AI SDK

PiperOrigin-RevId: 698239121

Author: vertex-sdk-bot

tests/unit/vertexai/feature_store_constants.py

@@ -393,3 +393,26 @@
 )
 _TEST_FG1_FM1_FEATURE_SELECTION_CONFIGS = [("my_fg1_f1", 0.3), ("my_fg1_f2", 0.4)]
 _TEST_FG1_FM1_SCHEDULE_CONFIG = "0 0 * * *"
+_TEST_FG1_FM2_ID = "my_fg1_fm2"
+_TEST_FG1_FM2_PATH = (
+    f"{_TEST_PARENT}/featureGroups/{_TEST_FG1_ID}/featureMonitors/{_TEST_FG1_FM2_ID}"
+)
+_TEST_FG1_FM2_DESCRIPTION = "My feature monitor 2 in feature group 1"
+_TEST_FG1_FM2_LABELS = {"my_fg1_feature_monitor": "fm2"}
+_TEST_FG1_FM2_FEATURE_SELECTION_CONFIGS = [("my_fg1_f2", 0.5)]
+_TEST_FG1_FM2_SCHEDULE_CONFIG = "8 0 * * *"
+_TEST_FG1_FM2 = types.feature_monitor.FeatureMonitor(
+    name=_TEST_FG1_FM2_PATH,
+    description=_TEST_FG1_FM2_DESCRIPTION,
+    labels=_TEST_FG1_FM2_LABELS,
+    schedule_config=types.feature_monitor.ScheduleConfig(cron="8 0 * * *"),
+    feature_selection_config=types.feature_monitor.FeatureSelectionConfig(
+        feature_configs=[
+            types.feature_monitor.FeatureSelectionConfig.FeatureConfig(
+                feature_id="my_fg1_f2",
+                drift_threshold=0.5,
+            ),
+        ]
+    ),
+)
+_TEST_FG1_FM_LIST = [_TEST_FG1_FM1, _TEST_FG1_FM2]

tests/unit/vertexai/test_feature_group.py

@@ -85,6 +85,13 @@
     _TEST_FG1_FM1_LABELS,
     _TEST_FG1_FM1_FEATURE_SELECTION_CONFIGS,
     _TEST_FG1_FM1_SCHEDULE_CONFIG,
+    _TEST_FG1_FM2_ID,
+    _TEST_FG1_FM2_PATH,
+    _TEST_FG1_FM2_DESCRIPTION,
+    _TEST_FG1_FM2_LABELS,
+    _TEST_FG1_FM2_FEATURE_SELECTION_CONFIGS,
+    _TEST_FG1_FM2_SCHEDULE_CONFIG,
+    _TEST_FG1_FM_LIST,
 )
 from test_feature import feature_eq
 from test_feature_monitor import feature_monitor_eq
@@ -182,6 +189,16 @@ def list_features_mock():
         yield list_features_mock
 
 
+@pytest.fixture
+def list_feature_monitors_mock():
+    with patch.object(
+        FeatureRegistryServiceClient,
+        "list_feature_monitors",
+    ) as list_feature_monitors_mock:
+        list_feature_monitors_mock.return_value = _TEST_FG1_FM_LIST
+        yield list_feature_monitors_mock
+
+
 @pytest.fixture()
 def mock_base_instantiate_client():
     with patch.object(
@@ -929,3 +946,36 @@ def test_create_feature_monitor(
             ),
         ]
     )
+
+
+def test_list_feature_monitors(get_fg_mock, list_feature_monitors_mock):
+    aiplatform.init(project=_TEST_PROJECT, location=_TEST_LOCATION)
+
+    feature_monitors = FeatureGroup(_TEST_FG1_ID).list_feature_monitors()
+
+    list_feature_monitors_mock.assert_called_once_with(
+        request={"parent": _TEST_FG1_PATH}
+    )
+    assert len(feature_monitors) == len(_TEST_FG1_FM_LIST)
+    feature_monitor_eq(
+        feature_monitors[0],
+        name=_TEST_FG1_FM1_ID,
+        resource_name=_TEST_FG1_FM1_PATH,
+        project=_TEST_PROJECT,
+        location=_TEST_LOCATION,
+        description=_TEST_FG1_FM1_DESCRIPTION,
+        labels=_TEST_FG1_FM1_LABELS,
+        schedule_config=_TEST_FG1_FM1_SCHEDULE_CONFIG,
+        feature_selection_configs=_TEST_FG1_FM1_FEATURE_SELECTION_CONFIGS,
+    )
+    feature_monitor_eq(
+        feature_monitors[1],
+        name=_TEST_FG1_FM2_ID,
+        resource_name=_TEST_FG1_FM2_PATH,
+        project=_TEST_PROJECT,
+        location=_TEST_LOCATION,
+        description=_TEST_FG1_FM2_DESCRIPTION,
+        labels=_TEST_FG1_FM2_LABELS,
+        schedule_config=_TEST_FG1_FM2_SCHEDULE_CONFIG,
+        feature_selection_configs=_TEST_FG1_FM2_FEATURE_SELECTION_CONFIGS,
+    )

vertexai/resources/preview/feature_store/feature_group.py

@@ -544,6 +544,36 @@ def create_feature_monitor(
 
         return feature_monitor_obj
 
+    def list_feature_monitors(
+        self,
+        project: Optional[str] = None,
+        location: Optional[str] = None,
+        credentials: Optional[auth_credentials.Credentials] = None,
+    ) -> List[FeatureMonitor]:
+        """Lists features monitors under this feature group.
+
+        Args:
+            project:
+                Project to list feature monitors in. If unset, the project set in
+                aiplatform.init will be used.
+            location:
+                Location to list feature monitors in. If not set, location set in
+                aiplatform.init will be used.
+            credentials:
+                Custom credentials to use to list feature monitors. Overrides
+                credentials set in aiplatform.init.
+
+        Returns:
+            List of feature monitors under this feature group.
+        """
+
+        return FeatureMonitor.list(
+            parent=self.resource_name,
+            project=project,
+            location=location,
+            credentials=credentials,
+        )
+
     @property
     def source(self) -> FeatureGroupBigQuerySource:
         return FeatureGroupBigQuerySource(