databricks-sdk 0.23.0__py3-none-any.whl → 0.24.0__py3-none-any.whl

databricks/sdk/service/serving.py

@@ -35,6 +35,51 @@ class Ai21LabsConfig:
         return cls(ai21labs_api_key=d.get('ai21labs_api_key', None))
 
 
+@dataclass
+class AmazonBedrockConfig:
+    aws_region: str
+    """The AWS region to use. Bedrock has to be enabled there."""
+
+    aws_access_key_id: str
+    """The Databricks secret key reference for an AWS Access Key ID with permissions to interact with
+    Bedrock services."""
+
+    aws_secret_access_key: str
+    """The Databricks secret key reference for an AWS Secret Access Key paired with the access key ID,
+    with permissions to interact with Bedrock services."""
+
+    bedrock_provider: AmazonBedrockConfigBedrockProvider
+    """The underlying provider in Amazon Bedrock. Supported values (case insensitive) include:
+    Anthropic, Cohere, AI21Labs, Amazon."""
+
+    def as_dict(self) -> dict:
+        """Serializes the AmazonBedrockConfig into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.aws_access_key_id is not None: body['aws_access_key_id'] = self.aws_access_key_id
+        if self.aws_region is not None: body['aws_region'] = self.aws_region
+        if self.aws_secret_access_key is not None: body['aws_secret_access_key'] = self.aws_secret_access_key
+        if self.bedrock_provider is not None: body['bedrock_provider'] = self.bedrock_provider.value
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> AmazonBedrockConfig:
+        """Deserializes the AmazonBedrockConfig from a dictionary."""
+        return cls(aws_access_key_id=d.get('aws_access_key_id', None),
+                   aws_region=d.get('aws_region', None),
+                   aws_secret_access_key=d.get('aws_secret_access_key', None),
+                   bedrock_provider=_enum(d, 'bedrock_provider', AmazonBedrockConfigBedrockProvider))
+
+
+class AmazonBedrockConfigBedrockProvider(Enum):
+    """The underlying provider in Amazon Bedrock. Supported values (case insensitive) include:
+    Anthropic, Cohere, AI21Labs, Amazon."""
+
+    AI21LABS = 'ai21labs'
+    AMAZON = 'amazon'
+    ANTHROPIC = 'anthropic'
+    COHERE = 'cohere'
+
+
 @dataclass
 class AnthropicConfig:
     anthropic_api_key: str
@@ -243,51 +288,6 @@ class AutoCaptureState:
         return cls(payload_table=_from_dict(d, 'payload_table', PayloadTable))
 
 
-@dataclass
-class AwsBedrockConfig:
-    aws_region: str
-    """The AWS region to use. Bedrock has to be enabled there."""
-
-    aws_access_key_id: str
-    """The Databricks secret key reference for an AWS Access Key ID with permissions to interact with
-    Bedrock services."""
-
-    aws_secret_access_key: str
-    """The Databricks secret key reference for an AWS Secret Access Key paired with the access key ID,
-    with permissions to interact with Bedrock services."""
-
-    bedrock_provider: AwsBedrockConfigBedrockProvider
-    """The underlying provider in AWS Bedrock. Supported values (case insensitive) include: Anthropic,
-    Cohere, AI21Labs, Amazon."""
-
-    def as_dict(self) -> dict:
-        """Serializes the AwsBedrockConfig into a dictionary suitable for use as a JSON request body."""
-        body = {}
-        if self.aws_access_key_id is not None: body['aws_access_key_id'] = self.aws_access_key_id
-        if self.aws_region is not None: body['aws_region'] = self.aws_region
-        if self.aws_secret_access_key is not None: body['aws_secret_access_key'] = self.aws_secret_access_key
-        if self.bedrock_provider is not None: body['bedrock_provider'] = self.bedrock_provider.value
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> AwsBedrockConfig:
-        """Deserializes the AwsBedrockConfig from a dictionary."""
-        return cls(aws_access_key_id=d.get('aws_access_key_id', None),
-                   aws_region=d.get('aws_region', None),
-                   aws_secret_access_key=d.get('aws_secret_access_key', None),
-                   bedrock_provider=_enum(d, 'bedrock_provider', AwsBedrockConfigBedrockProvider))
-
-
-class AwsBedrockConfigBedrockProvider(Enum):
-    """The underlying provider in AWS Bedrock. Supported values (case insensitive) include: Anthropic,
-    Cohere, AI21Labs, Amazon."""
-
-    AI21LABS = 'ai21labs'
-    AMAZON = 'amazon'
-    ANTHROPIC = 'anthropic'
-    COHERE = 'cohere'
-
-
 @dataclass
 class BuildLogsResponse:
     logs: str
@@ -789,7 +789,7 @@ class ExportMetricsResponse:
 class ExternalModel:
     provider: ExternalModelProvider
     """The name of the provider for the external model. Currently, the supported providers are
-    'ai21labs', 'anthropic', 'aws-bedrock', 'cohere', 'databricks-model-serving', 'openai', and
+    'ai21labs', 'anthropic', 'amazon-bedrock', 'cohere', 'databricks-model-serving', 'openai', and
     'palm'.","""
 
     name: str
@@ -801,12 +801,12 @@ class ExternalModel:
     ai21labs_config: Optional[Ai21LabsConfig] = None
     """AI21Labs Config. Only required if the provider is 'ai21labs'."""
 
+    amazon_bedrock_config: Optional[AmazonBedrockConfig] = None
+    """Amazon Bedrock Config. Only required if the provider is 'amazon-bedrock'."""
+
     anthropic_config: Optional[AnthropicConfig] = None
     """Anthropic Config. Only required if the provider is 'anthropic'."""
 
-    aws_bedrock_config: Optional[AwsBedrockConfig] = None
-    """AWS Bedrock Config. Only required if the provider is 'aws-bedrock'."""
-
     cohere_config: Optional[CohereConfig] = None
     """Cohere Config. Only required if the provider is 'cohere'."""
 
@@ -823,8 +823,8 @@ class ExternalModel:
         """Serializes the ExternalModel into a dictionary suitable for use as a JSON request body."""
         body = {}
         if self.ai21labs_config: body['ai21labs_config'] = self.ai21labs_config.as_dict()
+        if self.amazon_bedrock_config: body['amazon_bedrock_config'] = self.amazon_bedrock_config.as_dict()
         if self.anthropic_config: body['anthropic_config'] = self.anthropic_config.as_dict()
-        if self.aws_bedrock_config: body['aws_bedrock_config'] = self.aws_bedrock_config.as_dict()
         if self.cohere_config: body['cohere_config'] = self.cohere_config.as_dict()
         if self.databricks_model_serving_config:
             body['databricks_model_serving_config'] = self.databricks_model_serving_config.as_dict()
@@ -839,8 +839,8 @@ class ExternalModel:
     def from_dict(cls, d: Dict[str, any]) -> ExternalModel:
         """Deserializes the ExternalModel from a dictionary."""
         return cls(ai21labs_config=_from_dict(d, 'ai21labs_config', Ai21LabsConfig),
+                   amazon_bedrock_config=_from_dict(d, 'amazon_bedrock_config', AmazonBedrockConfig),
                    anthropic_config=_from_dict(d, 'anthropic_config', AnthropicConfig),
-                   aws_bedrock_config=_from_dict(d, 'aws_bedrock_config', AwsBedrockConfig),
                    cohere_config=_from_dict(d, 'cohere_config', CohereConfig),
                    databricks_model_serving_config=_from_dict(d, 'databricks_model_serving_config',
                                                               DatabricksModelServingConfig),
@@ -853,12 +853,12 @@ class ExternalModel:
 
 class ExternalModelProvider(Enum):
     """The name of the provider for the external model. Currently, the supported providers are
-    'ai21labs', 'anthropic', 'aws-bedrock', 'cohere', 'databricks-model-serving', 'openai', and
+    'ai21labs', 'anthropic', 'amazon-bedrock', 'cohere', 'databricks-model-serving', 'openai', and
     'palm'.","""
 
     AI21LABS = 'ai21labs'
+    AMAZON_BEDROCK = 'amazon-bedrock'
     ANTHROPIC = 'anthropic'
-    AWS_BEDROCK = 'aws-bedrock'
     COHERE = 'cohere'
     DATABRICKS_MODEL_SERVING = 'databricks-model-serving'
     OPENAI = 'openai'

databricks/sdk/service/workspace.py

@@ -71,10 +71,18 @@ class CreateCredentials:
     gitLabEnterpriseEdition and awsCodeCommit."""
 
     git_username: Optional[str] = None
-    """Git username."""
+    """The username or email provided with your Git provider account, depending on which provider you
+    are using. For GitHub, GitHub Enterprise Server, or Azure DevOps Services, either email or
+    username may be used. For GitLab, GitLab Enterprise Edition, email must be used. For AWS
+    CodeCommit, BitBucket or BitBucket Server, username must be used. For all other providers please
+    see your provider's Personal Access Token authentication documentation to see what is supported."""
 
     personal_access_token: Optional[str] = None
-    """The personal access token used to authenticate to the corresponding Git provider."""
+    """The personal access token used to authenticate to the corresponding Git provider. For certain
+    providers, support may exist for other types of scoped access tokens. [Learn more]. The personal
+    access token used to authenticate to the corresponding Git
+    
+    [Learn more]: https://docs.databricks.com/repos/get-access-tokens-from-git-provider.html"""
 
     def as_dict(self) -> dict:
         """Serializes the CreateCredentials into a dictionary suitable for use as a JSON request body."""
@@ -103,7 +111,11 @@ class CreateCredentialsResponse:
     gitLabEnterpriseEdition and awsCodeCommit."""
 
     git_username: Optional[str] = None
-    """Git username."""
+    """The username or email provided with your Git provider account, depending on which provider you
+    are using. For GitHub, GitHub Enterprise Server, or Azure DevOps Services, either email or
+    username may be used. For GitLab, GitLab Enterprise Edition, email must be used. For AWS
+    CodeCommit, BitBucket or BitBucket Server, username must be used. For all other providers please
+    see your provider's Personal Access Token authentication documentation to see what is supported."""
 
     def as_dict(self) -> dict:
         """Serializes the CreateCredentialsResponse into a dictionary suitable for use as a JSON request body."""
@@ -215,7 +227,11 @@ class CredentialInfo:
     gitLabEnterpriseEdition and awsCodeCommit."""
 
     git_username: Optional[str] = None
-    """Git username."""
+    """The username or email provided with your Git provider account, depending on which provider you
+    are using. For GitHub, GitHub Enterprise Server, or Azure DevOps Services, either email or
+    username may be used. For GitLab, GitLab Enterprise Edition, email must be used. For AWS
+    CodeCommit, BitBucket or BitBucket Server, username must be used. For all other providers please
+    see your provider's Personal Access Token authentication documentation to see what is supported."""
 
     def as_dict(self) -> dict:
         """Serializes the CredentialInfo into a dictionary suitable for use as a JSON request body."""
@@ -1171,10 +1187,18 @@ class UpdateCredentials:
     gitLabEnterpriseEdition and awsCodeCommit."""
 
     git_username: Optional[str] = None
-    """Git username."""
+    """The username or email provided with your Git provider account, depending on which provider you
+    are using. For GitHub, GitHub Enterprise Server, or Azure DevOps Services, either email or
+    username may be used. For GitLab, GitLab Enterprise Edition, email must be used. For AWS
+    CodeCommit, BitBucket or BitBucket Server, username must be used. For all other providers please
+    see your provider's Personal Access Token authentication documentation to see what is supported."""
 
     personal_access_token: Optional[str] = None
-    """The personal access token used to authenticate to the corresponding Git provider."""
+    """The personal access token used to authenticate to the corresponding Git provider. For certain
+    providers, support may exist for other types of scoped access tokens. [Learn more]. The personal
+    access token used to authenticate to the corresponding Git
+    
+    [Learn more]: https://docs.databricks.com/repos/get-access-tokens-from-git-provider.html"""
 
     def as_dict(self) -> dict:
         """Serializes the UpdateCredentials into a dictionary suitable for use as a JSON request body."""
@@ -1449,9 +1473,17 @@ class GitCredentialsAPI:
           bitbucketCloud, gitLab, azureDevOpsServices, gitHubEnterprise, bitbucketServer,
           gitLabEnterpriseEdition and awsCodeCommit.
         :param git_username: str (optional)
-          Git username.
+          The username or email provided with your Git provider account, depending on which provider you are
+          using. For GitHub, GitHub Enterprise Server, or Azure DevOps Services, either email or username may
+          be used. For GitLab, GitLab Enterprise Edition, email must be used. For AWS CodeCommit, BitBucket or
+          BitBucket Server, username must be used. For all other providers please see your provider's Personal
+          Access Token authentication documentation to see what is supported.
         :param personal_access_token: str (optional)
-          The personal access token used to authenticate to the corresponding Git provider.
+          The personal access token used to authenticate to the corresponding Git provider. For certain
+          providers, support may exist for other types of scoped access tokens. [Learn more]. The personal
+          access token used to authenticate to the corresponding Git
+          
+          [Learn more]: https://docs.databricks.com/repos/get-access-tokens-from-git-provider.html
         
         :returns: :class:`CreateCredentialsResponse`
         """
@@ -1526,9 +1558,17 @@ class GitCredentialsAPI:
           bitbucketCloud, gitLab, azureDevOpsServices, gitHubEnterprise, bitbucketServer,
           gitLabEnterpriseEdition and awsCodeCommit.
         :param git_username: str (optional)
-          Git username.
+          The username or email provided with your Git provider account, depending on which provider you are
+          using. For GitHub, GitHub Enterprise Server, or Azure DevOps Services, either email or username may
+          be used. For GitLab, GitLab Enterprise Edition, email must be used. For AWS CodeCommit, BitBucket or
+          BitBucket Server, username must be used. For all other providers please see your provider's Personal
+          Access Token authentication documentation to see what is supported.
         :param personal_access_token: str (optional)
-          The personal access token used to authenticate to the corresponding Git provider.
+          The personal access token used to authenticate to the corresponding Git provider. For certain
+          providers, support may exist for other types of scoped access tokens. [Learn more]. The personal
+          access token used to authenticate to the corresponding Git
+          
+          [Learn more]: https://docs.databricks.com/repos/get-access-tokens-from-git-provider.html
         
         
         """

databricks/sdk/version.py

@@ -1 +1 @@
-__version__ = '0.23.0'
+__version__ = '0.24.0'

{databricks_sdk-0.23.0.dist-info → databricks_sdk-0.24.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: databricks-sdk
-Version: 0.23.0
+Version: 0.24.0
 Summary: Databricks SDK for Python (Beta)
 Home-page: https://databricks-sdk-py.readthedocs.io
 Author: Serge Smertin

{databricks_sdk-0.23.0.dist-info → databricks_sdk-0.24.0.dist-info}/RECORD

@@ -11,7 +11,7 @@ databricks/sdk/environments.py,sha256=gStDfgI07ECd6Pb82Rf-nRjf48NH6hOY3UfTXm4YNZ
 databricks/sdk/oauth.py,sha256=jqe0yrrTUfRL8kpR21Odwn4R_X6Ns-hTLu3dKYDI1EM,18313
 databricks/sdk/py.typed,sha256=pSvaHpbY1UPNEXyVFUjlgBhjPFZMmVC_UNrPC7eMOHI,74
 databricks/sdk/retries.py,sha256=WgLh12bwdBc6fCQlaig3kKu18cVhPzFDGsspvq629Ew,2454
-databricks/sdk/version.py,sha256=pwluWaTPL8Xkb9G3PEN8_juXZgyuXodBFmQvwlw9pb4,23
+databricks/sdk/version.py,sha256=lbgaUA9yL_X3wnFgvRc5nIlmJ4gFm6Ala6TbPHN-5aY,23
 databricks/sdk/_widgets/__init__.py,sha256=Qm3JB8LmdPgEn_-VgxKkodTO4gn6OdaDPwsYcDmeIRI,2667
 databricks/sdk/_widgets/default_widgets_utils.py,sha256=Rk59AFzVYVpOektB_yC_7j-vSt5OdtZA85IlG0kw0xA,1202
 databricks/sdk/_widgets/ipywidgets_utils.py,sha256=P-AyGeahPiX3S59mxpAMgffi4gyJ0irEOY7Ekkn9nQ0,2850
@@ -27,27 +27,27 @@ databricks/sdk/mixins/workspace.py,sha256=dWMNvuEi8jJ5wMhrDt1LiqxNdWSsmEuDTzrcZR
 databricks/sdk/runtime/__init__.py,sha256=9NnZkBzeZXZRQxcE1qKzAszQEzcpIgpL7lQzW3_kxEU,7266
 databricks/sdk/runtime/dbutils_stub.py,sha256=UFbRZF-bBcwxjbv_pxma00bjNtktLLaYpo8oHRc4-9g,11421
 databricks/sdk/service/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-databricks/sdk/service/_internal.py,sha256=VvKT8sYgF8aVYp-nxS2s8QYc8GhhQvI3IcFJZJx1g50,1841
+databricks/sdk/service/_internal.py,sha256=nWbJfW5eJCQgAZ3TmA26xoWb6SNZ5N76ZA8bO1N4AsU,1961
 databricks/sdk/service/billing.py,sha256=Hbe5bMsBrpebuAl8yj-GwVRGktrzKwiZJj3gq1wUMaI,50625
 databricks/sdk/service/catalog.py,sha256=EytdKjPNqN6uhmUMRnQFJu5xaKP2xbQJ9GYsmnhRZLs,402330
 databricks/sdk/service/compute.py,sha256=fEDhQfFiycLWpOBz6ALhLXGvnNYVFsZvMlPm0j0lXpU,395632
-databricks/sdk/service/dashboards.py,sha256=9wsQNgrOxrZ3kBSyOfbWaWid9WJgozAu1qCVJ0H4YVM,15660
-databricks/sdk/service/files.py,sha256=myqpOOxwIaGwcepukUgV7QDhhZ-pHJg-SDvHU4NFX30,37599
-databricks/sdk/service/iam.py,sha256=Obbtlzl5G1Sd-ShmTDQq93ewDRzXbTH4D_lgNLS46uY,147465
-databricks/sdk/service/jobs.py,sha256=BmWC-6_xHZjQmM0BQAlObfe2BHzBsD0nxyS5OykRfCs,285521
+databricks/sdk/service/dashboards.py,sha256=PwhX73El3POXdblc7ZOm2PAkhf5TcSZ5Na73_ne2Zb4,18801
+databricks/sdk/service/files.py,sha256=VCt83YSI9rhQexmxaQdrUXHq2UCYfZcDMLvJx5X6n1M,38162
+databricks/sdk/service/iam.py,sha256=3Pe8ZpP413ys2GsaCEbvEKvv1eEji8OeuKrnGWbtb1E,147612
+databricks/sdk/service/jobs.py,sha256=Pi0tkKKIYqtWwKTucT2ZE0bsh4A4oPPuRQwwq7p8ock,304889
 databricks/sdk/service/ml.py,sha256=vohBdESClI3EOpO-ZZ44W-CMz1alq5Tw4oJnWa99Z2M,236128
 databricks/sdk/service/oauth2.py,sha256=zpEA7glY_EsPvMgkk-hmt4eVgrmtcSGgduI7XlShNUo,36215
 databricks/sdk/service/pipelines.py,sha256=2Kum7q_4hVZcEaww2c0NLJj0Fbe9G0BGsF71JKDCvv8,101546
 databricks/sdk/service/provisioning.py,sha256=DP4Df4X-p0JEUk4zAJQhjX_wxpMi673OKLXFhxl6YSE,142678
-databricks/sdk/service/serving.py,sha256=kRwmGuLbUZnxUdEYIxMkS2q1iOd0kw2SV6gkdo4v1HY,130549
+databricks/sdk/service/serving.py,sha256=TuF7GAN2w_IUWf-ymx1g1f6uf1g3ot83EBZn52edqbI,130618
 databricks/sdk/service/settings.py,sha256=7KITQbOPbaESxsCssckPdxLAS64auzfij1DLTRd9cQI,176630
 databricks/sdk/service/sharing.py,sha256=21BdjeHQTWvAfkoYfMBwj0hGYGOxRoGGYdz3M-20xV4,98520
 databricks/sdk/service/sql.py,sha256=FMbmD7h1ZrH5-28-fxt7A0o5KZYphenYzFQbzS8vVEw,256266
 databricks/sdk/service/vectorsearch.py,sha256=R5RACGGpM9w2yUylBTiV5Bk-YC6O2OorKlNUHqKFawg,51588
-databricks/sdk/service/workspace.py,sha256=Cc_fgKbZ5phzdjg7ryM-6uoVQ_ug3h4rhzQJFyk0RBM,96837
-databricks_sdk-0.23.0.dist-info/LICENSE,sha256=afBgTZo-JsYqj4VOjnejBetMuHKcFR30YobDdpVFkqY,11411
-databricks_sdk-0.23.0.dist-info/METADATA,sha256=NSpDj9c0We0p3P5PtXyoQ_3H4KSyPIZZ6Qq54i-vark,34642
-databricks_sdk-0.23.0.dist-info/NOTICE,sha256=Qnc0m8JjZNTDV80y0h1aJGvsr4GqM63m1nr2VTypg6E,963
-databricks_sdk-0.23.0.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-databricks_sdk-0.23.0.dist-info/top_level.txt,sha256=7kRdatoSgU0EUurRQJ_3F1Nv4EOSHWAr6ng25tJOJKU,11
-databricks_sdk-0.23.0.dist-info/RECORD,,
+databricks/sdk/service/workspace.py,sha256=4BZLbKu6yF9Jztb07uzUnEflA8JKOD9LIOQEJm5BK9c,100855
+databricks_sdk-0.24.0.dist-info/LICENSE,sha256=afBgTZo-JsYqj4VOjnejBetMuHKcFR30YobDdpVFkqY,11411
+databricks_sdk-0.24.0.dist-info/METADATA,sha256=yrgQWRk8_Wgb5AoiwfCXx32MFKu0SG4vN84-53lCGAo,34642
+databricks_sdk-0.24.0.dist-info/NOTICE,sha256=Qnc0m8JjZNTDV80y0h1aJGvsr4GqM63m1nr2VTypg6E,963
+databricks_sdk-0.24.0.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+databricks_sdk-0.24.0.dist-info/top_level.txt,sha256=7kRdatoSgU0EUurRQJ_3F1Nv4EOSHWAr6ng25tJOJKU,11
+databricks_sdk-0.24.0.dist-info/RECORD,,