pulumi-databricks 1.75.0a1756323569__py3-none-any.whl → 1.77.0__py3-none-any.whl

pulumi_databricks/get_jobs.py

@@ -13,6 +13,8 @@ if sys.version_info >= (3, 11):
 else:
     from typing_extensions import NotRequired, TypedDict, TypeAlias
 from . import _utilities
+from . import outputs
+from ._inputs import *
 
 __all__ = [
     'GetJobsResult',
@@ -26,7 +28,7 @@ class GetJobsResult:
     """
     A collection of values returned by getJobs.
     """
-    def __init__(__self__, id=None, ids=None, job_name_contains=None, key=None):
+    def __init__(__self__, id=None, ids=None, job_name_contains=None, key=None, provider_config=None):
         if id and not isinstance(id, str):
             raise TypeError("Expected argument 'id' to be a str")
         pulumi.set(__self__, "id", id)
@@ -39,6 +41,9 @@ class GetJobsResult:
         if key and not isinstance(key, str):
             raise TypeError("Expected argument 'key' to be a str")
         pulumi.set(__self__, "key", key)
+        if provider_config and not isinstance(provider_config, dict):
+            raise TypeError("Expected argument 'provider_config' to be a dict")
+        pulumi.set(__self__, "provider_config", provider_config)
 
     @_builtins.property
     @pulumi.getter
@@ -66,6 +71,11 @@ class GetJobsResult:
     def key(self) -> Optional[_builtins.str]:
         return pulumi.get(self, "key")
 
+    @_builtins.property
+    @pulumi.getter(name="providerConfig")
+    def provider_config(self) -> Optional['outputs.GetJobsProviderConfigResult']:
+        return pulumi.get(self, "provider_config")
+
 
 class AwaitableGetJobsResult(GetJobsResult):
     # pylint: disable=using-constant-test
@@ -76,12 +86,14 @@ class AwaitableGetJobsResult(GetJobsResult):
             id=self.id,
             ids=self.ids,
             job_name_contains=self.job_name_contains,
-            key=self.key)
+            key=self.key,
+            provider_config=self.provider_config)
 
 
 def get_jobs(ids: Optional[Mapping[str, _builtins.str]] = None,
              job_name_contains: Optional[_builtins.str] = None,
              key: Optional[_builtins.str] = None,
+             provider_config: Optional[Union['GetJobsProviderConfigArgs', 'GetJobsProviderConfigArgsDict']] = None,
              opts: Optional[pulumi.InvokeOptions] = None) -> AwaitableGetJobsResult:
     """
     Retrieves a list of Job ids, that were created by Pulumi or manually, so that special handling could be applied.
@@ -146,11 +158,13 @@ def get_jobs(ids: Optional[Mapping[str, _builtins.str]] = None,
     :param Mapping[str, _builtins.str] ids: map of Job names to ids
     :param _builtins.str job_name_contains: Only return Job ids that match the given name string (case-insensitive).
     :param _builtins.str key: Attribute to use for keys in the returned map of Job ids by. Possible values are `name` (default) or `id`. Setting to `id` uses the job ID as the map key, allowing duplicate job names.
+    :param Union['GetJobsProviderConfigArgs', 'GetJobsProviderConfigArgsDict'] provider_config: Configure the provider for management through account provider. This block consists of the following fields:
     """
     __args__ = dict()
     __args__['ids'] = ids
     __args__['jobNameContains'] = job_name_contains
     __args__['key'] = key
+    __args__['providerConfig'] = provider_config
     opts = pulumi.InvokeOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
     __ret__ = pulumi.runtime.invoke('databricks:index/getJobs:getJobs', __args__, opts=opts, typ=GetJobsResult).value
 
@@ -158,10 +172,12 @@ def get_jobs(ids: Optional[Mapping[str, _builtins.str]] = None,
         id=pulumi.get(__ret__, 'id'),
         ids=pulumi.get(__ret__, 'ids'),
         job_name_contains=pulumi.get(__ret__, 'job_name_contains'),
-        key=pulumi.get(__ret__, 'key'))
+        key=pulumi.get(__ret__, 'key'),
+        provider_config=pulumi.get(__ret__, 'provider_config'))
 def get_jobs_output(ids: Optional[pulumi.Input[Optional[Mapping[str, _builtins.str]]]] = None,
                     job_name_contains: Optional[pulumi.Input[Optional[_builtins.str]]] = None,
                     key: Optional[pulumi.Input[Optional[_builtins.str]]] = None,
+                    provider_config: Optional[pulumi.Input[Optional[Union['GetJobsProviderConfigArgs', 'GetJobsProviderConfigArgsDict']]]] = None,
                     opts: Optional[Union[pulumi.InvokeOptions, pulumi.InvokeOutputOptions]] = None) -> pulumi.Output[GetJobsResult]:
     """
     Retrieves a list of Job ids, that were created by Pulumi or manually, so that special handling could be applied.
@@ -226,15 +242,18 @@ def get_jobs_output(ids: Optional[pulumi.Input[Optional[Mapping[str, _builtins.s
     :param Mapping[str, _builtins.str] ids: map of Job names to ids
     :param _builtins.str job_name_contains: Only return Job ids that match the given name string (case-insensitive).
     :param _builtins.str key: Attribute to use for keys in the returned map of Job ids by. Possible values are `name` (default) or `id`. Setting to `id` uses the job ID as the map key, allowing duplicate job names.
+    :param Union['GetJobsProviderConfigArgs', 'GetJobsProviderConfigArgsDict'] provider_config: Configure the provider for management through account provider. This block consists of the following fields:
     """
     __args__ = dict()
     __args__['ids'] = ids
     __args__['jobNameContains'] = job_name_contains
     __args__['key'] = key
+    __args__['providerConfig'] = provider_config
     opts = pulumi.InvokeOutputOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
     __ret__ = pulumi.runtime.invoke_output('databricks:index/getJobs:getJobs', __args__, opts=opts, typ=GetJobsResult)
     return __ret__.apply(lambda __response__: GetJobsResult(
         id=pulumi.get(__response__, 'id'),
         ids=pulumi.get(__response__, 'ids'),
         job_name_contains=pulumi.get(__response__, 'job_name_contains'),
-        key=pulumi.get(__response__, 'key')))
+        key=pulumi.get(__response__, 'key'),
+        provider_config=pulumi.get(__response__, 'provider_config')))

pulumi_databricks/get_materialized_features_feature_tag.py

@@ -55,7 +55,7 @@ class GetMaterializedFeaturesFeatureTagResult:
 
     @_builtins.property
     @pulumi.getter
-    def value(self) -> Optional[_builtins.str]:
+    def value(self) -> _builtins.str:
         """
         (string)
         """
@@ -74,16 +74,12 @@ class AwaitableGetMaterializedFeaturesFeatureTagResult(GetMaterializedFeaturesFe
 
 
 def get_materialized_features_feature_tag(key: Optional[_builtins.str] = None,
-                                          value: Optional[_builtins.str] = None,
                                           opts: Optional[pulumi.InvokeOptions] = None) -> AwaitableGetMaterializedFeaturesFeatureTagResult:
     """
-    Use this data source to access information about an existing resource.
-
-    :param _builtins.str value: (string)
+    [![Private Preview](https://img.shields.io/badge/Release_Stage-Private_Preview-blueviolet)](https://docs.databricks.com/aws/en/release-notes/release-types)
     """
     __args__ = dict()
     __args__['key'] = key
-    __args__['value'] = value
     opts = pulumi.InvokeOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
     __ret__ = pulumi.runtime.invoke('databricks:index/getMaterializedFeaturesFeatureTag:getMaterializedFeaturesFeatureTag', __args__, opts=opts, typ=GetMaterializedFeaturesFeatureTagResult).value
 
@@ -92,16 +88,12 @@ def get_materialized_features_feature_tag(key: Optional[_builtins.str] = None,
         key=pulumi.get(__ret__, 'key'),
         value=pulumi.get(__ret__, 'value'))
 def get_materialized_features_feature_tag_output(key: Optional[pulumi.Input[_builtins.str]] = None,
-                                                 value: Optional[pulumi.Input[Optional[_builtins.str]]] = None,
                                                  opts: Optional[Union[pulumi.InvokeOptions, pulumi.InvokeOutputOptions]] = None) -> pulumi.Output[GetMaterializedFeaturesFeatureTagResult]:
     """
-    Use this data source to access information about an existing resource.
-
-    :param _builtins.str value: (string)
+    [![Private Preview](https://img.shields.io/badge/Release_Stage-Private_Preview-blueviolet)](https://docs.databricks.com/aws/en/release-notes/release-types)
     """
     __args__ = dict()
     __args__['key'] = key
-    __args__['value'] = value
     opts = pulumi.InvokeOutputOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
     __ret__ = pulumi.runtime.invoke_output('databricks:index/getMaterializedFeaturesFeatureTag:getMaterializedFeaturesFeatureTag', __args__, opts=opts, typ=GetMaterializedFeaturesFeatureTagResult)
     return __ret__.apply(lambda __response__: GetMaterializedFeaturesFeatureTagResult(

pulumi_databricks/get_materialized_features_feature_tags.py

@@ -27,13 +27,27 @@ class GetMaterializedFeaturesFeatureTagsResult:
     """
     A collection of values returned by getMaterializedFeaturesFeatureTags.
     """
-    def __init__(__self__, feature_tags=None, id=None):
+    def __init__(__self__, feature_name=None, feature_tags=None, id=None, page_size=None, table_name=None):
+        if feature_name and not isinstance(feature_name, str):
+            raise TypeError("Expected argument 'feature_name' to be a str")
+        pulumi.set(__self__, "feature_name", feature_name)
         if feature_tags and not isinstance(feature_tags, list):
             raise TypeError("Expected argument 'feature_tags' to be a list")
         pulumi.set(__self__, "feature_tags", feature_tags)
         if id and not isinstance(id, str):
             raise TypeError("Expected argument 'id' to be a str")
         pulumi.set(__self__, "id", id)
+        if page_size and not isinstance(page_size, int):
+            raise TypeError("Expected argument 'page_size' to be a int")
+        pulumi.set(__self__, "page_size", page_size)
+        if table_name and not isinstance(table_name, str):
+            raise TypeError("Expected argument 'table_name' to be a str")
+        pulumi.set(__self__, "table_name", table_name)
+
+    @_builtins.property
+    @pulumi.getter(name="featureName")
+    def feature_name(self) -> _builtins.str:
+        return pulumi.get(self, "feature_name")
 
     @_builtins.property
     @pulumi.getter(name="featureTags")
@@ -48,6 +62,16 @@ class GetMaterializedFeaturesFeatureTagsResult:
         """
         return pulumi.get(self, "id")
 
+    @_builtins.property
+    @pulumi.getter(name="pageSize")
+    def page_size(self) -> Optional[_builtins.int]:
+        return pulumi.get(self, "page_size")
+
+    @_builtins.property
+    @pulumi.getter(name="tableName")
+    def table_name(self) -> _builtins.str:
+        return pulumi.get(self, "table_name")
+
 
 class AwaitableGetMaterializedFeaturesFeatureTagsResult(GetMaterializedFeaturesFeatureTagsResult):
     # pylint: disable=using-constant-test
@@ -55,28 +79,55 @@ class AwaitableGetMaterializedFeaturesFeatureTagsResult(GetMaterializedFeaturesF
         if False:
             yield self
         return GetMaterializedFeaturesFeatureTagsResult(
+            feature_name=self.feature_name,
             feature_tags=self.feature_tags,
-            id=self.id)
+            id=self.id,
+            page_size=self.page_size,
+            table_name=self.table_name)
 
 
-def get_materialized_features_feature_tags(opts: Optional[pulumi.InvokeOptions] = None) -> AwaitableGetMaterializedFeaturesFeatureTagsResult:
+def get_materialized_features_feature_tags(feature_name: Optional[_builtins.str] = None,
+                                           page_size: Optional[_builtins.int] = None,
+                                           table_name: Optional[_builtins.str] = None,
+                                           opts: Optional[pulumi.InvokeOptions] = None) -> AwaitableGetMaterializedFeaturesFeatureTagsResult:
     """
-    Use this data source to access information about an existing resource.
+    [![Private Preview](https://img.shields.io/badge/Release_Stage-Private_Preview-blueviolet)](https://docs.databricks.com/aws/en/release-notes/release-types)
+
+
+    :param _builtins.int page_size: The maximum number of results to return
     """
     __args__ = dict()
+    __args__['featureName'] = feature_name
+    __args__['pageSize'] = page_size
+    __args__['tableName'] = table_name
     opts = pulumi.InvokeOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
     __ret__ = pulumi.runtime.invoke('databricks:index/getMaterializedFeaturesFeatureTags:getMaterializedFeaturesFeatureTags', __args__, opts=opts, typ=GetMaterializedFeaturesFeatureTagsResult).value
 
     return AwaitableGetMaterializedFeaturesFeatureTagsResult(
+        feature_name=pulumi.get(__ret__, 'feature_name'),
         feature_tags=pulumi.get(__ret__, 'feature_tags'),
-        id=pulumi.get(__ret__, 'id'))
-def get_materialized_features_feature_tags_output(opts: Optional[Union[pulumi.InvokeOptions, pulumi.InvokeOutputOptions]] = None) -> pulumi.Output[GetMaterializedFeaturesFeatureTagsResult]:
+        id=pulumi.get(__ret__, 'id'),
+        page_size=pulumi.get(__ret__, 'page_size'),
+        table_name=pulumi.get(__ret__, 'table_name'))
+def get_materialized_features_feature_tags_output(feature_name: Optional[pulumi.Input[_builtins.str]] = None,
+                                                  page_size: Optional[pulumi.Input[Optional[_builtins.int]]] = None,
+                                                  table_name: Optional[pulumi.Input[_builtins.str]] = None,
+                                                  opts: Optional[Union[pulumi.InvokeOptions, pulumi.InvokeOutputOptions]] = None) -> pulumi.Output[GetMaterializedFeaturesFeatureTagsResult]:
     """
-    Use this data source to access information about an existing resource.
+    [![Private Preview](https://img.shields.io/badge/Release_Stage-Private_Preview-blueviolet)](https://docs.databricks.com/aws/en/release-notes/release-types)
+
+
+    :param _builtins.int page_size: The maximum number of results to return
     """
     __args__ = dict()
+    __args__['featureName'] = feature_name
+    __args__['pageSize'] = page_size
+    __args__['tableName'] = table_name
     opts = pulumi.InvokeOutputOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
     __ret__ = pulumi.runtime.invoke_output('databricks:index/getMaterializedFeaturesFeatureTags:getMaterializedFeaturesFeatureTags', __args__, opts=opts, typ=GetMaterializedFeaturesFeatureTagsResult)
     return __ret__.apply(lambda __response__: GetMaterializedFeaturesFeatureTagsResult(
+        feature_name=pulumi.get(__response__, 'feature_name'),
         feature_tags=pulumi.get(__response__, 'feature_tags'),
-        id=pulumi.get(__response__, 'id')))
+        id=pulumi.get(__response__, 'id'),
+        page_size=pulumi.get(__response__, 'page_size'),
+        table_name=pulumi.get(__response__, 'table_name')))

pulumi_databricks/get_metastore.py

@@ -113,7 +113,7 @@ def get_metastore(id: Optional[_builtins.str] = None,
     import pulumi_aws as aws
     import pulumi_databricks as databricks
 
-    metastore = aws.s3.BucketV2("metastore",
+    metastore = aws.s3.Bucket("metastore",
         bucket=f"{prefix}-metastore",
         force_destroy=True)
     this_metastore = databricks.Metastore("this",
@@ -175,7 +175,7 @@ def get_metastore_output(id: Optional[pulumi.Input[Optional[_builtins.str]]] = N
     import pulumi_aws as aws
     import pulumi_databricks as databricks
 
-    metastore = aws.s3.BucketV2("metastore",
+    metastore = aws.s3.Bucket("metastore",
         bucket=f"{prefix}-metastore",
         force_destroy=True)
     this_metastore = databricks.Metastore("this",

pulumi_databricks/get_mlflow_experiment.py

@@ -28,7 +28,7 @@ class GetMlflowExperimentResult:
     """
     A collection of values returned by getMlflowExperiment.
     """
-    def __init__(__self__, artifact_location=None, creation_time=None, experiment_id=None, id=None, last_update_time=None, lifecycle_stage=None, name=None, tags=None):
+    def __init__(__self__, artifact_location=None, creation_time=None, experiment_id=None, id=None, last_update_time=None, lifecycle_stage=None, name=None, provider_config=None, tags=None):
         if artifact_location and not isinstance(artifact_location, str):
             raise TypeError("Expected argument 'artifact_location' to be a str")
         pulumi.set(__self__, "artifact_location", artifact_location)
@@ -50,6 +50,9 @@ class GetMlflowExperimentResult:
         if name and not isinstance(name, str):
             raise TypeError("Expected argument 'name' to be a str")
         pulumi.set(__self__, "name", name)
+        if provider_config and not isinstance(provider_config, dict):
+            raise TypeError("Expected argument 'provider_config' to be a dict")
+        pulumi.set(__self__, "provider_config", provider_config)
         if tags and not isinstance(tags, list):
             raise TypeError("Expected argument 'tags' to be a list")
         pulumi.set(__self__, "tags", tags)
@@ -110,6 +113,11 @@ class GetMlflowExperimentResult:
         """
         return pulumi.get(self, "name")
 
+    @_builtins.property
+    @pulumi.getter(name="providerConfig")
+    def provider_config(self) -> Optional['outputs.GetMlflowExperimentProviderConfigResult']:
+        return pulumi.get(self, "provider_config")
+
     @_builtins.property
     @pulumi.getter
     def tags(self) -> Sequence['outputs.GetMlflowExperimentTagResult']:
@@ -132,6 +140,7 @@ class AwaitableGetMlflowExperimentResult(GetMlflowExperimentResult):
             last_update_time=self.last_update_time,
             lifecycle_stage=self.lifecycle_stage,
             name=self.name,
+            provider_config=self.provider_config,
             tags=self.tags)
 
 
@@ -142,6 +151,7 @@ def get_mlflow_experiment(artifact_location: Optional[_builtins.str] = None,
                           last_update_time: Optional[_builtins.int] = None,
                           lifecycle_stage: Optional[_builtins.str] = None,
                           name: Optional[_builtins.str] = None,
+                          provider_config: Optional[Union['GetMlflowExperimentProviderConfigArgs', 'GetMlflowExperimentProviderConfigArgsDict']] = None,
                           tags: Optional[Sequence[Union['GetMlflowExperimentTagArgs', 'GetMlflowExperimentTagArgsDict']]] = None,
                           opts: Optional[pulumi.InvokeOptions] = None) -> AwaitableGetMlflowExperimentResult:
     """
@@ -157,6 +167,7 @@ def get_mlflow_experiment(artifact_location: Optional[_builtins.str] = None,
     :param _builtins.int last_update_time: Last update time in unix time stamp.
     :param _builtins.str lifecycle_stage: Current life cycle stage of the experiment: `active` or `deleted`.
     :param _builtins.str name: Path to experiment.
+    :param Union['GetMlflowExperimentProviderConfigArgs', 'GetMlflowExperimentProviderConfigArgsDict'] provider_config: Configure the provider for management through account provider. This block consists of the following fields:
     :param Sequence[Union['GetMlflowExperimentTagArgs', 'GetMlflowExperimentTagArgsDict']] tags: Additional metadata key-value pairs.
     """
     __args__ = dict()
@@ -167,6 +178,7 @@ def get_mlflow_experiment(artifact_location: Optional[_builtins.str] = None,
     __args__['lastUpdateTime'] = last_update_time
     __args__['lifecycleStage'] = lifecycle_stage
     __args__['name'] = name
+    __args__['providerConfig'] = provider_config
     __args__['tags'] = tags
     opts = pulumi.InvokeOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
     __ret__ = pulumi.runtime.invoke('databricks:index/getMlflowExperiment:getMlflowExperiment', __args__, opts=opts, typ=GetMlflowExperimentResult).value
@@ -179,6 +191,7 @@ def get_mlflow_experiment(artifact_location: Optional[_builtins.str] = None,
         last_update_time=pulumi.get(__ret__, 'last_update_time'),
         lifecycle_stage=pulumi.get(__ret__, 'lifecycle_stage'),
         name=pulumi.get(__ret__, 'name'),
+        provider_config=pulumi.get(__ret__, 'provider_config'),
         tags=pulumi.get(__ret__, 'tags'))
 def get_mlflow_experiment_output(artifact_location: Optional[pulumi.Input[Optional[_builtins.str]]] = None,
                                  creation_time: Optional[pulumi.Input[Optional[_builtins.int]]] = None,
@@ -187,6 +200,7 @@ def get_mlflow_experiment_output(artifact_location: Optional[pulumi.Input[Option
                                  last_update_time: Optional[pulumi.Input[Optional[_builtins.int]]] = None,
                                  lifecycle_stage: Optional[pulumi.Input[Optional[_builtins.str]]] = None,
                                  name: Optional[pulumi.Input[Optional[_builtins.str]]] = None,
+                                 provider_config: Optional[pulumi.Input[Optional[Union['GetMlflowExperimentProviderConfigArgs', 'GetMlflowExperimentProviderConfigArgsDict']]]] = None,
                                  tags: Optional[pulumi.Input[Optional[Sequence[Union['GetMlflowExperimentTagArgs', 'GetMlflowExperimentTagArgsDict']]]]] = None,
                                  opts: Optional[Union[pulumi.InvokeOptions, pulumi.InvokeOutputOptions]] = None) -> pulumi.Output[GetMlflowExperimentResult]:
     """
@@ -202,6 +216,7 @@ def get_mlflow_experiment_output(artifact_location: Optional[pulumi.Input[Option
     :param _builtins.int last_update_time: Last update time in unix time stamp.
     :param _builtins.str lifecycle_stage: Current life cycle stage of the experiment: `active` or `deleted`.
     :param _builtins.str name: Path to experiment.
+    :param Union['GetMlflowExperimentProviderConfigArgs', 'GetMlflowExperimentProviderConfigArgsDict'] provider_config: Configure the provider for management through account provider. This block consists of the following fields:
     :param Sequence[Union['GetMlflowExperimentTagArgs', 'GetMlflowExperimentTagArgsDict']] tags: Additional metadata key-value pairs.
     """
     __args__ = dict()
@@ -212,6 +227,7 @@ def get_mlflow_experiment_output(artifact_location: Optional[pulumi.Input[Option
     __args__['lastUpdateTime'] = last_update_time
     __args__['lifecycleStage'] = lifecycle_stage
     __args__['name'] = name
+    __args__['providerConfig'] = provider_config
     __args__['tags'] = tags
     opts = pulumi.InvokeOutputOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
     __ret__ = pulumi.runtime.invoke_output('databricks:index/getMlflowExperiment:getMlflowExperiment', __args__, opts=opts, typ=GetMlflowExperimentResult)
@@ -223,4 +239,5 @@ def get_mlflow_experiment_output(artifact_location: Optional[pulumi.Input[Option
         last_update_time=pulumi.get(__response__, 'last_update_time'),
         lifecycle_stage=pulumi.get(__response__, 'lifecycle_stage'),
         name=pulumi.get(__response__, 'name'),
+        provider_config=pulumi.get(__response__, 'provider_config'),
         tags=pulumi.get(__response__, 'tags')))

pulumi_databricks/get_mlflow_model.py

@@ -28,7 +28,7 @@ class GetMlflowModelResult:
     """
     A collection of values returned by getMlflowModel.
     """
-    def __init__(__self__, description=None, id=None, latest_versions=None, name=None, permission_level=None, tags=None, user_id=None):
+    def __init__(__self__, description=None, id=None, latest_versions=None, name=None, permission_level=None, provider_config=None, tags=None, user_id=None):
         if description and not isinstance(description, str):
             raise TypeError("Expected argument 'description' to be a str")
         pulumi.set(__self__, "description", description)
@@ -44,6 +44,9 @@ class GetMlflowModelResult:
         if permission_level and not isinstance(permission_level, str):
             raise TypeError("Expected argument 'permission_level' to be a str")
         pulumi.set(__self__, "permission_level", permission_level)
+        if provider_config and not isinstance(provider_config, dict):
+            raise TypeError("Expected argument 'provider_config' to be a dict")
+        pulumi.set(__self__, "provider_config", provider_config)
         if tags and not isinstance(tags, list):
             raise TypeError("Expected argument 'tags' to be a list")
         pulumi.set(__self__, "tags", tags)
@@ -91,6 +94,11 @@ class GetMlflowModelResult:
         """
         return pulumi.get(self, "permission_level")
 
+    @_builtins.property
+    @pulumi.getter(name="providerConfig")
+    def provider_config(self) -> Optional['outputs.GetMlflowModelProviderConfigResult']:
+        return pulumi.get(self, "provider_config")
+
     @_builtins.property
     @pulumi.getter
     def tags(self) -> Sequence['outputs.GetMlflowModelTagResult']:
@@ -119,6 +127,7 @@ class AwaitableGetMlflowModelResult(GetMlflowModelResult):
             latest_versions=self.latest_versions,
             name=self.name,
             permission_level=self.permission_level,
+            provider_config=self.provider_config,
             tags=self.tags,
             user_id=self.user_id)
 
@@ -127,6 +136,7 @@ def get_mlflow_model(description: Optional[_builtins.str] = None,
                      latest_versions: Optional[Sequence[Union['GetMlflowModelLatestVersionArgs', 'GetMlflowModelLatestVersionArgsDict']]] = None,
                      name: Optional[_builtins.str] = None,
                      permission_level: Optional[_builtins.str] = None,
+                     provider_config: Optional[Union['GetMlflowModelProviderConfigArgs', 'GetMlflowModelProviderConfigArgsDict']] = None,
                      tags: Optional[Sequence[Union['GetMlflowModelTagArgs', 'GetMlflowModelTagArgsDict']]] = None,
                      user_id: Optional[_builtins.str] = None,
                      opts: Optional[pulumi.InvokeOptions] = None) -> AwaitableGetMlflowModelResult:
@@ -181,6 +191,7 @@ def get_mlflow_model(description: Optional[_builtins.str] = None,
     :param Sequence[Union['GetMlflowModelLatestVersionArgs', 'GetMlflowModelLatestVersionArgsDict']] latest_versions: Array of model versions, each the latest version for its stage.
     :param _builtins.str name: Name of the registered model.
     :param _builtins.str permission_level: Permission level of the requesting user on the object. For what is allowed at each level, see MLflow Model permissions.
+    :param Union['GetMlflowModelProviderConfigArgs', 'GetMlflowModelProviderConfigArgsDict'] provider_config: Configure the provider for management through account provider. This block consists of the following fields:
     :param Sequence[Union['GetMlflowModelTagArgs', 'GetMlflowModelTagArgsDict']] tags: Array of tags associated with the model.
     :param _builtins.str user_id: The username of the user that created the object.
     """
@@ -189,6 +200,7 @@ def get_mlflow_model(description: Optional[_builtins.str] = None,
     __args__['latestVersions'] = latest_versions
     __args__['name'] = name
     __args__['permissionLevel'] = permission_level
+    __args__['providerConfig'] = provider_config
     __args__['tags'] = tags
     __args__['userId'] = user_id
     opts = pulumi.InvokeOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
@@ -200,12 +212,14 @@ def get_mlflow_model(description: Optional[_builtins.str] = None,
         latest_versions=pulumi.get(__ret__, 'latest_versions'),
         name=pulumi.get(__ret__, 'name'),
         permission_level=pulumi.get(__ret__, 'permission_level'),
+        provider_config=pulumi.get(__ret__, 'provider_config'),
         tags=pulumi.get(__ret__, 'tags'),
         user_id=pulumi.get(__ret__, 'user_id'))
 def get_mlflow_model_output(description: Optional[pulumi.Input[Optional[_builtins.str]]] = None,
                             latest_versions: Optional[pulumi.Input[Optional[Sequence[Union['GetMlflowModelLatestVersionArgs', 'GetMlflowModelLatestVersionArgsDict']]]]] = None,
                             name: Optional[pulumi.Input[_builtins.str]] = None,
                             permission_level: Optional[pulumi.Input[Optional[_builtins.str]]] = None,
+                            provider_config: Optional[pulumi.Input[Optional[Union['GetMlflowModelProviderConfigArgs', 'GetMlflowModelProviderConfigArgsDict']]]] = None,
                             tags: Optional[pulumi.Input[Optional[Sequence[Union['GetMlflowModelTagArgs', 'GetMlflowModelTagArgsDict']]]]] = None,
                             user_id: Optional[pulumi.Input[Optional[_builtins.str]]] = None,
                             opts: Optional[Union[pulumi.InvokeOptions, pulumi.InvokeOutputOptions]] = None) -> pulumi.Output[GetMlflowModelResult]:
@@ -260,6 +274,7 @@ def get_mlflow_model_output(description: Optional[pulumi.Input[Optional[_builtin
     :param Sequence[Union['GetMlflowModelLatestVersionArgs', 'GetMlflowModelLatestVersionArgsDict']] latest_versions: Array of model versions, each the latest version for its stage.
     :param _builtins.str name: Name of the registered model.
     :param _builtins.str permission_level: Permission level of the requesting user on the object. For what is allowed at each level, see MLflow Model permissions.
+    :param Union['GetMlflowModelProviderConfigArgs', 'GetMlflowModelProviderConfigArgsDict'] provider_config: Configure the provider for management through account provider. This block consists of the following fields:
     :param Sequence[Union['GetMlflowModelTagArgs', 'GetMlflowModelTagArgsDict']] tags: Array of tags associated with the model.
     :param _builtins.str user_id: The username of the user that created the object.
     """
@@ -268,6 +283,7 @@ def get_mlflow_model_output(description: Optional[pulumi.Input[Optional[_builtin
     __args__['latestVersions'] = latest_versions
     __args__['name'] = name
     __args__['permissionLevel'] = permission_level
+    __args__['providerConfig'] = provider_config
     __args__['tags'] = tags
     __args__['userId'] = user_id
     opts = pulumi.InvokeOutputOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
@@ -278,5 +294,6 @@ def get_mlflow_model_output(description: Optional[pulumi.Input[Optional[_builtin
         latest_versions=pulumi.get(__response__, 'latest_versions'),
         name=pulumi.get(__response__, 'name'),
         permission_level=pulumi.get(__response__, 'permission_level'),
+        provider_config=pulumi.get(__response__, 'provider_config'),
         tags=pulumi.get(__response__, 'tags'),
         user_id=pulumi.get(__response__, 'user_id')))

pulumi_databricks/get_mlflow_models.py

@@ -13,6 +13,8 @@ if sys.version_info >= (3, 11):
 else:
     from typing_extensions import NotRequired, TypedDict, TypeAlias
 from . import _utilities
+from . import outputs
+from ._inputs import *
 
 __all__ = [
     'GetMlflowModelsResult',
@@ -26,13 +28,16 @@ class GetMlflowModelsResult:
     """
     A collection of values returned by getMlflowModels.
     """
-    def __init__(__self__, id=None, names=None):
+    def __init__(__self__, id=None, names=None, provider_config=None):
         if id and not isinstance(id, str):
             raise TypeError("Expected argument 'id' to be a str")
         pulumi.set(__self__, "id", id)
         if names and not isinstance(names, list):
             raise TypeError("Expected argument 'names' to be a list")
         pulumi.set(__self__, "names", names)
+        if provider_config and not isinstance(provider_config, dict):
+            raise TypeError("Expected argument 'provider_config' to be a dict")
+        pulumi.set(__self__, "provider_config", provider_config)
 
     @_builtins.property
     @pulumi.getter
@@ -50,6 +55,11 @@ class GetMlflowModelsResult:
         """
         return pulumi.get(self, "names")
 
+    @_builtins.property
+    @pulumi.getter(name="providerConfig")
+    def provider_config(self) -> Optional['outputs.GetMlflowModelsProviderConfigResult']:
+        return pulumi.get(self, "provider_config")
+
 
 class AwaitableGetMlflowModelsResult(GetMlflowModelsResult):
     # pylint: disable=using-constant-test
@@ -58,10 +68,12 @@ class AwaitableGetMlflowModelsResult(GetMlflowModelsResult):
             yield self
         return GetMlflowModelsResult(
             id=self.id,
-            names=self.names)
+            names=self.names,
+            provider_config=self.provider_config)
 
 
 def get_mlflow_models(names: Optional[Sequence[_builtins.str]] = None,
+                      provider_config: Optional[Union['GetMlflowModelsProviderConfigArgs', 'GetMlflowModelsProviderConfigArgsDict']] = None,
                       opts: Optional[pulumi.InvokeOptions] = None) -> AwaitableGetMlflowModelsResult:
     """
     Retrieves a list of MlflowModel objects, that were created by Pulumi or manually, so that special handling could be applied.
@@ -80,16 +92,20 @@ def get_mlflow_models(names: Optional[Sequence[_builtins.str]] = None,
 
 
     :param Sequence[_builtins.str] names: List of names of databricks_mlflow_model
+    :param Union['GetMlflowModelsProviderConfigArgs', 'GetMlflowModelsProviderConfigArgsDict'] provider_config: Configure the provider for management through account provider. This block consists of the following fields:
     """
     __args__ = dict()
     __args__['names'] = names
+    __args__['providerConfig'] = provider_config
     opts = pulumi.InvokeOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
     __ret__ = pulumi.runtime.invoke('databricks:index/getMlflowModels:getMlflowModels', __args__, opts=opts, typ=GetMlflowModelsResult).value
 
     return AwaitableGetMlflowModelsResult(
         id=pulumi.get(__ret__, 'id'),
-        names=pulumi.get(__ret__, 'names'))
+        names=pulumi.get(__ret__, 'names'),
+        provider_config=pulumi.get(__ret__, 'provider_config'))
 def get_mlflow_models_output(names: Optional[pulumi.Input[Optional[Sequence[_builtins.str]]]] = None,
+                             provider_config: Optional[pulumi.Input[Optional[Union['GetMlflowModelsProviderConfigArgs', 'GetMlflowModelsProviderConfigArgsDict']]]] = None,
                              opts: Optional[Union[pulumi.InvokeOptions, pulumi.InvokeOutputOptions]] = None) -> pulumi.Output[GetMlflowModelsResult]:
     """
     Retrieves a list of MlflowModel objects, that were created by Pulumi or manually, so that special handling could be applied.
@@ -108,11 +124,14 @@ def get_mlflow_models_output(names: Optional[pulumi.Input[Optional[Sequence[_bui
 
 
     :param Sequence[_builtins.str] names: List of names of databricks_mlflow_model
+    :param Union['GetMlflowModelsProviderConfigArgs', 'GetMlflowModelsProviderConfigArgsDict'] provider_config: Configure the provider for management through account provider. This block consists of the following fields:
     """
     __args__ = dict()
     __args__['names'] = names
+    __args__['providerConfig'] = provider_config
     opts = pulumi.InvokeOutputOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
     __ret__ = pulumi.runtime.invoke_output('databricks:index/getMlflowModels:getMlflowModels', __args__, opts=opts, typ=GetMlflowModelsResult)
     return __ret__.apply(lambda __response__: GetMlflowModelsResult(
         id=pulumi.get(__response__, 'id'),
-        names=pulumi.get(__response__, 'names')))
+        names=pulumi.get(__response__, 'names'),
+        provider_config=pulumi.get(__response__, 'provider_config')))