databricks-sdk 0.47.0__py3-none-any.whl → 0.48.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of databricks-sdk might be problematic. Click here for more details.
- databricks/sdk/service/catalog.py +0 -2
- databricks/sdk/service/compute.py +181 -376
- databricks/sdk/service/dashboards.py +0 -2
- databricks/sdk/service/iam.py +29 -12
- databricks/sdk/service/jobs.py +0 -1
- databricks/sdk/service/marketplace.py +0 -2
- databricks/sdk/service/ml.py +45 -20
- databricks/sdk/service/oauth2.py +0 -12
- databricks/sdk/service/pipelines.py +28 -25
- databricks/sdk/service/serving.py +0 -193
- databricks/sdk/service/sharing.py +71 -71
- databricks/sdk/version.py +1 -1
- {databricks_sdk-0.47.0.dist-info → databricks_sdk-0.48.0.dist-info}/METADATA +1 -1
- {databricks_sdk-0.47.0.dist-info → databricks_sdk-0.48.0.dist-info}/RECORD +18 -18
- {databricks_sdk-0.47.0.dist-info → databricks_sdk-0.48.0.dist-info}/WHEEL +1 -1
- {databricks_sdk-0.47.0.dist-info → databricks_sdk-0.48.0.dist-info}/licenses/LICENSE +0 -0
- {databricks_sdk-0.47.0.dist-info → databricks_sdk-0.48.0.dist-info}/licenses/NOTICE +0 -0
- {databricks_sdk-0.47.0.dist-info → databricks_sdk-0.48.0.dist-info}/top_level.txt +0 -0
|
@@ -63,10 +63,6 @@ class Ai21LabsConfig:
|
|
|
63
63
|
|
|
64
64
|
@dataclass
|
|
65
65
|
class AiGatewayConfig:
|
|
66
|
-
fallback_config: Optional[FallbackConfig] = None
|
|
67
|
-
"""Configuration for traffic fallback which auto fallbacks to other served entities if the request
|
|
68
|
-
to a served entity fails with certain error codes, to increase availability."""
|
|
69
|
-
|
|
70
66
|
guardrails: Optional[AiGatewayGuardrails] = None
|
|
71
67
|
"""Configuration for AI Guardrails to prevent unwanted data and unsafe data in requests and
|
|
72
68
|
responses."""
|
|
@@ -85,8 +81,6 @@ class AiGatewayConfig:
|
|
|
85
81
|
def as_dict(self) -> dict:
|
|
86
82
|
"""Serializes the AiGatewayConfig into a dictionary suitable for use as a JSON request body."""
|
|
87
83
|
body = {}
|
|
88
|
-
if self.fallback_config:
|
|
89
|
-
body["fallback_config"] = self.fallback_config.as_dict()
|
|
90
84
|
if self.guardrails:
|
|
91
85
|
body["guardrails"] = self.guardrails.as_dict()
|
|
92
86
|
if self.inference_table_config:
|
|
@@ -100,8 +94,6 @@ class AiGatewayConfig:
|
|
|
100
94
|
def as_shallow_dict(self) -> dict:
|
|
101
95
|
"""Serializes the AiGatewayConfig into a shallow dictionary of its immediate attributes."""
|
|
102
96
|
body = {}
|
|
103
|
-
if self.fallback_config:
|
|
104
|
-
body["fallback_config"] = self.fallback_config
|
|
105
97
|
if self.guardrails:
|
|
106
98
|
body["guardrails"] = self.guardrails
|
|
107
99
|
if self.inference_table_config:
|
|
@@ -116,7 +108,6 @@ class AiGatewayConfig:
|
|
|
116
108
|
def from_dict(cls, d: Dict[str, Any]) -> AiGatewayConfig:
|
|
117
109
|
"""Deserializes the AiGatewayConfig from a dictionary."""
|
|
118
110
|
return cls(
|
|
119
|
-
fallback_config=_from_dict(d, "fallback_config", FallbackConfig),
|
|
120
111
|
guardrails=_from_dict(d, "guardrails", AiGatewayGuardrails),
|
|
121
112
|
inference_table_config=_from_dict(d, "inference_table_config", AiGatewayInferenceTableConfig),
|
|
122
113
|
rate_limits=_repeated_dict(d, "rate_limits", AiGatewayRateLimit),
|
|
@@ -515,47 +506,6 @@ class AnthropicConfig:
|
|
|
515
506
|
)
|
|
516
507
|
|
|
517
508
|
|
|
518
|
-
@dataclass
|
|
519
|
-
class ApiKeyAuth:
|
|
520
|
-
key: str
|
|
521
|
-
"""The name of the API key parameter used for authentication."""
|
|
522
|
-
|
|
523
|
-
value: Optional[str] = None
|
|
524
|
-
"""The Databricks secret key reference for an API Key. If you prefer to paste your token directly,
|
|
525
|
-
see `value_plaintext`."""
|
|
526
|
-
|
|
527
|
-
value_plaintext: Optional[str] = None
|
|
528
|
-
"""The API Key provided as a plaintext string. If you prefer to reference your token using
|
|
529
|
-
Databricks Secrets, see `value`."""
|
|
530
|
-
|
|
531
|
-
def as_dict(self) -> dict:
|
|
532
|
-
"""Serializes the ApiKeyAuth into a dictionary suitable for use as a JSON request body."""
|
|
533
|
-
body = {}
|
|
534
|
-
if self.key is not None:
|
|
535
|
-
body["key"] = self.key
|
|
536
|
-
if self.value is not None:
|
|
537
|
-
body["value"] = self.value
|
|
538
|
-
if self.value_plaintext is not None:
|
|
539
|
-
body["value_plaintext"] = self.value_plaintext
|
|
540
|
-
return body
|
|
541
|
-
|
|
542
|
-
def as_shallow_dict(self) -> dict:
|
|
543
|
-
"""Serializes the ApiKeyAuth into a shallow dictionary of its immediate attributes."""
|
|
544
|
-
body = {}
|
|
545
|
-
if self.key is not None:
|
|
546
|
-
body["key"] = self.key
|
|
547
|
-
if self.value is not None:
|
|
548
|
-
body["value"] = self.value
|
|
549
|
-
if self.value_plaintext is not None:
|
|
550
|
-
body["value_plaintext"] = self.value_plaintext
|
|
551
|
-
return body
|
|
552
|
-
|
|
553
|
-
@classmethod
|
|
554
|
-
def from_dict(cls, d: Dict[str, Any]) -> ApiKeyAuth:
|
|
555
|
-
"""Deserializes the ApiKeyAuth from a dictionary."""
|
|
556
|
-
return cls(key=d.get("key", None), value=d.get("value", None), value_plaintext=d.get("value_plaintext", None))
|
|
557
|
-
|
|
558
|
-
|
|
559
509
|
@dataclass
|
|
560
510
|
class AutoCaptureConfigInput:
|
|
561
511
|
catalog_name: Optional[str] = None
|
|
@@ -695,40 +645,6 @@ class AutoCaptureState:
|
|
|
695
645
|
return cls(payload_table=_from_dict(d, "payload_table", PayloadTable))
|
|
696
646
|
|
|
697
647
|
|
|
698
|
-
@dataclass
|
|
699
|
-
class BearerTokenAuth:
|
|
700
|
-
token: Optional[str] = None
|
|
701
|
-
"""The Databricks secret key reference for a token. If you prefer to paste your token directly, see
|
|
702
|
-
`token_plaintext`."""
|
|
703
|
-
|
|
704
|
-
token_plaintext: Optional[str] = None
|
|
705
|
-
"""The token provided as a plaintext string. If you prefer to reference your token using Databricks
|
|
706
|
-
Secrets, see `token`."""
|
|
707
|
-
|
|
708
|
-
def as_dict(self) -> dict:
|
|
709
|
-
"""Serializes the BearerTokenAuth into a dictionary suitable for use as a JSON request body."""
|
|
710
|
-
body = {}
|
|
711
|
-
if self.token is not None:
|
|
712
|
-
body["token"] = self.token
|
|
713
|
-
if self.token_plaintext is not None:
|
|
714
|
-
body["token_plaintext"] = self.token_plaintext
|
|
715
|
-
return body
|
|
716
|
-
|
|
717
|
-
def as_shallow_dict(self) -> dict:
|
|
718
|
-
"""Serializes the BearerTokenAuth into a shallow dictionary of its immediate attributes."""
|
|
719
|
-
body = {}
|
|
720
|
-
if self.token is not None:
|
|
721
|
-
body["token"] = self.token
|
|
722
|
-
if self.token_plaintext is not None:
|
|
723
|
-
body["token_plaintext"] = self.token_plaintext
|
|
724
|
-
return body
|
|
725
|
-
|
|
726
|
-
@classmethod
|
|
727
|
-
def from_dict(cls, d: Dict[str, Any]) -> BearerTokenAuth:
|
|
728
|
-
"""Deserializes the BearerTokenAuth from a dictionary."""
|
|
729
|
-
return cls(token=d.get("token", None), token_plaintext=d.get("token_plaintext", None))
|
|
730
|
-
|
|
731
|
-
|
|
732
648
|
@dataclass
|
|
733
649
|
class BuildLogsResponse:
|
|
734
650
|
logs: str
|
|
@@ -920,53 +836,6 @@ class CreateServingEndpoint:
|
|
|
920
836
|
)
|
|
921
837
|
|
|
922
838
|
|
|
923
|
-
@dataclass
|
|
924
|
-
class CustomProviderConfig:
|
|
925
|
-
"""Configs needed to create a custom provider model route."""
|
|
926
|
-
|
|
927
|
-
custom_provider_url: str
|
|
928
|
-
"""This is a field to provide the URL of the custom provider API."""
|
|
929
|
-
|
|
930
|
-
api_key_auth: Optional[ApiKeyAuth] = None
|
|
931
|
-
"""This is a field to provide API key authentication for the custom provider API. You can only
|
|
932
|
-
specify one authentication method."""
|
|
933
|
-
|
|
934
|
-
bearer_token_auth: Optional[BearerTokenAuth] = None
|
|
935
|
-
"""This is a field to provide bearer token authentication for the custom provider API. You can only
|
|
936
|
-
specify one authentication method."""
|
|
937
|
-
|
|
938
|
-
def as_dict(self) -> dict:
|
|
939
|
-
"""Serializes the CustomProviderConfig into a dictionary suitable for use as a JSON request body."""
|
|
940
|
-
body = {}
|
|
941
|
-
if self.api_key_auth:
|
|
942
|
-
body["api_key_auth"] = self.api_key_auth.as_dict()
|
|
943
|
-
if self.bearer_token_auth:
|
|
944
|
-
body["bearer_token_auth"] = self.bearer_token_auth.as_dict()
|
|
945
|
-
if self.custom_provider_url is not None:
|
|
946
|
-
body["custom_provider_url"] = self.custom_provider_url
|
|
947
|
-
return body
|
|
948
|
-
|
|
949
|
-
def as_shallow_dict(self) -> dict:
|
|
950
|
-
"""Serializes the CustomProviderConfig into a shallow dictionary of its immediate attributes."""
|
|
951
|
-
body = {}
|
|
952
|
-
if self.api_key_auth:
|
|
953
|
-
body["api_key_auth"] = self.api_key_auth
|
|
954
|
-
if self.bearer_token_auth:
|
|
955
|
-
body["bearer_token_auth"] = self.bearer_token_auth
|
|
956
|
-
if self.custom_provider_url is not None:
|
|
957
|
-
body["custom_provider_url"] = self.custom_provider_url
|
|
958
|
-
return body
|
|
959
|
-
|
|
960
|
-
@classmethod
|
|
961
|
-
def from_dict(cls, d: Dict[str, Any]) -> CustomProviderConfig:
|
|
962
|
-
"""Deserializes the CustomProviderConfig from a dictionary."""
|
|
963
|
-
return cls(
|
|
964
|
-
api_key_auth=_from_dict(d, "api_key_auth", ApiKeyAuth),
|
|
965
|
-
bearer_token_auth=_from_dict(d, "bearer_token_auth", BearerTokenAuth),
|
|
966
|
-
custom_provider_url=d.get("custom_provider_url", None),
|
|
967
|
-
)
|
|
968
|
-
|
|
969
|
-
|
|
970
839
|
@dataclass
|
|
971
840
|
class DataPlaneInfo:
|
|
972
841
|
"""Details necessary to query this object's API through the DataPlane APIs."""
|
|
@@ -1626,9 +1495,6 @@ class ExternalModel:
|
|
|
1626
1495
|
cohere_config: Optional[CohereConfig] = None
|
|
1627
1496
|
"""Cohere Config. Only required if the provider is 'cohere'."""
|
|
1628
1497
|
|
|
1629
|
-
custom_provider_config: Optional[CustomProviderConfig] = None
|
|
1630
|
-
"""Custom Provider Config. Only required if the provider is 'custom'."""
|
|
1631
|
-
|
|
1632
1498
|
databricks_model_serving_config: Optional[DatabricksModelServingConfig] = None
|
|
1633
1499
|
"""Databricks Model Serving Config. Only required if the provider is 'databricks-model-serving'."""
|
|
1634
1500
|
|
|
@@ -1652,8 +1518,6 @@ class ExternalModel:
|
|
|
1652
1518
|
body["anthropic_config"] = self.anthropic_config.as_dict()
|
|
1653
1519
|
if self.cohere_config:
|
|
1654
1520
|
body["cohere_config"] = self.cohere_config.as_dict()
|
|
1655
|
-
if self.custom_provider_config:
|
|
1656
|
-
body["custom_provider_config"] = self.custom_provider_config.as_dict()
|
|
1657
1521
|
if self.databricks_model_serving_config:
|
|
1658
1522
|
body["databricks_model_serving_config"] = self.databricks_model_serving_config.as_dict()
|
|
1659
1523
|
if self.google_cloud_vertex_ai_config:
|
|
@@ -1681,8 +1545,6 @@ class ExternalModel:
|
|
|
1681
1545
|
body["anthropic_config"] = self.anthropic_config
|
|
1682
1546
|
if self.cohere_config:
|
|
1683
1547
|
body["cohere_config"] = self.cohere_config
|
|
1684
|
-
if self.custom_provider_config:
|
|
1685
|
-
body["custom_provider_config"] = self.custom_provider_config
|
|
1686
1548
|
if self.databricks_model_serving_config:
|
|
1687
1549
|
body["databricks_model_serving_config"] = self.databricks_model_serving_config
|
|
1688
1550
|
if self.google_cloud_vertex_ai_config:
|
|
@@ -1707,7 +1569,6 @@ class ExternalModel:
|
|
|
1707
1569
|
amazon_bedrock_config=_from_dict(d, "amazon_bedrock_config", AmazonBedrockConfig),
|
|
1708
1570
|
anthropic_config=_from_dict(d, "anthropic_config", AnthropicConfig),
|
|
1709
1571
|
cohere_config=_from_dict(d, "cohere_config", CohereConfig),
|
|
1710
|
-
custom_provider_config=_from_dict(d, "custom_provider_config", CustomProviderConfig),
|
|
1711
1572
|
databricks_model_serving_config=_from_dict(
|
|
1712
1573
|
d, "databricks_model_serving_config", DatabricksModelServingConfig
|
|
1713
1574
|
),
|
|
@@ -1726,7 +1587,6 @@ class ExternalModelProvider(Enum):
|
|
|
1726
1587
|
AMAZON_BEDROCK = "amazon-bedrock"
|
|
1727
1588
|
ANTHROPIC = "anthropic"
|
|
1728
1589
|
COHERE = "cohere"
|
|
1729
|
-
CUSTOM = "custom"
|
|
1730
1590
|
DATABRICKS_MODEL_SERVING = "databricks-model-serving"
|
|
1731
1591
|
GOOGLE_CLOUD_VERTEX_AI = "google-cloud-vertex-ai"
|
|
1732
1592
|
OPENAI = "openai"
|
|
@@ -1776,35 +1636,6 @@ class ExternalModelUsageElement:
|
|
|
1776
1636
|
)
|
|
1777
1637
|
|
|
1778
1638
|
|
|
1779
|
-
@dataclass
|
|
1780
|
-
class FallbackConfig:
|
|
1781
|
-
enabled: bool
|
|
1782
|
-
"""Whether to enable traffic fallback. When a served entity in the serving endpoint returns
|
|
1783
|
-
specific error codes (e.g. 500), the request will automatically be round-robin attempted with
|
|
1784
|
-
other served entities in the same endpoint, following the order of served entity list, until a
|
|
1785
|
-
successful response is returned. If all attempts fail, return the last response with the error
|
|
1786
|
-
code."""
|
|
1787
|
-
|
|
1788
|
-
def as_dict(self) -> dict:
|
|
1789
|
-
"""Serializes the FallbackConfig into a dictionary suitable for use as a JSON request body."""
|
|
1790
|
-
body = {}
|
|
1791
|
-
if self.enabled is not None:
|
|
1792
|
-
body["enabled"] = self.enabled
|
|
1793
|
-
return body
|
|
1794
|
-
|
|
1795
|
-
def as_shallow_dict(self) -> dict:
|
|
1796
|
-
"""Serializes the FallbackConfig into a shallow dictionary of its immediate attributes."""
|
|
1797
|
-
body = {}
|
|
1798
|
-
if self.enabled is not None:
|
|
1799
|
-
body["enabled"] = self.enabled
|
|
1800
|
-
return body
|
|
1801
|
-
|
|
1802
|
-
@classmethod
|
|
1803
|
-
def from_dict(cls, d: Dict[str, Any]) -> FallbackConfig:
|
|
1804
|
-
"""Deserializes the FallbackConfig from a dictionary."""
|
|
1805
|
-
return cls(enabled=d.get("enabled", None))
|
|
1806
|
-
|
|
1807
|
-
|
|
1808
1639
|
@dataclass
|
|
1809
1640
|
class FoundationModel:
|
|
1810
1641
|
"""All fields are not sensitive as they are hard-coded in the system and made available to
|
|
@@ -2293,10 +2124,6 @@ class PayloadTable:
|
|
|
2293
2124
|
|
|
2294
2125
|
@dataclass
|
|
2295
2126
|
class PutAiGatewayRequest:
|
|
2296
|
-
fallback_config: Optional[FallbackConfig] = None
|
|
2297
|
-
"""Configuration for traffic fallback which auto fallbacks to other served entities if the request
|
|
2298
|
-
to a served entity fails with certain error codes, to increase availability."""
|
|
2299
|
-
|
|
2300
2127
|
guardrails: Optional[AiGatewayGuardrails] = None
|
|
2301
2128
|
"""Configuration for AI Guardrails to prevent unwanted data and unsafe data in requests and
|
|
2302
2129
|
responses."""
|
|
@@ -2318,8 +2145,6 @@ class PutAiGatewayRequest:
|
|
|
2318
2145
|
def as_dict(self) -> dict:
|
|
2319
2146
|
"""Serializes the PutAiGatewayRequest into a dictionary suitable for use as a JSON request body."""
|
|
2320
2147
|
body = {}
|
|
2321
|
-
if self.fallback_config:
|
|
2322
|
-
body["fallback_config"] = self.fallback_config.as_dict()
|
|
2323
2148
|
if self.guardrails:
|
|
2324
2149
|
body["guardrails"] = self.guardrails.as_dict()
|
|
2325
2150
|
if self.inference_table_config:
|
|
@@ -2335,8 +2160,6 @@ class PutAiGatewayRequest:
|
|
|
2335
2160
|
def as_shallow_dict(self) -> dict:
|
|
2336
2161
|
"""Serializes the PutAiGatewayRequest into a shallow dictionary of its immediate attributes."""
|
|
2337
2162
|
body = {}
|
|
2338
|
-
if self.fallback_config:
|
|
2339
|
-
body["fallback_config"] = self.fallback_config
|
|
2340
2163
|
if self.guardrails:
|
|
2341
2164
|
body["guardrails"] = self.guardrails
|
|
2342
2165
|
if self.inference_table_config:
|
|
@@ -2353,7 +2176,6 @@ class PutAiGatewayRequest:
|
|
|
2353
2176
|
def from_dict(cls, d: Dict[str, Any]) -> PutAiGatewayRequest:
|
|
2354
2177
|
"""Deserializes the PutAiGatewayRequest from a dictionary."""
|
|
2355
2178
|
return cls(
|
|
2356
|
-
fallback_config=_from_dict(d, "fallback_config", FallbackConfig),
|
|
2357
2179
|
guardrails=_from_dict(d, "guardrails", AiGatewayGuardrails),
|
|
2358
2180
|
inference_table_config=_from_dict(d, "inference_table_config", AiGatewayInferenceTableConfig),
|
|
2359
2181
|
name=d.get("name", None),
|
|
@@ -2364,10 +2186,6 @@ class PutAiGatewayRequest:
|
|
|
2364
2186
|
|
|
2365
2187
|
@dataclass
|
|
2366
2188
|
class PutAiGatewayResponse:
|
|
2367
|
-
fallback_config: Optional[FallbackConfig] = None
|
|
2368
|
-
"""Configuration for traffic fallback which auto fallbacks to other served entities if the request
|
|
2369
|
-
to a served entity fails with certain error codes, to increase availability."""
|
|
2370
|
-
|
|
2371
2189
|
guardrails: Optional[AiGatewayGuardrails] = None
|
|
2372
2190
|
"""Configuration for AI Guardrails to prevent unwanted data and unsafe data in requests and
|
|
2373
2191
|
responses."""
|
|
@@ -2386,8 +2204,6 @@ class PutAiGatewayResponse:
|
|
|
2386
2204
|
def as_dict(self) -> dict:
|
|
2387
2205
|
"""Serializes the PutAiGatewayResponse into a dictionary suitable for use as a JSON request body."""
|
|
2388
2206
|
body = {}
|
|
2389
|
-
if self.fallback_config:
|
|
2390
|
-
body["fallback_config"] = self.fallback_config.as_dict()
|
|
2391
2207
|
if self.guardrails:
|
|
2392
2208
|
body["guardrails"] = self.guardrails.as_dict()
|
|
2393
2209
|
if self.inference_table_config:
|
|
@@ -2401,8 +2217,6 @@ class PutAiGatewayResponse:
|
|
|
2401
2217
|
def as_shallow_dict(self) -> dict:
|
|
2402
2218
|
"""Serializes the PutAiGatewayResponse into a shallow dictionary of its immediate attributes."""
|
|
2403
2219
|
body = {}
|
|
2404
|
-
if self.fallback_config:
|
|
2405
|
-
body["fallback_config"] = self.fallback_config
|
|
2406
2220
|
if self.guardrails:
|
|
2407
2221
|
body["guardrails"] = self.guardrails
|
|
2408
2222
|
if self.inference_table_config:
|
|
@@ -2417,7 +2231,6 @@ class PutAiGatewayResponse:
|
|
|
2417
2231
|
def from_dict(cls, d: Dict[str, Any]) -> PutAiGatewayResponse:
|
|
2418
2232
|
"""Deserializes the PutAiGatewayResponse from a dictionary."""
|
|
2419
2233
|
return cls(
|
|
2420
|
-
fallback_config=_from_dict(d, "fallback_config", FallbackConfig),
|
|
2421
2234
|
guardrails=_from_dict(d, "guardrails", AiGatewayGuardrails),
|
|
2422
2235
|
inference_table_config=_from_dict(d, "inference_table_config", AiGatewayInferenceTableConfig),
|
|
2423
2236
|
rate_limits=_repeated_dict(d, "rate_limits", AiGatewayRateLimit),
|
|
@@ -4556,7 +4369,6 @@ class ServingEndpointsAPI:
|
|
|
4556
4369
|
self,
|
|
4557
4370
|
name: str,
|
|
4558
4371
|
*,
|
|
4559
|
-
fallback_config: Optional[FallbackConfig] = None,
|
|
4560
4372
|
guardrails: Optional[AiGatewayGuardrails] = None,
|
|
4561
4373
|
inference_table_config: Optional[AiGatewayInferenceTableConfig] = None,
|
|
4562
4374
|
rate_limits: Optional[List[AiGatewayRateLimit]] = None,
|
|
@@ -4569,9 +4381,6 @@ class ServingEndpointsAPI:
|
|
|
4569
4381
|
|
|
4570
4382
|
:param name: str
|
|
4571
4383
|
The name of the serving endpoint whose AI Gateway is being updated. This field is required.
|
|
4572
|
-
:param fallback_config: :class:`FallbackConfig` (optional)
|
|
4573
|
-
Configuration for traffic fallback which auto fallbacks to other served entities if the request to a
|
|
4574
|
-
served entity fails with certain error codes, to increase availability.
|
|
4575
4384
|
:param guardrails: :class:`AiGatewayGuardrails` (optional)
|
|
4576
4385
|
Configuration for AI Guardrails to prevent unwanted data and unsafe data in requests and responses.
|
|
4577
4386
|
:param inference_table_config: :class:`AiGatewayInferenceTableConfig` (optional)
|
|
@@ -4586,8 +4395,6 @@ class ServingEndpointsAPI:
|
|
|
4586
4395
|
:returns: :class:`PutAiGatewayResponse`
|
|
4587
4396
|
"""
|
|
4588
4397
|
body = {}
|
|
4589
|
-
if fallback_config is not None:
|
|
4590
|
-
body["fallback_config"] = fallback_config.as_dict()
|
|
4591
4398
|
if guardrails is not None:
|
|
4592
4399
|
body["guardrails"] = guardrails.as_dict()
|
|
4593
4400
|
if inference_table_config is not None:
|
|
@@ -324,7 +324,71 @@ class DeltaSharingDependencyList:
|
|
|
324
324
|
|
|
325
325
|
|
|
326
326
|
@dataclass
|
|
327
|
-
class
|
|
327
|
+
class DeltaSharingFunctionDependency:
|
|
328
|
+
"""A Function in UC as a dependency."""
|
|
329
|
+
|
|
330
|
+
function_name: Optional[str] = None
|
|
331
|
+
|
|
332
|
+
schema_name: Optional[str] = None
|
|
333
|
+
|
|
334
|
+
def as_dict(self) -> dict:
|
|
335
|
+
"""Serializes the DeltaSharingFunctionDependency into a dictionary suitable for use as a JSON request body."""
|
|
336
|
+
body = {}
|
|
337
|
+
if self.function_name is not None:
|
|
338
|
+
body["function_name"] = self.function_name
|
|
339
|
+
if self.schema_name is not None:
|
|
340
|
+
body["schema_name"] = self.schema_name
|
|
341
|
+
return body
|
|
342
|
+
|
|
343
|
+
def as_shallow_dict(self) -> dict:
|
|
344
|
+
"""Serializes the DeltaSharingFunctionDependency into a shallow dictionary of its immediate attributes."""
|
|
345
|
+
body = {}
|
|
346
|
+
if self.function_name is not None:
|
|
347
|
+
body["function_name"] = self.function_name
|
|
348
|
+
if self.schema_name is not None:
|
|
349
|
+
body["schema_name"] = self.schema_name
|
|
350
|
+
return body
|
|
351
|
+
|
|
352
|
+
@classmethod
|
|
353
|
+
def from_dict(cls, d: Dict[str, Any]) -> DeltaSharingFunctionDependency:
|
|
354
|
+
"""Deserializes the DeltaSharingFunctionDependency from a dictionary."""
|
|
355
|
+
return cls(function_name=d.get("function_name", None), schema_name=d.get("schema_name", None))
|
|
356
|
+
|
|
357
|
+
|
|
358
|
+
@dataclass
|
|
359
|
+
class DeltaSharingTableDependency:
|
|
360
|
+
"""A Table in UC as a dependency."""
|
|
361
|
+
|
|
362
|
+
schema_name: Optional[str] = None
|
|
363
|
+
|
|
364
|
+
table_name: Optional[str] = None
|
|
365
|
+
|
|
366
|
+
def as_dict(self) -> dict:
|
|
367
|
+
"""Serializes the DeltaSharingTableDependency into a dictionary suitable for use as a JSON request body."""
|
|
368
|
+
body = {}
|
|
369
|
+
if self.schema_name is not None:
|
|
370
|
+
body["schema_name"] = self.schema_name
|
|
371
|
+
if self.table_name is not None:
|
|
372
|
+
body["table_name"] = self.table_name
|
|
373
|
+
return body
|
|
374
|
+
|
|
375
|
+
def as_shallow_dict(self) -> dict:
|
|
376
|
+
"""Serializes the DeltaSharingTableDependency into a shallow dictionary of its immediate attributes."""
|
|
377
|
+
body = {}
|
|
378
|
+
if self.schema_name is not None:
|
|
379
|
+
body["schema_name"] = self.schema_name
|
|
380
|
+
if self.table_name is not None:
|
|
381
|
+
body["table_name"] = self.table_name
|
|
382
|
+
return body
|
|
383
|
+
|
|
384
|
+
@classmethod
|
|
385
|
+
def from_dict(cls, d: Dict[str, Any]) -> DeltaSharingTableDependency:
|
|
386
|
+
"""Deserializes the DeltaSharingTableDependency from a dictionary."""
|
|
387
|
+
return cls(schema_name=d.get("schema_name", None), table_name=d.get("table_name", None))
|
|
388
|
+
|
|
389
|
+
|
|
390
|
+
@dataclass
|
|
391
|
+
class Function:
|
|
328
392
|
aliases: Optional[List[RegisteredModelAlias]] = None
|
|
329
393
|
"""The aliass of registered model."""
|
|
330
394
|
|
|
@@ -374,7 +438,7 @@ class DeltaSharingFunction:
|
|
|
374
438
|
"""The tags of the function."""
|
|
375
439
|
|
|
376
440
|
def as_dict(self) -> dict:
|
|
377
|
-
"""Serializes the
|
|
441
|
+
"""Serializes the Function into a dictionary suitable for use as a JSON request body."""
|
|
378
442
|
body = {}
|
|
379
443
|
if self.aliases:
|
|
380
444
|
body["aliases"] = [v.as_dict() for v in self.aliases]
|
|
@@ -411,7 +475,7 @@ class DeltaSharingFunction:
|
|
|
411
475
|
return body
|
|
412
476
|
|
|
413
477
|
def as_shallow_dict(self) -> dict:
|
|
414
|
-
"""Serializes the
|
|
478
|
+
"""Serializes the Function into a shallow dictionary of its immediate attributes."""
|
|
415
479
|
body = {}
|
|
416
480
|
if self.aliases:
|
|
417
481
|
body["aliases"] = self.aliases
|
|
@@ -448,8 +512,8 @@ class DeltaSharingFunction:
|
|
|
448
512
|
return body
|
|
449
513
|
|
|
450
514
|
@classmethod
|
|
451
|
-
def from_dict(cls, d: Dict[str, Any]) ->
|
|
452
|
-
"""Deserializes the
|
|
515
|
+
def from_dict(cls, d: Dict[str, Any]) -> Function:
|
|
516
|
+
"""Deserializes the Function from a dictionary."""
|
|
453
517
|
return cls(
|
|
454
518
|
aliases=_repeated_dict(d, "aliases", RegisteredModelAlias),
|
|
455
519
|
comment=d.get("comment", None),
|
|
@@ -470,70 +534,6 @@ class DeltaSharingFunction:
|
|
|
470
534
|
)
|
|
471
535
|
|
|
472
536
|
|
|
473
|
-
@dataclass
|
|
474
|
-
class DeltaSharingFunctionDependency:
|
|
475
|
-
"""A Function in UC as a dependency."""
|
|
476
|
-
|
|
477
|
-
function_name: Optional[str] = None
|
|
478
|
-
|
|
479
|
-
schema_name: Optional[str] = None
|
|
480
|
-
|
|
481
|
-
def as_dict(self) -> dict:
|
|
482
|
-
"""Serializes the DeltaSharingFunctionDependency into a dictionary suitable for use as a JSON request body."""
|
|
483
|
-
body = {}
|
|
484
|
-
if self.function_name is not None:
|
|
485
|
-
body["function_name"] = self.function_name
|
|
486
|
-
if self.schema_name is not None:
|
|
487
|
-
body["schema_name"] = self.schema_name
|
|
488
|
-
return body
|
|
489
|
-
|
|
490
|
-
def as_shallow_dict(self) -> dict:
|
|
491
|
-
"""Serializes the DeltaSharingFunctionDependency into a shallow dictionary of its immediate attributes."""
|
|
492
|
-
body = {}
|
|
493
|
-
if self.function_name is not None:
|
|
494
|
-
body["function_name"] = self.function_name
|
|
495
|
-
if self.schema_name is not None:
|
|
496
|
-
body["schema_name"] = self.schema_name
|
|
497
|
-
return body
|
|
498
|
-
|
|
499
|
-
@classmethod
|
|
500
|
-
def from_dict(cls, d: Dict[str, Any]) -> DeltaSharingFunctionDependency:
|
|
501
|
-
"""Deserializes the DeltaSharingFunctionDependency from a dictionary."""
|
|
502
|
-
return cls(function_name=d.get("function_name", None), schema_name=d.get("schema_name", None))
|
|
503
|
-
|
|
504
|
-
|
|
505
|
-
@dataclass
|
|
506
|
-
class DeltaSharingTableDependency:
|
|
507
|
-
"""A Table in UC as a dependency."""
|
|
508
|
-
|
|
509
|
-
schema_name: Optional[str] = None
|
|
510
|
-
|
|
511
|
-
table_name: Optional[str] = None
|
|
512
|
-
|
|
513
|
-
def as_dict(self) -> dict:
|
|
514
|
-
"""Serializes the DeltaSharingTableDependency into a dictionary suitable for use as a JSON request body."""
|
|
515
|
-
body = {}
|
|
516
|
-
if self.schema_name is not None:
|
|
517
|
-
body["schema_name"] = self.schema_name
|
|
518
|
-
if self.table_name is not None:
|
|
519
|
-
body["table_name"] = self.table_name
|
|
520
|
-
return body
|
|
521
|
-
|
|
522
|
-
def as_shallow_dict(self) -> dict:
|
|
523
|
-
"""Serializes the DeltaSharingTableDependency into a shallow dictionary of its immediate attributes."""
|
|
524
|
-
body = {}
|
|
525
|
-
if self.schema_name is not None:
|
|
526
|
-
body["schema_name"] = self.schema_name
|
|
527
|
-
if self.table_name is not None:
|
|
528
|
-
body["table_name"] = self.table_name
|
|
529
|
-
return body
|
|
530
|
-
|
|
531
|
-
@classmethod
|
|
532
|
-
def from_dict(cls, d: Dict[str, Any]) -> DeltaSharingTableDependency:
|
|
533
|
-
"""Deserializes the DeltaSharingTableDependency from a dictionary."""
|
|
534
|
-
return cls(schema_name=d.get("schema_name", None), table_name=d.get("table_name", None))
|
|
535
|
-
|
|
536
|
-
|
|
537
537
|
@dataclass
|
|
538
538
|
class FunctionParameterInfo:
|
|
539
539
|
"""Represents a parameter of a function. The same message is used for both input and output
|
|
@@ -809,7 +809,7 @@ class IpAccessList:
|
|
|
809
809
|
class ListProviderShareAssetsResponse:
|
|
810
810
|
"""Response to ListProviderShareAssets, which contains the list of assets of a share."""
|
|
811
811
|
|
|
812
|
-
functions: Optional[List[
|
|
812
|
+
functions: Optional[List[Function]] = None
|
|
813
813
|
"""The list of functions in the share."""
|
|
814
814
|
|
|
815
815
|
notebooks: Optional[List[NotebookFile]] = None
|
|
@@ -851,7 +851,7 @@ class ListProviderShareAssetsResponse:
|
|
|
851
851
|
def from_dict(cls, d: Dict[str, Any]) -> ListProviderShareAssetsResponse:
|
|
852
852
|
"""Deserializes the ListProviderShareAssetsResponse from a dictionary."""
|
|
853
853
|
return cls(
|
|
854
|
-
functions=_repeated_dict(d, "functions",
|
|
854
|
+
functions=_repeated_dict(d, "functions", Function),
|
|
855
855
|
notebooks=_repeated_dict(d, "notebooks", NotebookFile),
|
|
856
856
|
tables=_repeated_dict(d, "tables", Table),
|
|
857
857
|
volumes=_repeated_dict(d, "volumes", Volume),
|
databricks/sdk/version.py
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
__version__ = "0.
|
|
1
|
+
__version__ = "0.48.0"
|
|
@@ -15,7 +15,7 @@ databricks/sdk/oauth.py,sha256=8Cod3Fa4gPuVhd7zFBkVOVLB1fwiAKkyibU6_wnpqDE,28411
|
|
|
15
15
|
databricks/sdk/py.typed,sha256=pSvaHpbY1UPNEXyVFUjlgBhjPFZMmVC_UNrPC7eMOHI,74
|
|
16
16
|
databricks/sdk/retries.py,sha256=7k2kEexGqGKXHNAWHbPFSZSugU8UIU0qtyly_hix22Q,2581
|
|
17
17
|
databricks/sdk/useragent.py,sha256=boEgzTv-Zmo6boipZKjSopNy0CXg4GShC1_lTKpJgqs,7361
|
|
18
|
-
databricks/sdk/version.py,sha256=
|
|
18
|
+
databricks/sdk/version.py,sha256=bkYe4lEQZCEmFm0XRZaZkxTV1niMqR_lbp-tzKL6s6c,23
|
|
19
19
|
databricks/sdk/_widgets/__init__.py,sha256=VhI-VvLlr3rKUT1nbROslHJIbmZX_tPJ9rRhrdFsYUA,2811
|
|
20
20
|
databricks/sdk/_widgets/default_widgets_utils.py,sha256=_hwCbptLbRzWEmknco0H1wQNAYcuy2pjFO9NiRbvFeo,1127
|
|
21
21
|
databricks/sdk/_widgets/ipywidgets_utils.py,sha256=mg3rEPG9z76e0yVjGgcLybUvd_zSuN5ziGeKiZ-c8Ew,2927
|
|
@@ -44,27 +44,27 @@ databricks/sdk/service/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3
|
|
|
44
44
|
databricks/sdk/service/_internal.py,sha256=PY83MPehEwGuMzCnyvolqglnfZeQ7-eS38kedTa7KDU,1985
|
|
45
45
|
databricks/sdk/service/apps.py,sha256=NdPhpycOpsYlXlBB5JTvt-AmiaCxIqFKMz7370CKKQo,54550
|
|
46
46
|
databricks/sdk/service/billing.py,sha256=s-QN57TyMMfJSA40CfIjjno8iWHBssjpxcy0xcEcwyw,98590
|
|
47
|
-
databricks/sdk/service/catalog.py,sha256=
|
|
47
|
+
databricks/sdk/service/catalog.py,sha256=eu1zLzoeQxirFPMbthS5d5Mt0EjdbuqYP4sESotNwHc,602387
|
|
48
48
|
databricks/sdk/service/cleanrooms.py,sha256=TON2V0-T_TOk31K9bUypuRJaAJhHMM0bQDbMjJWlCDQ,58613
|
|
49
|
-
databricks/sdk/service/compute.py,sha256=
|
|
50
|
-
databricks/sdk/service/dashboards.py,sha256=
|
|
49
|
+
databricks/sdk/service/compute.py,sha256=R8GilWUVYVljZHfdNuiP4AJiAFWqKG4_XKn6Wg-um68,553503
|
|
50
|
+
databricks/sdk/service/dashboards.py,sha256=MhKFjglM4odyOiNzagII4pprkPMh8FOfzvwm9Ntlyr8,105263
|
|
51
51
|
databricks/sdk/service/files.py,sha256=gWjtORuRe7B1VZol0rr_dcSHq1gMmuKszGHaMqunslU,46234
|
|
52
|
-
databricks/sdk/service/iam.py,sha256=
|
|
53
|
-
databricks/sdk/service/jobs.py,sha256=
|
|
54
|
-
databricks/sdk/service/marketplace.py,sha256=
|
|
55
|
-
databricks/sdk/service/ml.py,sha256=
|
|
56
|
-
databricks/sdk/service/oauth2.py,sha256=
|
|
57
|
-
databricks/sdk/service/pipelines.py,sha256=
|
|
52
|
+
databricks/sdk/service/iam.py,sha256=TFKlhh7A1INGVcfAZQ1f0ORpAIzWLJ78l23vkerTe4M,174846
|
|
53
|
+
databricks/sdk/service/jobs.py,sha256=2DQIqaeYq2tOeRzeeq23Q8_ntXaeJNK0LHhlcbd_8A4,443424
|
|
54
|
+
databricks/sdk/service/marketplace.py,sha256=h-hGo2GHrjbVe0lXuengvqIDEN5VGhBdna4di9at5hA,175736
|
|
55
|
+
databricks/sdk/service/ml.py,sha256=q5vdOguyLiTpGqT6R5w41l_cCendUx4Hm-fQeW1-08k,316482
|
|
56
|
+
databricks/sdk/service/oauth2.py,sha256=LLp1uMvxfPOXzuekLjnLH6si6YOCvie7qZSpjSNDYd4,79788
|
|
57
|
+
databricks/sdk/service/pipelines.py,sha256=_2RpczckyhexbCtLoB_ewVKgZNFseN8bCwABod6YUqs,166666
|
|
58
58
|
databricks/sdk/service/provisioning.py,sha256=-Ly2o02i-jhNmiP9zLPeYF8H2usoB-oTG0RLF5gkIpc,169311
|
|
59
|
-
databricks/sdk/service/serving.py,sha256=
|
|
59
|
+
databricks/sdk/service/serving.py,sha256=eHPRoxeOpqV-dyVf7V8_6_FHBoTTwXqr5W9Y6b_iZtI,204406
|
|
60
60
|
databricks/sdk/service/settings.py,sha256=9I8NHnVTtTn_u91fYQi-QpDMjKPF2c2Sec0gEub6on0,323112
|
|
61
|
-
databricks/sdk/service/sharing.py,sha256=
|
|
61
|
+
databricks/sdk/service/sharing.py,sha256=jP3XWl3qojFWmZE9xkgCixqEZPuGZAJC_M-eGXMrPQs,142473
|
|
62
62
|
databricks/sdk/service/sql.py,sha256=2t8QhrEUyLekRYHnRYKkxVY8qpDbxE7JayfMSds6OKM,409593
|
|
63
63
|
databricks/sdk/service/vectorsearch.py,sha256=stdRmV-iE8L0rGzhSSG5tmoVRXb9bPItSE_ss6HQVDo,79897
|
|
64
64
|
databricks/sdk/service/workspace.py,sha256=T0ZbnG1qcPjKysGO_tBzl5x1PyalydeYJRBZbooYNm0,130893
|
|
65
|
-
databricks_sdk-0.
|
|
66
|
-
databricks_sdk-0.
|
|
67
|
-
databricks_sdk-0.
|
|
68
|
-
databricks_sdk-0.
|
|
69
|
-
databricks_sdk-0.
|
|
70
|
-
databricks_sdk-0.
|
|
65
|
+
databricks_sdk-0.48.0.dist-info/licenses/LICENSE,sha256=afBgTZo-JsYqj4VOjnejBetMuHKcFR30YobDdpVFkqY,11411
|
|
66
|
+
databricks_sdk-0.48.0.dist-info/licenses/NOTICE,sha256=tkRcQYA1k68wDLcnOWbg2xJDsUOJw8G8DGBhb8dnI3w,1588
|
|
67
|
+
databricks_sdk-0.48.0.dist-info/METADATA,sha256=QVB1xlNzOVnFrOEVSKUVtW-ZIf9GAoGBzH_HdPwl3sA,38334
|
|
68
|
+
databricks_sdk-0.48.0.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
|
|
69
|
+
databricks_sdk-0.48.0.dist-info/top_level.txt,sha256=7kRdatoSgU0EUurRQJ_3F1Nv4EOSHWAr6ng25tJOJKU,11
|
|
70
|
+
databricks_sdk-0.48.0.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|