databricks-sdk 0.63.0__py3-none-any.whl → 0.65.0__py3-none-any.whl

databricks/sdk/service/ml.py

@@ -1210,51 +1210,6 @@ class ExperimentTag:
         return cls(key=d.get("key", None), value=d.get("value", None))
 
 
-@dataclass
-class Feature:
-    """Feature for model version."""
-
-    feature_name: Optional[str] = None
-    """Feature name"""
-
-    feature_table_id: Optional[str] = None
-    """Feature table id"""
-
-    feature_table_name: Optional[str] = None
-    """Feature table name"""
-
-    def as_dict(self) -> dict:
-        """Serializes the Feature into a dictionary suitable for use as a JSON request body."""
-        body = {}
-        if self.feature_name is not None:
-            body["feature_name"] = self.feature_name
-        if self.feature_table_id is not None:
-            body["feature_table_id"] = self.feature_table_id
-        if self.feature_table_name is not None:
-            body["feature_table_name"] = self.feature_table_name
-        return body
-
-    def as_shallow_dict(self) -> dict:
-        """Serializes the Feature into a shallow dictionary of its immediate attributes."""
-        body = {}
-        if self.feature_name is not None:
-            body["feature_name"] = self.feature_name
-        if self.feature_table_id is not None:
-            body["feature_table_id"] = self.feature_table_id
-        if self.feature_table_name is not None:
-            body["feature_table_name"] = self.feature_table_name
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, Any]) -> Feature:
-        """Deserializes the Feature from a dictionary."""
-        return cls(
-            feature_name=d.get("feature_name", None),
-            feature_table_id=d.get("feature_table_id", None),
-            feature_table_name=d.get("feature_table_name", None),
-        )
-
-
 @dataclass
 class FeatureLineage:
     feature_specs: Optional[List[FeatureLineageFeatureSpec]] = None
@@ -1391,7 +1346,7 @@ class FeatureLineageOnlineFeature:
 class FeatureList:
     """Feature list wrap all the features for a model version"""
 
-    features: Optional[List[Feature]] = None
+    features: Optional[List[LinkedFeature]] = None
 
     def as_dict(self) -> dict:
         """Serializes the FeatureList into a dictionary suitable for use as a JSON request body."""
@@ -1410,7 +1365,7 @@ class FeatureList:
     @classmethod
     def from_dict(cls, d: Dict[str, Any]) -> FeatureList:
         """Deserializes the FeatureList from a dictionary."""
-        return cls(features=_repeated_dict(d, "features", Feature))
+        return cls(features=_repeated_dict(d, "features", LinkedFeature))
 
 
 @dataclass
@@ -2054,6 +2009,51 @@ class JobSpecWithoutSecret:
         return cls(job_id=d.get("job_id", None), workspace_url=d.get("workspace_url", None))
 
 
+@dataclass
+class LinkedFeature:
+    """Feature for model version. ([ML-57150] Renamed from Feature to LinkedFeature)"""
+
+    feature_name: Optional[str] = None
+    """Feature name"""
+
+    feature_table_id: Optional[str] = None
+    """Feature table id"""
+
+    feature_table_name: Optional[str] = None
+    """Feature table name"""
+
+    def as_dict(self) -> dict:
+        """Serializes the LinkedFeature into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.feature_name is not None:
+            body["feature_name"] = self.feature_name
+        if self.feature_table_id is not None:
+            body["feature_table_id"] = self.feature_table_id
+        if self.feature_table_name is not None:
+            body["feature_table_name"] = self.feature_table_name
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the LinkedFeature into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.feature_name is not None:
+            body["feature_name"] = self.feature_name
+        if self.feature_table_id is not None:
+            body["feature_table_id"] = self.feature_table_id
+        if self.feature_table_name is not None:
+            body["feature_table_name"] = self.feature_table_name
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> LinkedFeature:
+        """Deserializes the LinkedFeature from a dictionary."""
+        return cls(
+            feature_name=d.get("feature_name", None),
+            feature_table_id=d.get("feature_table_id", None),
+            feature_table_name=d.get("feature_table_name", None),
+        )
+
+
 @dataclass
 class ListArtifactsResponse:
     files: Optional[List[FileInfo]] = None

databricks/sdk/service/pipelines.py

@@ -613,6 +613,9 @@ class IngestionPipelineDefinition:
     objects: Optional[List[IngestionConfig]] = None
     """Required. Settings specifying tables to replicate and the destination for the replicated tables."""
 
+    source_configurations: Optional[List[SourceConfig]] = None
+    """Top-level source configurations"""
+
     source_type: Optional[IngestionSourceType] = None
     """The type of the foreign source. The source type will be inferred from the source connection or
     ingestion gateway. This field is output only and will be ignored if provided."""
@@ -630,6 +633,8 @@ class IngestionPipelineDefinition:
             body["ingestion_gateway_id"] = self.ingestion_gateway_id
         if self.objects:
             body["objects"] = [v.as_dict() for v in self.objects]
+        if self.source_configurations:
+            body["source_configurations"] = [v.as_dict() for v in self.source_configurations]
         if self.source_type is not None:
             body["source_type"] = self.source_type.value
         if self.table_configuration:
@@ -645,6 +650,8 @@ class IngestionPipelineDefinition:
             body["ingestion_gateway_id"] = self.ingestion_gateway_id
         if self.objects:
             body["objects"] = self.objects
+        if self.source_configurations:
+            body["source_configurations"] = self.source_configurations
         if self.source_type is not None:
             body["source_type"] = self.source_type
         if self.table_configuration:
@@ -658,6 +665,7 @@ class IngestionPipelineDefinition:
             connection_name=d.get("connection_name", None),
             ingestion_gateway_id=d.get("ingestion_gateway_id", None),
             objects=_repeated_dict(d, "objects", IngestionConfig),
+            source_configurations=_repeated_dict(d, "source_configurations", SourceConfig),
             source_type=_enum(d, "source_type", IngestionSourceType),
             table_configuration=_from_dict(d, "table_configuration", TableSpecificConfig),
         )
@@ -2239,6 +2247,67 @@ class PipelinesEnvironment:
         return cls(dependencies=d.get("dependencies", None))
 
 
+@dataclass
+class PostgresCatalogConfig:
+    """PG-specific catalog-level configuration parameters"""
+
+    slot_config: Optional[PostgresSlotConfig] = None
+    """Optional. The Postgres slot configuration to use for logical replication"""
+
+    def as_dict(self) -> dict:
+        """Serializes the PostgresCatalogConfig into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.slot_config:
+            body["slot_config"] = self.slot_config.as_dict()
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the PostgresCatalogConfig into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.slot_config:
+            body["slot_config"] = self.slot_config
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> PostgresCatalogConfig:
+        """Deserializes the PostgresCatalogConfig from a dictionary."""
+        return cls(slot_config=_from_dict(d, "slot_config", PostgresSlotConfig))
+
+
+@dataclass
+class PostgresSlotConfig:
+    """PostgresSlotConfig contains the configuration for a Postgres logical replication slot"""
+
+    publication_name: Optional[str] = None
+    """The name of the publication to use for the Postgres source"""
+
+    slot_name: Optional[str] = None
+    """The name of the logical replication slot to use for the Postgres source"""
+
+    def as_dict(self) -> dict:
+        """Serializes the PostgresSlotConfig into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.publication_name is not None:
+            body["publication_name"] = self.publication_name
+        if self.slot_name is not None:
+            body["slot_name"] = self.slot_name
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the PostgresSlotConfig into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.publication_name is not None:
+            body["publication_name"] = self.publication_name
+        if self.slot_name is not None:
+            body["slot_name"] = self.slot_name
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> PostgresSlotConfig:
+        """Deserializes the PostgresSlotConfig from a dictionary."""
+        return cls(publication_name=d.get("publication_name", None), slot_name=d.get("slot_name", None))
+
+
 @dataclass
 class ReportSpec:
     source_url: str
@@ -2527,6 +2596,67 @@ class SerializedException:
         )
 
 
+@dataclass
+class SourceCatalogConfig:
+    """SourceCatalogConfig contains catalog-level custom configuration parameters for each source"""
+
+    postgres: Optional[PostgresCatalogConfig] = None
+    """Postgres-specific catalog-level configuration parameters"""
+
+    source_catalog: Optional[str] = None
+    """Source catalog name"""
+
+    def as_dict(self) -> dict:
+        """Serializes the SourceCatalogConfig into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.postgres:
+            body["postgres"] = self.postgres.as_dict()
+        if self.source_catalog is not None:
+            body["source_catalog"] = self.source_catalog
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the SourceCatalogConfig into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.postgres:
+            body["postgres"] = self.postgres
+        if self.source_catalog is not None:
+            body["source_catalog"] = self.source_catalog
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> SourceCatalogConfig:
+        """Deserializes the SourceCatalogConfig from a dictionary."""
+        return cls(
+            postgres=_from_dict(d, "postgres", PostgresCatalogConfig), source_catalog=d.get("source_catalog", None)
+        )
+
+
+@dataclass
+class SourceConfig:
+    catalog: Optional[SourceCatalogConfig] = None
+    """Catalog-level source configuration parameters"""
+
+    def as_dict(self) -> dict:
+        """Serializes the SourceConfig into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.catalog:
+            body["catalog"] = self.catalog.as_dict()
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the SourceConfig into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.catalog:
+            body["catalog"] = self.catalog
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> SourceConfig:
+        """Deserializes the SourceConfig from a dictionary."""
+        return cls(catalog=_from_dict(d, "catalog", SourceCatalogConfig))
+
+
 @dataclass
 class StackFrame:
     declaring_class: Optional[str] = None

databricks/sdk/service/serving.py

@@ -314,6 +314,9 @@ class AiGatewayRateLimit:
     """Principal field for a user, user group, or service principal to apply rate limiting to. Accepts
     a user email, group name, or service principal application ID."""
 
+    tokens: Optional[int] = None
+    """Used to specify how many tokens are allowed for a key within the renewal_period."""
+
     def as_dict(self) -> dict:
         """Serializes the AiGatewayRateLimit into a dictionary suitable for use as a JSON request body."""
         body = {}
@@ -325,6 +328,8 @@ class AiGatewayRateLimit:
             body["principal"] = self.principal
         if self.renewal_period is not None:
             body["renewal_period"] = self.renewal_period.value
+        if self.tokens is not None:
+            body["tokens"] = self.tokens
         return body
 
     def as_shallow_dict(self) -> dict:
@@ -338,6 +343,8 @@ class AiGatewayRateLimit:
             body["principal"] = self.principal
         if self.renewal_period is not None:
             body["renewal_period"] = self.renewal_period
+        if self.tokens is not None:
+            body["tokens"] = self.tokens
         return body
 
     @classmethod
@@ -348,6 +355,7 @@ class AiGatewayRateLimit:
             key=_enum(d, "key", AiGatewayRateLimitKey),
             principal=d.get("principal", None),
             renewal_period=_enum(d, "renewal_period", AiGatewayRateLimitRenewalPeriod),
+            tokens=d.get("tokens", None),
         )
 
 
@@ -3374,6 +3382,9 @@ class ServingEndpoint:
     task: Optional[str] = None
     """The task type of the serving endpoint."""
 
+    usage_policy_id: Optional[str] = None
+    """The usage policy associated with serving endpoint."""
+
     def as_dict(self) -> dict:
         """Serializes the ServingEndpoint into a dictionary suitable for use as a JSON request body."""
         body = {}
@@ -3401,6 +3412,8 @@ class ServingEndpoint:
             body["tags"] = [v.as_dict() for v in self.tags]
         if self.task is not None:
             body["task"] = self.task
+        if self.usage_policy_id is not None:
+            body["usage_policy_id"] = self.usage_policy_id
         return body
 
     def as_shallow_dict(self) -> dict:
@@ -3430,6 +3443,8 @@ class ServingEndpoint:
             body["tags"] = self.tags
         if self.task is not None:
             body["task"] = self.task
+        if self.usage_policy_id is not None:
+            body["usage_policy_id"] = self.usage_policy_id
         return body
 
     @classmethod
@@ -3448,6 +3463,7 @@ class ServingEndpoint:
             state=_from_dict(d, "state", EndpointState),
             tags=_repeated_dict(d, "tags", EndpointTag),
             task=d.get("task", None),
+            usage_policy_id=d.get("usage_policy_id", None),
         )
 
 

databricks/sdk/service/settings.py

@@ -3587,8 +3587,32 @@ class LlmProxyPartnerPoweredWorkspace:
 
 @dataclass
 class MicrosoftTeamsConfig:
+    app_id: Optional[str] = None
+    """[Input-Only] App ID for Microsoft Teams App."""
+
+    app_id_set: Optional[bool] = None
+    """[Output-Only] Whether App ID is set."""
+
+    auth_secret: Optional[str] = None
+    """[Input-Only] Secret for Microsoft Teams App authentication."""
+
+    auth_secret_set: Optional[bool] = None
+    """[Output-Only] Whether secret is set."""
+
+    channel_url: Optional[str] = None
+    """[Input-Only] Channel URL for Microsoft Teams App."""
+
+    channel_url_set: Optional[bool] = None
+    """[Output-Only] Whether Channel URL is set."""
+
+    tenant_id: Optional[str] = None
+    """[Input-Only] Tenant ID for Microsoft Teams App."""
+
+    tenant_id_set: Optional[bool] = None
+    """[Output-Only] Whether Tenant ID is set."""
+
     url: Optional[str] = None
-    """[Input-Only] URL for Microsoft Teams."""
+    """[Input-Only] URL for Microsoft Teams webhook."""
 
     url_set: Optional[bool] = None
     """[Output-Only] Whether URL is set."""
@@ -3596,6 +3620,22 @@ class MicrosoftTeamsConfig:
     def as_dict(self) -> dict:
         """Serializes the MicrosoftTeamsConfig into a dictionary suitable for use as a JSON request body."""
         body = {}
+        if self.app_id is not None:
+            body["app_id"] = self.app_id
+        if self.app_id_set is not None:
+            body["app_id_set"] = self.app_id_set
+        if self.auth_secret is not None:
+            body["auth_secret"] = self.auth_secret
+        if self.auth_secret_set is not None:
+            body["auth_secret_set"] = self.auth_secret_set
+        if self.channel_url is not None:
+            body["channel_url"] = self.channel_url
+        if self.channel_url_set is not None:
+            body["channel_url_set"] = self.channel_url_set
+        if self.tenant_id is not None:
+            body["tenant_id"] = self.tenant_id
+        if self.tenant_id_set is not None:
+            body["tenant_id_set"] = self.tenant_id_set
         if self.url is not None:
             body["url"] = self.url
         if self.url_set is not None:
@@ -3605,6 +3645,22 @@ class MicrosoftTeamsConfig:
     def as_shallow_dict(self) -> dict:
         """Serializes the MicrosoftTeamsConfig into a shallow dictionary of its immediate attributes."""
         body = {}
+        if self.app_id is not None:
+            body["app_id"] = self.app_id
+        if self.app_id_set is not None:
+            body["app_id_set"] = self.app_id_set
+        if self.auth_secret is not None:
+            body["auth_secret"] = self.auth_secret
+        if self.auth_secret_set is not None:
+            body["auth_secret_set"] = self.auth_secret_set
+        if self.channel_url is not None:
+            body["channel_url"] = self.channel_url
+        if self.channel_url_set is not None:
+            body["channel_url_set"] = self.channel_url_set
+        if self.tenant_id is not None:
+            body["tenant_id"] = self.tenant_id
+        if self.tenant_id_set is not None:
+            body["tenant_id_set"] = self.tenant_id_set
         if self.url is not None:
             body["url"] = self.url
         if self.url_set is not None:
@@ -3614,7 +3670,18 @@ class MicrosoftTeamsConfig:
     @classmethod
     def from_dict(cls, d: Dict[str, Any]) -> MicrosoftTeamsConfig:
         """Deserializes the MicrosoftTeamsConfig from a dictionary."""
-        return cls(url=d.get("url", None), url_set=d.get("url_set", None))
+        return cls(
+            app_id=d.get("app_id", None),
+            app_id_set=d.get("app_id_set", None),
+            auth_secret=d.get("auth_secret", None),
+            auth_secret_set=d.get("auth_secret_set", None),
+            channel_url=d.get("channel_url", None),
+            channel_url_set=d.get("channel_url_set", None),
+            tenant_id=d.get("tenant_id", None),
+            tenant_id_set=d.get("tenant_id_set", None),
+            url=d.get("url", None),
+            url_set=d.get("url_set", None),
+        )
 
 
 @dataclass