get_last_job interface

Author: guptadivyank

ads/feature_store/feature_group.py

@@ -750,12 +750,8 @@ def get_features_df(self) -> "pd.DataFrame":
         """
         records = []
         for feature in self.features:
-            records.append(
-                {
-                    "name": feature.feature_name,
-                    "type": feature.feature_type
-                }
-            )
+            records.append({"name": feature.feature_name, "type": feature.feature_type})
+
         return pd.DataFrame.from_records(records)
 
     def get_input_features_df(self) -> "pd.DataFrame":
@@ -926,10 +922,18 @@ def get_last_job(self) -> "FeatureGroupJob":
             )
 
         if not self.job_id:
-            raise ValueError(
-                "Associated jobs cannot be retrieved before calling 'materialise' or 'delete'."
+            fg_job = FeatureGroupJob.list(
+                feature_group_id=self.id,
+                compartment_id=self.compartment_id,
+                sort_by="timeCreated",
+                limit="1",
             )
-
+            if not fg_job:
+                raise ValueError(
+                    "Associated jobs cannot be retrieved before calling 'materialise' or 'delete'."
+                )
+            self.with_job_id(fg_job[0].id)
+            return fg_job[0]
         return FeatureGroupJob.from_id(self.job_id)
 
     def select(self, features: Optional[List[str]] = []) -> Query:

ads/feature_store/mixin/oci_feature_store.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python
 # -*- coding: utf-8; -*-
+from ads.common.decorator.utils import class_or_instance_method
 
 # Copyright (c) 2023 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
@@ -77,3 +78,44 @@ def init_client(
     @property
     def client(self) -> oci.feature_store.feature_store_client.FeatureStoreClient:
         return super().client
+
+    @class_or_instance_method
+    def list_resource(
+        cls, compartment_id: str = None, limit: int = 0, **kwargs
+    ) -> list:
+        """Generic method to list OCI resources
+
+        Parameters
+        ----------
+        compartment_id : str
+            Compartment ID of the OCI resources. Defaults to None.
+            If compartment_id is not specified,
+            the value of NB_SESSION_COMPARTMENT_OCID in environment variable will be used.
+        limit : int
+            The maximum number of items to return. Defaults to 0, All items will be returned
+        **kwargs :
+            Additional keyword arguments to filter the resource.
+            The kwargs are passed into OCI API.
+
+        Returns
+        -------
+        list
+            A list of OCI resources
+
+        Raises
+        ------
+        NotImplementedError
+            List method is not supported or implemented.
+
+        """
+        if limit:
+            items = cls._find_oci_method("list")(
+                cls.check_compartment_id(compartment_id), limit=limit, **kwargs
+            ).data.items
+        else:
+            items = oci.pagination.list_call_get_all_results(
+                cls._find_oci_method("list"),
+                cls.check_compartment_id(compartment_id),
+                **kwargs,
+            ).data
+        return [cls.from_oci_model(item) for item in items]