databricks_uc_registry_messages_pb2.pyi
1 import databricks_pb2 as _databricks_pb2 2 from scalapb import scalapb_pb2 as _scalapb_pb2 3 from google.protobuf.internal import containers as _containers 4 from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper 5 from google.protobuf import descriptor as _descriptor 6 from google.protobuf import message as _message 7 from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union 8 9 DESCRIPTOR: _descriptor.FileDescriptor 10 11 class ModelVersionStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 12 __slots__ = () 13 UNSPECIFIED: _ClassVar[ModelVersionStatus] 14 PENDING_REGISTRATION: _ClassVar[ModelVersionStatus] 15 FAILED_REGISTRATION: _ClassVar[ModelVersionStatus] 16 READY: _ClassVar[ModelVersionStatus] 17 18 class ModelVersionOperation(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 19 __slots__ = () 20 MODEL_VERSION_OPERATION_UNSPECIFIED: _ClassVar[ModelVersionOperation] 21 MODEL_VERSION_OPERATION_READ: _ClassVar[ModelVersionOperation] 22 MODEL_VERSION_OPERATION_READ_WRITE: _ClassVar[ModelVersionOperation] 23 24 class StorageMode(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 25 __slots__ = () 26 STORAGE_MODE_UNSPECIFIED: _ClassVar[StorageMode] 27 CUSTOMER_HOSTED: _ClassVar[StorageMode] 28 DEFAULT_STORAGE: _ClassVar[StorageMode] 29 30 class SseEncryptionAlgorithm(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 31 __slots__ = () 32 SSE_ENCRYPTION_ALGORITHM_UNSPECIFIED: _ClassVar[SseEncryptionAlgorithm] 33 AWS_SSE_S3: _ClassVar[SseEncryptionAlgorithm] 34 AWS_SSE_KMS: _ClassVar[SseEncryptionAlgorithm] 35 36 class DependencyType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 37 __slots__ = () 38 DEPENDENCY_TYPE_UNSPECIFIED: _ClassVar[DependencyType] 39 DATABRICKS_VECTOR_INDEX: _ClassVar[DependencyType] 40 DATABRICKS_MODEL_ENDPOINT: _ClassVar[DependencyType] 41 DATABRICKS_UC_FUNCTION: _ClassVar[DependencyType] 42 DATABRICKS_UC_CONNECTION: _ClassVar[DependencyType] 43 DATABRICKS_TABLE: _ClassVar[DependencyType] 44 45 class TableType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 46 __slots__ = () 47 TABLE: _ClassVar[TableType] 48 PERSISTED_VIEW: _ClassVar[TableType] 49 TEMP_VIEW: _ClassVar[TableType] 50 MATERIALIZED_VIEW: _ClassVar[TableType] 51 STREAMING_LIVE_TABLE: _ClassVar[TableType] 52 PATH: _ClassVar[TableType] 53 54 class ModelVersionLineageDirection(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 55 __slots__ = () 56 UPSTREAM: _ClassVar[ModelVersionLineageDirection] 57 DOWNSTREAM: _ClassVar[ModelVersionLineageDirection] 58 UNSPECIFIED: ModelVersionStatus 59 PENDING_REGISTRATION: ModelVersionStatus 60 FAILED_REGISTRATION: ModelVersionStatus 61 READY: ModelVersionStatus 62 MODEL_VERSION_OPERATION_UNSPECIFIED: ModelVersionOperation 63 MODEL_VERSION_OPERATION_READ: ModelVersionOperation 64 MODEL_VERSION_OPERATION_READ_WRITE: ModelVersionOperation 65 STORAGE_MODE_UNSPECIFIED: StorageMode 66 CUSTOMER_HOSTED: StorageMode 67 DEFAULT_STORAGE: StorageMode 68 SSE_ENCRYPTION_ALGORITHM_UNSPECIFIED: SseEncryptionAlgorithm 69 AWS_SSE_S3: SseEncryptionAlgorithm 70 AWS_SSE_KMS: SseEncryptionAlgorithm 71 DEPENDENCY_TYPE_UNSPECIFIED: DependencyType 72 DATABRICKS_VECTOR_INDEX: DependencyType 73 DATABRICKS_MODEL_ENDPOINT: DependencyType 74 DATABRICKS_UC_FUNCTION: DependencyType 75 DATABRICKS_UC_CONNECTION: DependencyType 76 DATABRICKS_TABLE: DependencyType 77 TABLE: TableType 78 PERSISTED_VIEW: TableType 79 TEMP_VIEW: TableType 80 MATERIALIZED_VIEW: TableType 81 STREAMING_LIVE_TABLE: TableType 82 PATH: TableType 83 UPSTREAM: ModelVersionLineageDirection 84 DOWNSTREAM: ModelVersionLineageDirection 85 86 class RegisteredModel(_message.Message): 87 __slots__ = ("name", "creation_timestamp", "last_updated_timestamp", "user_id", "description", "aliases", "tags", "deployment_job_id", "deployment_job_state") 88 NAME_FIELD_NUMBER: _ClassVar[int] 89 CREATION_TIMESTAMP_FIELD_NUMBER: _ClassVar[int] 90 LAST_UPDATED_TIMESTAMP_FIELD_NUMBER: _ClassVar[int] 91 USER_ID_FIELD_NUMBER: _ClassVar[int] 92 DESCRIPTION_FIELD_NUMBER: _ClassVar[int] 93 ALIASES_FIELD_NUMBER: _ClassVar[int] 94 TAGS_FIELD_NUMBER: _ClassVar[int] 95 DEPLOYMENT_JOB_ID_FIELD_NUMBER: _ClassVar[int] 96 DEPLOYMENT_JOB_STATE_FIELD_NUMBER: _ClassVar[int] 97 name: str 98 creation_timestamp: int 99 last_updated_timestamp: int 100 user_id: str 101 description: str 102 aliases: _containers.RepeatedCompositeFieldContainer[RegisteredModelAlias] 103 tags: _containers.RepeatedCompositeFieldContainer[RegisteredModelTag] 104 deployment_job_id: str 105 deployment_job_state: DeploymentJobConnection.State 106 def __init__(self, name: _Optional[str] = ..., creation_timestamp: _Optional[int] = ..., last_updated_timestamp: _Optional[int] = ..., user_id: _Optional[str] = ..., description: _Optional[str] = ..., aliases: _Optional[_Iterable[_Union[RegisteredModelAlias, _Mapping]]] = ..., tags: _Optional[_Iterable[_Union[RegisteredModelTag, _Mapping]]] = ..., deployment_job_id: _Optional[str] = ..., deployment_job_state: _Optional[_Union[DeploymentJobConnection.State, str]] = ...) -> None: ... 107 108 class RegisteredModelAlias(_message.Message): 109 __slots__ = ("alias", "version") 110 ALIAS_FIELD_NUMBER: _ClassVar[int] 111 VERSION_FIELD_NUMBER: _ClassVar[int] 112 alias: str 113 version: str 114 def __init__(self, alias: _Optional[str] = ..., version: _Optional[str] = ...) -> None: ... 115 116 class RegisteredModelTag(_message.Message): 117 __slots__ = ("key", "value") 118 KEY_FIELD_NUMBER: _ClassVar[int] 119 VALUE_FIELD_NUMBER: _ClassVar[int] 120 key: str 121 value: str 122 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 123 124 class ModelParam(_message.Message): 125 __slots__ = ("name", "value") 126 NAME_FIELD_NUMBER: _ClassVar[int] 127 VALUE_FIELD_NUMBER: _ClassVar[int] 128 name: str 129 value: str 130 def __init__(self, name: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 131 132 class ModelMetric(_message.Message): 133 __slots__ = ("key", "value", "timestamp", "step", "dataset_name", "dataset_digest", "model_id", "run_id") 134 KEY_FIELD_NUMBER: _ClassVar[int] 135 VALUE_FIELD_NUMBER: _ClassVar[int] 136 TIMESTAMP_FIELD_NUMBER: _ClassVar[int] 137 STEP_FIELD_NUMBER: _ClassVar[int] 138 DATASET_NAME_FIELD_NUMBER: _ClassVar[int] 139 DATASET_DIGEST_FIELD_NUMBER: _ClassVar[int] 140 MODEL_ID_FIELD_NUMBER: _ClassVar[int] 141 RUN_ID_FIELD_NUMBER: _ClassVar[int] 142 key: str 143 value: float 144 timestamp: int 145 step: int 146 dataset_name: str 147 dataset_digest: str 148 model_id: str 149 run_id: str 150 def __init__(self, key: _Optional[str] = ..., value: _Optional[float] = ..., timestamp: _Optional[int] = ..., step: _Optional[int] = ..., dataset_name: _Optional[str] = ..., dataset_digest: _Optional[str] = ..., model_id: _Optional[str] = ..., run_id: _Optional[str] = ...) -> None: ... 151 152 class ModelVersionTag(_message.Message): 153 __slots__ = ("key", "value") 154 KEY_FIELD_NUMBER: _ClassVar[int] 155 VALUE_FIELD_NUMBER: _ClassVar[int] 156 key: str 157 value: str 158 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 159 160 class ModelVersion(_message.Message): 161 __slots__ = ("name", "version", "creation_timestamp", "last_updated_timestamp", "user_id", "description", "source", "run_id", "run_experiment_id", "run_tracking_server_id", "status", "status_message", "storage_location", "aliases", "tags", "model_id", "model_params", "model_metrics", "deployment_job_state") 162 NAME_FIELD_NUMBER: _ClassVar[int] 163 VERSION_FIELD_NUMBER: _ClassVar[int] 164 CREATION_TIMESTAMP_FIELD_NUMBER: _ClassVar[int] 165 LAST_UPDATED_TIMESTAMP_FIELD_NUMBER: _ClassVar[int] 166 USER_ID_FIELD_NUMBER: _ClassVar[int] 167 DESCRIPTION_FIELD_NUMBER: _ClassVar[int] 168 SOURCE_FIELD_NUMBER: _ClassVar[int] 169 RUN_ID_FIELD_NUMBER: _ClassVar[int] 170 RUN_EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 171 RUN_TRACKING_SERVER_ID_FIELD_NUMBER: _ClassVar[int] 172 STATUS_FIELD_NUMBER: _ClassVar[int] 173 STATUS_MESSAGE_FIELD_NUMBER: _ClassVar[int] 174 STORAGE_LOCATION_FIELD_NUMBER: _ClassVar[int] 175 ALIASES_FIELD_NUMBER: _ClassVar[int] 176 TAGS_FIELD_NUMBER: _ClassVar[int] 177 MODEL_ID_FIELD_NUMBER: _ClassVar[int] 178 MODEL_PARAMS_FIELD_NUMBER: _ClassVar[int] 179 MODEL_METRICS_FIELD_NUMBER: _ClassVar[int] 180 DEPLOYMENT_JOB_STATE_FIELD_NUMBER: _ClassVar[int] 181 name: str 182 version: str 183 creation_timestamp: int 184 last_updated_timestamp: int 185 user_id: str 186 description: str 187 source: str 188 run_id: str 189 run_experiment_id: str 190 run_tracking_server_id: str 191 status: ModelVersionStatus 192 status_message: str 193 storage_location: str 194 aliases: _containers.RepeatedCompositeFieldContainer[RegisteredModelAlias] 195 tags: _containers.RepeatedCompositeFieldContainer[ModelVersionTag] 196 model_id: str 197 model_params: _containers.RepeatedCompositeFieldContainer[ModelParam] 198 model_metrics: _containers.RepeatedCompositeFieldContainer[ModelMetric] 199 deployment_job_state: ModelVersionDeploymentJobState 200 def __init__(self, name: _Optional[str] = ..., version: _Optional[str] = ..., creation_timestamp: _Optional[int] = ..., last_updated_timestamp: _Optional[int] = ..., user_id: _Optional[str] = ..., description: _Optional[str] = ..., source: _Optional[str] = ..., run_id: _Optional[str] = ..., run_experiment_id: _Optional[str] = ..., run_tracking_server_id: _Optional[str] = ..., status: _Optional[_Union[ModelVersionStatus, str]] = ..., status_message: _Optional[str] = ..., storage_location: _Optional[str] = ..., aliases: _Optional[_Iterable[_Union[RegisteredModelAlias, _Mapping]]] = ..., tags: _Optional[_Iterable[_Union[ModelVersionTag, _Mapping]]] = ..., model_id: _Optional[str] = ..., model_params: _Optional[_Iterable[_Union[ModelParam, _Mapping]]] = ..., model_metrics: _Optional[_Iterable[_Union[ModelMetric, _Mapping]]] = ..., deployment_job_state: _Optional[_Union[ModelVersionDeploymentJobState, _Mapping]] = ...) -> None: ... 201 202 class DeploymentJobConnection(_message.Message): 203 __slots__ = () 204 class State(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 205 __slots__ = () 206 DEPLOYMENT_JOB_CONNECTION_STATE_UNSPECIFIED: _ClassVar[DeploymentJobConnection.State] 207 NOT_SET_UP: _ClassVar[DeploymentJobConnection.State] 208 CONNECTED: _ClassVar[DeploymentJobConnection.State] 209 NOT_FOUND: _ClassVar[DeploymentJobConnection.State] 210 REQUIRED_PARAMETERS_CHANGED: _ClassVar[DeploymentJobConnection.State] 211 DEPLOYMENT_JOB_CONNECTION_STATE_UNSPECIFIED: DeploymentJobConnection.State 212 NOT_SET_UP: DeploymentJobConnection.State 213 CONNECTED: DeploymentJobConnection.State 214 NOT_FOUND: DeploymentJobConnection.State 215 REQUIRED_PARAMETERS_CHANGED: DeploymentJobConnection.State 216 def __init__(self) -> None: ... 217 218 class ModelVersionDeploymentJobState(_message.Message): 219 __slots__ = ("job_id", "run_id", "job_state", "run_state", "current_task_name") 220 class DeploymentJobRunState(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 221 __slots__ = () 222 DEPLOYMENT_JOB_RUN_STATE_UNSPECIFIED: _ClassVar[ModelVersionDeploymentJobState.DeploymentJobRunState] 223 NO_VALID_DEPLOYMENT_JOB_FOUND: _ClassVar[ModelVersionDeploymentJobState.DeploymentJobRunState] 224 RUNNING: _ClassVar[ModelVersionDeploymentJobState.DeploymentJobRunState] 225 SUCCEEDED: _ClassVar[ModelVersionDeploymentJobState.DeploymentJobRunState] 226 FAILED: _ClassVar[ModelVersionDeploymentJobState.DeploymentJobRunState] 227 PENDING: _ClassVar[ModelVersionDeploymentJobState.DeploymentJobRunState] 228 APPROVAL: _ClassVar[ModelVersionDeploymentJobState.DeploymentJobRunState] 229 DEPLOYMENT_JOB_RUN_STATE_UNSPECIFIED: ModelVersionDeploymentJobState.DeploymentJobRunState 230 NO_VALID_DEPLOYMENT_JOB_FOUND: ModelVersionDeploymentJobState.DeploymentJobRunState 231 RUNNING: ModelVersionDeploymentJobState.DeploymentJobRunState 232 SUCCEEDED: ModelVersionDeploymentJobState.DeploymentJobRunState 233 FAILED: ModelVersionDeploymentJobState.DeploymentJobRunState 234 PENDING: ModelVersionDeploymentJobState.DeploymentJobRunState 235 APPROVAL: ModelVersionDeploymentJobState.DeploymentJobRunState 236 JOB_ID_FIELD_NUMBER: _ClassVar[int] 237 RUN_ID_FIELD_NUMBER: _ClassVar[int] 238 JOB_STATE_FIELD_NUMBER: _ClassVar[int] 239 RUN_STATE_FIELD_NUMBER: _ClassVar[int] 240 CURRENT_TASK_NAME_FIELD_NUMBER: _ClassVar[int] 241 job_id: str 242 run_id: str 243 job_state: DeploymentJobConnection.State 244 run_state: ModelVersionDeploymentJobState.DeploymentJobRunState 245 current_task_name: str 246 def __init__(self, job_id: _Optional[str] = ..., run_id: _Optional[str] = ..., job_state: _Optional[_Union[DeploymentJobConnection.State, str]] = ..., run_state: _Optional[_Union[ModelVersionDeploymentJobState.DeploymentJobRunState, str]] = ..., current_task_name: _Optional[str] = ...) -> None: ... 247 248 class TemporaryCredentials(_message.Message): 249 __slots__ = ("aws_temp_credentials", "azure_user_delegation_sas", "gcp_oauth_token", "r2_temp_credentials", "expiration_time", "storage_mode", "encryption_details") 250 AWS_TEMP_CREDENTIALS_FIELD_NUMBER: _ClassVar[int] 251 AZURE_USER_DELEGATION_SAS_FIELD_NUMBER: _ClassVar[int] 252 GCP_OAUTH_TOKEN_FIELD_NUMBER: _ClassVar[int] 253 R2_TEMP_CREDENTIALS_FIELD_NUMBER: _ClassVar[int] 254 EXPIRATION_TIME_FIELD_NUMBER: _ClassVar[int] 255 STORAGE_MODE_FIELD_NUMBER: _ClassVar[int] 256 ENCRYPTION_DETAILS_FIELD_NUMBER: _ClassVar[int] 257 aws_temp_credentials: AwsCredentials 258 azure_user_delegation_sas: AzureUserDelegationSAS 259 gcp_oauth_token: GcpOauthToken 260 r2_temp_credentials: R2Credentials 261 expiration_time: int 262 storage_mode: StorageMode 263 encryption_details: EncryptionDetails 264 def __init__(self, aws_temp_credentials: _Optional[_Union[AwsCredentials, _Mapping]] = ..., azure_user_delegation_sas: _Optional[_Union[AzureUserDelegationSAS, _Mapping]] = ..., gcp_oauth_token: _Optional[_Union[GcpOauthToken, _Mapping]] = ..., r2_temp_credentials: _Optional[_Union[R2Credentials, _Mapping]] = ..., expiration_time: _Optional[int] = ..., storage_mode: _Optional[_Union[StorageMode, str]] = ..., encryption_details: _Optional[_Union[EncryptionDetails, _Mapping]] = ...) -> None: ... 265 266 class AwsCredentials(_message.Message): 267 __slots__ = ("access_key_id", "secret_access_key", "session_token") 268 ACCESS_KEY_ID_FIELD_NUMBER: _ClassVar[int] 269 SECRET_ACCESS_KEY_FIELD_NUMBER: _ClassVar[int] 270 SESSION_TOKEN_FIELD_NUMBER: _ClassVar[int] 271 access_key_id: str 272 secret_access_key: str 273 session_token: str 274 def __init__(self, access_key_id: _Optional[str] = ..., secret_access_key: _Optional[str] = ..., session_token: _Optional[str] = ...) -> None: ... 275 276 class AzureUserDelegationSAS(_message.Message): 277 __slots__ = ("sas_token",) 278 SAS_TOKEN_FIELD_NUMBER: _ClassVar[int] 279 sas_token: str 280 def __init__(self, sas_token: _Optional[str] = ...) -> None: ... 281 282 class GcpOauthToken(_message.Message): 283 __slots__ = ("oauth_token",) 284 OAUTH_TOKEN_FIELD_NUMBER: _ClassVar[int] 285 oauth_token: str 286 def __init__(self, oauth_token: _Optional[str] = ...) -> None: ... 287 288 class R2Credentials(_message.Message): 289 __slots__ = ("access_key_id", "secret_access_key", "session_token") 290 ACCESS_KEY_ID_FIELD_NUMBER: _ClassVar[int] 291 SECRET_ACCESS_KEY_FIELD_NUMBER: _ClassVar[int] 292 SESSION_TOKEN_FIELD_NUMBER: _ClassVar[int] 293 access_key_id: str 294 secret_access_key: str 295 session_token: str 296 def __init__(self, access_key_id: _Optional[str] = ..., secret_access_key: _Optional[str] = ..., session_token: _Optional[str] = ...) -> None: ... 297 298 class EncryptionDetails(_message.Message): 299 __slots__ = ("sse_encryption_details",) 300 SSE_ENCRYPTION_DETAILS_FIELD_NUMBER: _ClassVar[int] 301 sse_encryption_details: SseEncryptionDetails 302 def __init__(self, sse_encryption_details: _Optional[_Union[SseEncryptionDetails, _Mapping]] = ...) -> None: ... 303 304 class SseEncryptionDetails(_message.Message): 305 __slots__ = ("algorithm", "aws_kms_key_arn") 306 ALGORITHM_FIELD_NUMBER: _ClassVar[int] 307 AWS_KMS_KEY_ARN_FIELD_NUMBER: _ClassVar[int] 308 algorithm: SseEncryptionAlgorithm 309 aws_kms_key_arn: str 310 def __init__(self, algorithm: _Optional[_Union[SseEncryptionAlgorithm, str]] = ..., aws_kms_key_arn: _Optional[str] = ...) -> None: ... 311 312 class CreateRegisteredModelRequest(_message.Message): 313 __slots__ = ("name", "tags", "description", "deployment_job_id") 314 NAME_FIELD_NUMBER: _ClassVar[int] 315 TAGS_FIELD_NUMBER: _ClassVar[int] 316 DESCRIPTION_FIELD_NUMBER: _ClassVar[int] 317 DEPLOYMENT_JOB_ID_FIELD_NUMBER: _ClassVar[int] 318 name: str 319 tags: _containers.RepeatedCompositeFieldContainer[RegisteredModelTag] 320 description: str 321 deployment_job_id: str 322 def __init__(self, name: _Optional[str] = ..., tags: _Optional[_Iterable[_Union[RegisteredModelTag, _Mapping]]] = ..., description: _Optional[str] = ..., deployment_job_id: _Optional[str] = ...) -> None: ... 323 324 class CreateRegisteredModelResponse(_message.Message): 325 __slots__ = ("registered_model",) 326 REGISTERED_MODEL_FIELD_NUMBER: _ClassVar[int] 327 registered_model: RegisteredModel 328 def __init__(self, registered_model: _Optional[_Union[RegisteredModel, _Mapping]] = ...) -> None: ... 329 330 class UpdateRegisteredModelRequest(_message.Message): 331 __slots__ = ("name", "new_name", "description", "deployment_job_id") 332 NAME_FIELD_NUMBER: _ClassVar[int] 333 NEW_NAME_FIELD_NUMBER: _ClassVar[int] 334 DESCRIPTION_FIELD_NUMBER: _ClassVar[int] 335 DEPLOYMENT_JOB_ID_FIELD_NUMBER: _ClassVar[int] 336 name: str 337 new_name: str 338 description: str 339 deployment_job_id: str 340 def __init__(self, name: _Optional[str] = ..., new_name: _Optional[str] = ..., description: _Optional[str] = ..., deployment_job_id: _Optional[str] = ...) -> None: ... 341 342 class UpdateRegisteredModelResponse(_message.Message): 343 __slots__ = ("registered_model",) 344 REGISTERED_MODEL_FIELD_NUMBER: _ClassVar[int] 345 registered_model: RegisteredModel 346 def __init__(self, registered_model: _Optional[_Union[RegisteredModel, _Mapping]] = ...) -> None: ... 347 348 class DeleteRegisteredModelRequest(_message.Message): 349 __slots__ = ("name",) 350 NAME_FIELD_NUMBER: _ClassVar[int] 351 name: str 352 def __init__(self, name: _Optional[str] = ...) -> None: ... 353 354 class DeleteRegisteredModelResponse(_message.Message): 355 __slots__ = () 356 def __init__(self) -> None: ... 357 358 class GetRegisteredModelRequest(_message.Message): 359 __slots__ = ("name",) 360 NAME_FIELD_NUMBER: _ClassVar[int] 361 name: str 362 def __init__(self, name: _Optional[str] = ...) -> None: ... 363 364 class GetRegisteredModelResponse(_message.Message): 365 __slots__ = ("registered_model",) 366 REGISTERED_MODEL_FIELD_NUMBER: _ClassVar[int] 367 registered_model: RegisteredModel 368 def __init__(self, registered_model: _Optional[_Union[RegisteredModel, _Mapping]] = ...) -> None: ... 369 370 class SearchRegisteredModelsRequest(_message.Message): 371 __slots__ = ("max_results", "page_token") 372 MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] 373 PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 374 max_results: int 375 page_token: str 376 def __init__(self, max_results: _Optional[int] = ..., page_token: _Optional[str] = ...) -> None: ... 377 378 class SearchRegisteredModelsResponse(_message.Message): 379 __slots__ = ("registered_models", "next_page_token") 380 REGISTERED_MODELS_FIELD_NUMBER: _ClassVar[int] 381 NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 382 registered_models: _containers.RepeatedCompositeFieldContainer[RegisteredModel] 383 next_page_token: str 384 def __init__(self, registered_models: _Optional[_Iterable[_Union[RegisteredModel, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... 385 386 class Dependency(_message.Message): 387 __slots__ = ("type", "name") 388 TYPE_FIELD_NUMBER: _ClassVar[int] 389 NAME_FIELD_NUMBER: _ClassVar[int] 390 type: DependencyType 391 name: str 392 def __init__(self, type: _Optional[_Union[DependencyType, str]] = ..., name: _Optional[str] = ...) -> None: ... 393 394 class CreateModelVersionRequest(_message.Message): 395 __slots__ = ("name", "source", "run_id", "description", "run_tracking_server_id", "feature_deps", "tags", "model_version_dependencies", "model_id") 396 NAME_FIELD_NUMBER: _ClassVar[int] 397 SOURCE_FIELD_NUMBER: _ClassVar[int] 398 RUN_ID_FIELD_NUMBER: _ClassVar[int] 399 DESCRIPTION_FIELD_NUMBER: _ClassVar[int] 400 RUN_TRACKING_SERVER_ID_FIELD_NUMBER: _ClassVar[int] 401 FEATURE_DEPS_FIELD_NUMBER: _ClassVar[int] 402 TAGS_FIELD_NUMBER: _ClassVar[int] 403 MODEL_VERSION_DEPENDENCIES_FIELD_NUMBER: _ClassVar[int] 404 MODEL_ID_FIELD_NUMBER: _ClassVar[int] 405 name: str 406 source: str 407 run_id: str 408 description: str 409 run_tracking_server_id: str 410 feature_deps: str 411 tags: _containers.RepeatedCompositeFieldContainer[ModelVersionTag] 412 model_version_dependencies: _containers.RepeatedCompositeFieldContainer[Dependency] 413 model_id: str 414 def __init__(self, name: _Optional[str] = ..., source: _Optional[str] = ..., run_id: _Optional[str] = ..., description: _Optional[str] = ..., run_tracking_server_id: _Optional[str] = ..., feature_deps: _Optional[str] = ..., tags: _Optional[_Iterable[_Union[ModelVersionTag, _Mapping]]] = ..., model_version_dependencies: _Optional[_Iterable[_Union[Dependency, _Mapping]]] = ..., model_id: _Optional[str] = ...) -> None: ... 415 416 class CreateModelVersionResponse(_message.Message): 417 __slots__ = ("model_version",) 418 MODEL_VERSION_FIELD_NUMBER: _ClassVar[int] 419 model_version: ModelVersion 420 def __init__(self, model_version: _Optional[_Union[ModelVersion, _Mapping]] = ...) -> None: ... 421 422 class UpdateModelVersionRequest(_message.Message): 423 __slots__ = ("name", "version", "description") 424 NAME_FIELD_NUMBER: _ClassVar[int] 425 VERSION_FIELD_NUMBER: _ClassVar[int] 426 DESCRIPTION_FIELD_NUMBER: _ClassVar[int] 427 name: str 428 version: str 429 description: str 430 def __init__(self, name: _Optional[str] = ..., version: _Optional[str] = ..., description: _Optional[str] = ...) -> None: ... 431 432 class UpdateModelVersionResponse(_message.Message): 433 __slots__ = ("model_version",) 434 MODEL_VERSION_FIELD_NUMBER: _ClassVar[int] 435 model_version: ModelVersion 436 def __init__(self, model_version: _Optional[_Union[ModelVersion, _Mapping]] = ...) -> None: ... 437 438 class DeleteModelVersionRequest(_message.Message): 439 __slots__ = ("name", "version") 440 NAME_FIELD_NUMBER: _ClassVar[int] 441 VERSION_FIELD_NUMBER: _ClassVar[int] 442 name: str 443 version: str 444 def __init__(self, name: _Optional[str] = ..., version: _Optional[str] = ...) -> None: ... 445 446 class DeleteModelVersionResponse(_message.Message): 447 __slots__ = () 448 def __init__(self) -> None: ... 449 450 class GetModelVersionRequest(_message.Message): 451 __slots__ = ("name", "version") 452 NAME_FIELD_NUMBER: _ClassVar[int] 453 VERSION_FIELD_NUMBER: _ClassVar[int] 454 name: str 455 version: str 456 def __init__(self, name: _Optional[str] = ..., version: _Optional[str] = ...) -> None: ... 457 458 class GetModelVersionResponse(_message.Message): 459 __slots__ = ("model_version",) 460 MODEL_VERSION_FIELD_NUMBER: _ClassVar[int] 461 model_version: ModelVersion 462 def __init__(self, model_version: _Optional[_Union[ModelVersion, _Mapping]] = ...) -> None: ... 463 464 class SearchModelVersionsRequest(_message.Message): 465 __slots__ = ("filter", "max_results", "page_token") 466 FILTER_FIELD_NUMBER: _ClassVar[int] 467 MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] 468 PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 469 filter: str 470 max_results: int 471 page_token: str 472 def __init__(self, filter: _Optional[str] = ..., max_results: _Optional[int] = ..., page_token: _Optional[str] = ...) -> None: ... 473 474 class SearchModelVersionsResponse(_message.Message): 475 __slots__ = ("model_versions", "next_page_token") 476 MODEL_VERSIONS_FIELD_NUMBER: _ClassVar[int] 477 NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 478 model_versions: _containers.RepeatedCompositeFieldContainer[ModelVersion] 479 next_page_token: str 480 def __init__(self, model_versions: _Optional[_Iterable[_Union[ModelVersion, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... 481 482 class GenerateTemporaryModelVersionCredentialsRequest(_message.Message): 483 __slots__ = ("name", "version", "operation") 484 NAME_FIELD_NUMBER: _ClassVar[int] 485 VERSION_FIELD_NUMBER: _ClassVar[int] 486 OPERATION_FIELD_NUMBER: _ClassVar[int] 487 name: str 488 version: str 489 operation: ModelVersionOperation 490 def __init__(self, name: _Optional[str] = ..., version: _Optional[str] = ..., operation: _Optional[_Union[ModelVersionOperation, str]] = ...) -> None: ... 491 492 class GenerateTemporaryModelVersionCredentialsResponse(_message.Message): 493 __slots__ = ("credentials",) 494 CREDENTIALS_FIELD_NUMBER: _ClassVar[int] 495 credentials: TemporaryCredentials 496 def __init__(self, credentials: _Optional[_Union[TemporaryCredentials, _Mapping]] = ...) -> None: ... 497 498 class GetModelVersionDownloadUriRequest(_message.Message): 499 __slots__ = ("name", "version") 500 NAME_FIELD_NUMBER: _ClassVar[int] 501 VERSION_FIELD_NUMBER: _ClassVar[int] 502 name: str 503 version: str 504 def __init__(self, name: _Optional[str] = ..., version: _Optional[str] = ...) -> None: ... 505 506 class GetModelVersionDownloadUriResponse(_message.Message): 507 __slots__ = ("artifact_uri",) 508 ARTIFACT_URI_FIELD_NUMBER: _ClassVar[int] 509 artifact_uri: str 510 def __init__(self, artifact_uri: _Optional[str] = ...) -> None: ... 511 512 class FinalizeModelVersionRequest(_message.Message): 513 __slots__ = ("name", "version") 514 NAME_FIELD_NUMBER: _ClassVar[int] 515 VERSION_FIELD_NUMBER: _ClassVar[int] 516 name: str 517 version: str 518 def __init__(self, name: _Optional[str] = ..., version: _Optional[str] = ...) -> None: ... 519 520 class FinalizeModelVersionResponse(_message.Message): 521 __slots__ = ("model_version",) 522 MODEL_VERSION_FIELD_NUMBER: _ClassVar[int] 523 model_version: ModelVersion 524 def __init__(self, model_version: _Optional[_Union[ModelVersion, _Mapping]] = ...) -> None: ... 525 526 class SetRegisteredModelAliasRequest(_message.Message): 527 __slots__ = ("name", "alias", "version") 528 NAME_FIELD_NUMBER: _ClassVar[int] 529 ALIAS_FIELD_NUMBER: _ClassVar[int] 530 VERSION_FIELD_NUMBER: _ClassVar[int] 531 name: str 532 alias: str 533 version: str 534 def __init__(self, name: _Optional[str] = ..., alias: _Optional[str] = ..., version: _Optional[str] = ...) -> None: ... 535 536 class SetRegisteredModelAliasResponse(_message.Message): 537 __slots__ = () 538 def __init__(self) -> None: ... 539 540 class DeleteRegisteredModelAliasRequest(_message.Message): 541 __slots__ = ("name", "alias") 542 NAME_FIELD_NUMBER: _ClassVar[int] 543 ALIAS_FIELD_NUMBER: _ClassVar[int] 544 name: str 545 alias: str 546 def __init__(self, name: _Optional[str] = ..., alias: _Optional[str] = ...) -> None: ... 547 548 class DeleteRegisteredModelAliasResponse(_message.Message): 549 __slots__ = () 550 def __init__(self) -> None: ... 551 552 class SetRegisteredModelTagRequest(_message.Message): 553 __slots__ = ("name", "key", "value") 554 NAME_FIELD_NUMBER: _ClassVar[int] 555 KEY_FIELD_NUMBER: _ClassVar[int] 556 VALUE_FIELD_NUMBER: _ClassVar[int] 557 name: str 558 key: str 559 value: str 560 def __init__(self, name: _Optional[str] = ..., key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 561 562 class SetRegisteredModelTagResponse(_message.Message): 563 __slots__ = () 564 def __init__(self) -> None: ... 565 566 class DeleteRegisteredModelTagRequest(_message.Message): 567 __slots__ = ("name", "key") 568 NAME_FIELD_NUMBER: _ClassVar[int] 569 KEY_FIELD_NUMBER: _ClassVar[int] 570 name: str 571 key: str 572 def __init__(self, name: _Optional[str] = ..., key: _Optional[str] = ...) -> None: ... 573 574 class DeleteRegisteredModelTagResponse(_message.Message): 575 __slots__ = () 576 def __init__(self) -> None: ... 577 578 class SetModelVersionTagRequest(_message.Message): 579 __slots__ = ("name", "version", "key", "value") 580 NAME_FIELD_NUMBER: _ClassVar[int] 581 VERSION_FIELD_NUMBER: _ClassVar[int] 582 KEY_FIELD_NUMBER: _ClassVar[int] 583 VALUE_FIELD_NUMBER: _ClassVar[int] 584 name: str 585 version: str 586 key: str 587 value: str 588 def __init__(self, name: _Optional[str] = ..., version: _Optional[str] = ..., key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 589 590 class SetModelVersionTagResponse(_message.Message): 591 __slots__ = () 592 def __init__(self) -> None: ... 593 594 class DeleteModelVersionTagRequest(_message.Message): 595 __slots__ = ("name", "version", "key") 596 NAME_FIELD_NUMBER: _ClassVar[int] 597 VERSION_FIELD_NUMBER: _ClassVar[int] 598 KEY_FIELD_NUMBER: _ClassVar[int] 599 name: str 600 version: str 601 key: str 602 def __init__(self, name: _Optional[str] = ..., version: _Optional[str] = ..., key: _Optional[str] = ...) -> None: ... 603 604 class DeleteModelVersionTagResponse(_message.Message): 605 __slots__ = () 606 def __init__(self) -> None: ... 607 608 class GetModelVersionByAliasRequest(_message.Message): 609 __slots__ = ("name", "alias") 610 NAME_FIELD_NUMBER: _ClassVar[int] 611 ALIAS_FIELD_NUMBER: _ClassVar[int] 612 name: str 613 alias: str 614 def __init__(self, name: _Optional[str] = ..., alias: _Optional[str] = ...) -> None: ... 615 616 class GetModelVersionByAliasResponse(_message.Message): 617 __slots__ = ("model_version",) 618 MODEL_VERSION_FIELD_NUMBER: _ClassVar[int] 619 model_version: ModelVersion 620 def __init__(self, model_version: _Optional[_Union[ModelVersion, _Mapping]] = ...) -> None: ... 621 622 class Entity(_message.Message): 623 __slots__ = ("job", "notebook", "pipeline") 624 JOB_FIELD_NUMBER: _ClassVar[int] 625 NOTEBOOK_FIELD_NUMBER: _ClassVar[int] 626 PIPELINE_FIELD_NUMBER: _ClassVar[int] 627 job: Job 628 notebook: Notebook 629 pipeline: Pipeline 630 def __init__(self, job: _Optional[_Union[Job, _Mapping]] = ..., notebook: _Optional[_Union[Notebook, _Mapping]] = ..., pipeline: _Optional[_Union[Pipeline, _Mapping]] = ...) -> None: ... 631 632 class Pipeline(_message.Message): 633 __slots__ = ("pipeline_id", "pipeline_update_id") 634 PIPELINE_ID_FIELD_NUMBER: _ClassVar[int] 635 PIPELINE_UPDATE_ID_FIELD_NUMBER: _ClassVar[int] 636 pipeline_id: str 637 pipeline_update_id: str 638 def __init__(self, pipeline_id: _Optional[str] = ..., pipeline_update_id: _Optional[str] = ...) -> None: ... 639 640 class Job(_message.Message): 641 __slots__ = ("id", "task_key", "job_run_id", "task_run_id") 642 ID_FIELD_NUMBER: _ClassVar[int] 643 TASK_KEY_FIELD_NUMBER: _ClassVar[int] 644 JOB_RUN_ID_FIELD_NUMBER: _ClassVar[int] 645 TASK_RUN_ID_FIELD_NUMBER: _ClassVar[int] 646 id: str 647 task_key: str 648 job_run_id: str 649 task_run_id: str 650 def __init__(self, id: _Optional[str] = ..., task_key: _Optional[str] = ..., job_run_id: _Optional[str] = ..., task_run_id: _Optional[str] = ...) -> None: ... 651 652 class Notebook(_message.Message): 653 __slots__ = ("id", "command_id", "notebook_run_id") 654 ID_FIELD_NUMBER: _ClassVar[int] 655 COMMAND_ID_FIELD_NUMBER: _ClassVar[int] 656 NOTEBOOK_RUN_ID_FIELD_NUMBER: _ClassVar[int] 657 id: str 658 command_id: str 659 notebook_run_id: str 660 def __init__(self, id: _Optional[str] = ..., command_id: _Optional[str] = ..., notebook_run_id: _Optional[str] = ...) -> None: ... 661 662 class Table(_message.Message): 663 __slots__ = ("name", "table_type", "location", "table_id") 664 NAME_FIELD_NUMBER: _ClassVar[int] 665 TABLE_TYPE_FIELD_NUMBER: _ClassVar[int] 666 LOCATION_FIELD_NUMBER: _ClassVar[int] 667 TABLE_ID_FIELD_NUMBER: _ClassVar[int] 668 name: str 669 table_type: TableType 670 location: str 671 table_id: str 672 def __init__(self, name: _Optional[str] = ..., table_type: _Optional[_Union[TableType, str]] = ..., location: _Optional[str] = ..., table_id: _Optional[str] = ...) -> None: ... 673 674 class Securable(_message.Message): 675 __slots__ = ("table",) 676 TABLE_FIELD_NUMBER: _ClassVar[int] 677 table: Table 678 def __init__(self, table: _Optional[_Union[Table, _Mapping]] = ...) -> None: ... 679 680 class Lineage(_message.Message): 681 __slots__ = ("target_securable", "source_securables") 682 TARGET_SECURABLE_FIELD_NUMBER: _ClassVar[int] 683 SOURCE_SECURABLES_FIELD_NUMBER: _ClassVar[int] 684 target_securable: Securable 685 source_securables: _containers.RepeatedCompositeFieldContainer[Securable] 686 def __init__(self, target_securable: _Optional[_Union[Securable, _Mapping]] = ..., source_securables: _Optional[_Iterable[_Union[Securable, _Mapping]]] = ...) -> None: ... 687 688 class LineageHeaderInfo(_message.Message): 689 __slots__ = ("entities", "lineages") 690 ENTITIES_FIELD_NUMBER: _ClassVar[int] 691 LINEAGES_FIELD_NUMBER: _ClassVar[int] 692 entities: _containers.RepeatedCompositeFieldContainer[Entity] 693 lineages: _containers.RepeatedCompositeFieldContainer[Lineage] 694 def __init__(self, entities: _Optional[_Iterable[_Union[Entity, _Mapping]]] = ..., lineages: _Optional[_Iterable[_Union[Lineage, _Mapping]]] = ...) -> None: ... 695 696 class ModelVersionLineageInfo(_message.Message): 697 __slots__ = ("entities", "direction") 698 ENTITIES_FIELD_NUMBER: _ClassVar[int] 699 DIRECTION_FIELD_NUMBER: _ClassVar[int] 700 entities: _containers.RepeatedCompositeFieldContainer[Entity] 701 direction: ModelVersionLineageDirection 702 def __init__(self, entities: _Optional[_Iterable[_Union[Entity, _Mapping]]] = ..., direction: _Optional[_Union[ModelVersionLineageDirection, str]] = ...) -> None: ... 703 704 class EmitModelVersionLineageRequest(_message.Message): 705 __slots__ = ("name", "version", "model_version_lineage_info", "securable") 706 NAME_FIELD_NUMBER: _ClassVar[int] 707 VERSION_FIELD_NUMBER: _ClassVar[int] 708 MODEL_VERSION_LINEAGE_INFO_FIELD_NUMBER: _ClassVar[int] 709 SECURABLE_FIELD_NUMBER: _ClassVar[int] 710 name: str 711 version: str 712 model_version_lineage_info: ModelVersionLineageInfo 713 securable: Securable 714 def __init__(self, name: _Optional[str] = ..., version: _Optional[str] = ..., model_version_lineage_info: _Optional[_Union[ModelVersionLineageInfo, _Mapping]] = ..., securable: _Optional[_Union[Securable, _Mapping]] = ...) -> None: ... 715 716 class EmitModelVersionLineageResponse(_message.Message): 717 __slots__ = () 718 def __init__(self) -> None: ... 719 720 class IsDatabricksSdkModelsArtifactRepositoryEnabledRequest(_message.Message): 721 __slots__ = () 722 def __init__(self) -> None: ... 723 724 class IsDatabricksSdkModelsArtifactRepositoryEnabledResponse(_message.Message): 725 __slots__ = ("is_databricks_sdk_models_artifact_repository_enabled",) 726 IS_DATABRICKS_SDK_MODELS_ARTIFACT_REPOSITORY_ENABLED_FIELD_NUMBER: _ClassVar[int] 727 is_databricks_sdk_models_artifact_repository_enabled: bool 728 def __init__(self, is_databricks_sdk_models_artifact_repository_enabled: bool = ...) -> None: ...