added supported static method in ResourcePrincipal

Author: govarsha

ads/common/auth.py

@@ -687,6 +687,10 @@ def create_signer(self) -> Dict:
  logger.info(f"Using 'resource_principal' authentication.")
  return signer_dict
 
+ @staticmethod
+ def supported():
+ return any(os.environ.get(var) for var in ['JOB_RUN_OCID', 'NB_SESSION_OCID', 'DATAFLOW_RUN_ID'])
+
 
 class InstancePrincipal(AuthSignerGenerator):
  """

ads/opctl/config/merger.py

@@ -11,7 +11,7 @@
 
 import yaml
 
-from ads.common.auth import AuthType
+from ads.common.auth import AuthType, ResourcePrincipal
 from ads.opctl import logger
 from ads.opctl.config.base import ConfigProcessor
 from ads.opctl.config.utils import read_from_ini, _DefaultNoneDict
@@ -115,7 +115,7 @@ def _fill_config_with_defaults(self, ads_config_path: str) -> None:
  )
  # set default auth
  if not self.config["execution"].get("auth", None):
- if is_in_notebook_session() or is_in_jobrun():
+ if ResourcePrincipal.supported():
  self.config["execution"]["auth"] = (
  exec_config.get("auth") or AuthType.RESOURCE_PRINCIPAL
  )