qwak-core 0.1.37__py3-none-any.whl → 0.5.19__py3-none-any.whl

_qwak_proto/qwak/deployment/deployment_pb2.pyi

@@ -142,6 +142,7 @@ class _GpuTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTy
     NVIDIA_A100: _GpuType.ValueType  # 3
     NVIDIA_T4: _GpuType.ValueType  # 4
     NVIDIA_A10G: _GpuType.ValueType  # 5
+    NVIDIA_L4: _GpuType.ValueType  # 6
 
 class GpuType(_GpuType, metaclass=_GpuTypeEnumTypeWrapper): ...
 
@@ -151,6 +152,7 @@ NVIDIA_V100: GpuType.ValueType  # 2
 NVIDIA_A100: GpuType.ValueType  # 3
 NVIDIA_T4: GpuType.ValueType  # 4
 NVIDIA_A10G: GpuType.ValueType  # 5
+NVIDIA_L4: GpuType.ValueType  # 6
 global___GpuType = GpuType
 
 class _MemoryUnit:
@@ -299,6 +301,21 @@ DEPLOYMENT_PHASE_FAILED: DeploymentPhaseStatus.ValueType  # 5
 DEPLOYMENT_PHASE_ABORTED: DeploymentPhaseStatus.ValueType  # 6
 global___DeploymentPhaseStatus = DeploymentPhaseStatus
 
+class QwakModeRegistryIdentifier(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ID_FIELD_NUMBER: builtins.int
+    id: builtins.str
+    """The model registry id"""
+    def __init__(
+        self,
+        *,
+        id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["id", b"id"]) -> None: ...
+
+global___QwakModeRegistryIdentifier = QwakModeRegistryIdentifier
+
 class HostingService(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -671,6 +688,7 @@ class AdvancedDeploymentOptions(google.protobuf.message.Message):
     MAX_BATCH_SIZE_FIELD_NUMBER: builtins.int
     DEPLOYMENT_PROCESS_TIMEOUT_LIMIT_FIELD_NUMBER: builtins.int
     PURCHASE_OPTION_FIELD_NUMBER: builtins.int
+    SERVICE_ACCOUNT_KEY_SECRET_NAME_FIELD_NUMBER: builtins.int
     number_of_http_server_workers: builtins.int
     """Amount of http server workers"""
     http_request_timeout_ms: builtins.int
@@ -686,6 +704,8 @@ class AdvancedDeploymentOptions(google.protobuf.message.Message):
     """The timeout for the deployment (in seconds)"""
     purchase_option: builtins.str
     """Whether it is spot/ondemand (default - spot)"""
+    service_account_key_secret_name: builtins.str
+    """Service account key secret name for gcp"""
     def __init__(
         self,
         *,
@@ -696,9 +716,10 @@ class AdvancedDeploymentOptions(google.protobuf.message.Message):
         max_batch_size: builtins.int = ...,
         deployment_process_timeout_limit: builtins.int = ...,
         purchase_option: builtins.str = ...,
+        service_account_key_secret_name: builtins.str = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["daemon_mode", b"daemon_mode"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["custom_iam_role_arn", b"custom_iam_role_arn", "daemon_mode", b"daemon_mode", "deployment_process_timeout_limit", b"deployment_process_timeout_limit", "http_request_timeout_ms", b"http_request_timeout_ms", "max_batch_size", b"max_batch_size", "number_of_http_server_workers", b"number_of_http_server_workers", "purchase_option", b"purchase_option"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["custom_iam_role_arn", b"custom_iam_role_arn", "daemon_mode", b"daemon_mode", "deployment_process_timeout_limit", b"deployment_process_timeout_limit", "http_request_timeout_ms", b"http_request_timeout_ms", "max_batch_size", b"max_batch_size", "number_of_http_server_workers", b"number_of_http_server_workers", "purchase_option", b"purchase_option", "service_account_key_secret_name", b"service_account_key_secret_name"]) -> None: ...
 
 global___AdvancedDeploymentOptions = AdvancedDeploymentOptions
 
@@ -722,6 +743,7 @@ class DeploymentDetails(google.protobuf.message.Message):
 
     LAST_DEPLOYED_FIELD_NUMBER: builtins.int
     BUILD_ID_FIELD_NUMBER: builtins.int
+    QWAK_MODEL_REGISTRY_IDENTIFIER_FIELD_NUMBER: builtins.int
     NUMBER_OF_PODS_FIELD_NUMBER: builtins.int
     CPU_FRACTION_FIELD_NUMBER: builtins.int
     MEMORY_AMOUNT_FIELD_NUMBER: builtins.int
@@ -745,11 +767,14 @@ class DeploymentDetails(google.protobuf.message.Message):
     STATUS_FIELD_NUMBER: builtins.int
     DEPLOYMENT_ID_FIELD_NUMBER: builtins.int
     FAILURE_DETAILS_FIELD_NUMBER: builtins.int
+    SERVICE_ACCOUNT_KEY_SECRET_NAME_FIELD_NUMBER: builtins.int
     @property
     def last_deployed(self) -> google.protobuf.timestamp_pb2.Timestamp:
         """Deployed date"""
     build_id: builtins.str
     """The build id that deployed"""
+    @property
+    def qwak_model_registry_identifier(self) -> global___QwakModeRegistryIdentifier: ...
     number_of_pods: builtins.int
     """Number of pods to deploy"""
     cpu_fraction: builtins.float
@@ -801,11 +826,14 @@ class DeploymentDetails(google.protobuf.message.Message):
     """Deployment id"""
     @property
     def failure_details(self) -> global___DeploymentFailureDetails: ...
+    service_account_key_secret_name: builtins.str
+    """Service account key secret name for gcp"""
     def __init__(
         self,
         *,
         last_deployed: google.protobuf.timestamp_pb2.Timestamp | None = ...,
         build_id: builtins.str = ...,
+        qwak_model_registry_identifier: global___QwakModeRegistryIdentifier | None = ...,
         number_of_pods: builtins.int = ...,
         cpu_fraction: builtins.float = ...,
         memory_amount: builtins.int = ...,
@@ -829,9 +857,11 @@ class DeploymentDetails(google.protobuf.message.Message):
         status: global___ModelDeploymentStatus.ValueType = ...,
         deployment_id: builtins.str = ...,
         failure_details: global___DeploymentFailureDetails | None = ...,
+        service_account_key_secret_name: builtins.str = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["daemon_mode", b"daemon_mode", "failure_details", b"failure_details", "last_deployed", b"last_deployed", "serving_strategy", b"serving_strategy", "undeployment_time", b"undeployment_time", "variation", b"variation"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["available_replicas", b"available_replicas", "build_id", b"build_id", "cpu_fraction", b"cpu_fraction", "custom_iam_role_arn", b"custom_iam_role_arn", "daemon_mode", b"daemon_mode", "deployment_id", b"deployment_id", "deployment_process_timeout_limit", b"deployment_process_timeout_limit", "environment_id", b"environment_id", "environment_variables", b"environment_variables", "failure_details", b"failure_details", "gpu_amount", b"gpu_amount", "gpu_type", b"gpu_type", "http_request_timeout_ms", b"http_request_timeout_ms", "kube_deployment_type", b"kube_deployment_type", "last_deployed", b"last_deployed", "max_batch_size", b"max_batch_size", "memory_amount", b"memory_amount", "memory_units", b"memory_units", "number_of_pods", b"number_of_pods", "number_of_workers", b"number_of_workers", "purchase_option", b"purchase_option", "serving_strategy", b"serving_strategy", "status", b"status", "undeployment_time", b"undeployment_time", "variation", b"variation"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["build_id", b"build_id", "daemon_mode", b"daemon_mode", "failure_details", b"failure_details", "image_identifier", b"image_identifier", "last_deployed", b"last_deployed", "qwak_model_registry_identifier", b"qwak_model_registry_identifier", "serving_strategy", b"serving_strategy", "undeployment_time", b"undeployment_time", "variation", b"variation"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["available_replicas", b"available_replicas", "build_id", b"build_id", "cpu_fraction", b"cpu_fraction", "custom_iam_role_arn", b"custom_iam_role_arn", "daemon_mode", b"daemon_mode", "deployment_id", b"deployment_id", "deployment_process_timeout_limit", b"deployment_process_timeout_limit", "environment_id", b"environment_id", "environment_variables", b"environment_variables", "failure_details", b"failure_details", "gpu_amount", b"gpu_amount", "gpu_type", b"gpu_type", "http_request_timeout_ms", b"http_request_timeout_ms", "image_identifier", b"image_identifier", "kube_deployment_type", b"kube_deployment_type", "last_deployed", b"last_deployed", "max_batch_size", b"max_batch_size", "memory_amount", b"memory_amount", "memory_units", b"memory_units", "number_of_pods", b"number_of_pods", "number_of_workers", b"number_of_workers", "purchase_option", b"purchase_option", "qwak_model_registry_identifier", b"qwak_model_registry_identifier", "service_account_key_secret_name", b"service_account_key_secret_name", "serving_strategy", b"serving_strategy", "status", b"status", "undeployment_time", b"undeployment_time", "variation", b"variation"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["image_identifier", b"image_identifier"]) -> typing_extensions.Literal["build_id", "qwak_model_registry_identifier"] | None: ...
 
 global___DeploymentDetails = DeploymentDetails
 
@@ -888,6 +918,7 @@ class DeploymentBrief(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     BUILD_ID_FIELD_NUMBER: builtins.int
+    QWAK_MODEL_REGISTRY_IDENTIFIER_FIELD_NUMBER: builtins.int
     STATUS_FIELD_NUMBER: builtins.int
     DEPLOYED_BY_FIELD_NUMBER: builtins.int
     FAILURE_REASON_CODE_FIELD_NUMBER: builtins.int
@@ -900,6 +931,8 @@ class DeploymentBrief(google.protobuf.message.Message):
     DEPLOYMENT_ID_FIELD_NUMBER: builtins.int
     build_id: builtins.str
     """the deployed build id"""
+    @property
+    def qwak_model_registry_identifier(self) -> global___QwakModeRegistryIdentifier: ...
     status: global___ModelDeploymentStatus.ValueType
     """The Deployment status - SUCCESSFUL_DEPLOYMENT or FAILED_DEPLOYMENT"""
     deployed_by: builtins.str
@@ -924,6 +957,7 @@ class DeploymentBrief(google.protobuf.message.Message):
         self,
         *,
         build_id: builtins.str = ...,
+        qwak_model_registry_identifier: global___QwakModeRegistryIdentifier | None = ...,
         status: global___ModelDeploymentStatus.ValueType = ...,
         deployed_by: builtins.str = ...,
         failure_reason_code: builtins.str = ...,
@@ -935,8 +969,9 @@ class DeploymentBrief(google.protobuf.message.Message):
         variation_protect_state: builtins.bool = ...,
         deployment_id: builtins.str = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["deployment_timestamp", b"deployment_timestamp"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id", "deployed_by", b"deployed_by", "deployment_id", b"deployment_id", "deployment_timestamp", b"deployment_timestamp", "environment_id", b"environment_id", "failure_message", b"failure_message", "failure_reason_code", b"failure_reason_code", "status", b"status", "technical_details", b"technical_details", "variation_name", b"variation_name", "variation_protect_state", b"variation_protect_state"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["build_id", b"build_id", "deployment_timestamp", b"deployment_timestamp", "image_identifier", b"image_identifier", "qwak_model_registry_identifier", b"qwak_model_registry_identifier"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id", "deployed_by", b"deployed_by", "deployment_id", b"deployment_id", "deployment_timestamp", b"deployment_timestamp", "environment_id", b"environment_id", "failure_message", b"failure_message", "failure_reason_code", b"failure_reason_code", "image_identifier", b"image_identifier", "qwak_model_registry_identifier", b"qwak_model_registry_identifier", "status", b"status", "technical_details", b"technical_details", "variation_name", b"variation_name", "variation_protect_state", b"variation_protect_state"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["image_identifier", b"image_identifier"]) -> typing_extensions.Literal["build_id", "qwak_model_registry_identifier"] | None: ...
 
 global___DeploymentBrief = DeploymentBrief
 
@@ -1246,6 +1281,7 @@ class EnvironmentDeploymentMessage(google.protobuf.message.Message):
     MODEL_ID_FIELD_NUMBER: builtins.int
     BRANCH_ID_FIELD_NUMBER: builtins.int
     BUILD_ID_FIELD_NUMBER: builtins.int
+    QWAK_MODEL_REGISTRY_IDENTIFIER_FIELD_NUMBER: builtins.int
     HOSTING_SERVICE_FIELD_NUMBER: builtins.int
     MODEL_UUID_FIELD_NUMBER: builtins.int
     model_id: builtins.str
@@ -1255,6 +1291,8 @@ class EnvironmentDeploymentMessage(google.protobuf.message.Message):
     build_id: builtins.str
     """Build ID (which must be successful) to deploy"""
     @property
+    def qwak_model_registry_identifier(self) -> global___QwakModeRegistryIdentifier: ...
+    @property
     def hosting_service(self) -> global___HostingService:
         """Downstream hosting service that will serve the Qwak model"""
     model_uuid: builtins.str
@@ -1264,11 +1302,13 @@ class EnvironmentDeploymentMessage(google.protobuf.message.Message):
         model_id: builtins.str = ...,
         branch_id: builtins.str = ...,
         build_id: builtins.str = ...,
+        qwak_model_registry_identifier: global___QwakModeRegistryIdentifier | None = ...,
         hosting_service: global___HostingService | None = ...,
         model_uuid: builtins.str = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["hosting_service", b"hosting_service"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["branch_id", b"branch_id", "build_id", b"build_id", "hosting_service", b"hosting_service", "model_id", b"model_id", "model_uuid", b"model_uuid"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["build_id", b"build_id", "hosting_service", b"hosting_service", "image_identifier", b"image_identifier", "qwak_model_registry_identifier", b"qwak_model_registry_identifier"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["branch_id", b"branch_id", "build_id", b"build_id", "hosting_service", b"hosting_service", "image_identifier", b"image_identifier", "model_id", b"model_id", "model_uuid", b"model_uuid", "qwak_model_registry_identifier", b"qwak_model_registry_identifier"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["image_identifier", b"image_identifier"]) -> typing_extensions.Literal["build_id", "qwak_model_registry_identifier"] | None: ...
 
 global___EnvironmentDeploymentMessage = EnvironmentDeploymentMessage
 
@@ -1507,15 +1547,18 @@ class BuildDeploymentStatus(google.protobuf.message.Message):
 
     BUILD_UUID_FIELD_NUMBER: builtins.int
     IS_DEPLOYED_FIELD_NUMBER: builtins.int
+    IS_IN_USE_FIELD_NUMBER: builtins.int
     build_uuid: builtins.str
     is_deployed: builtins.bool
+    is_in_use: builtins.bool
     def __init__(
         self,
         *,
         build_uuid: builtins.str = ...,
         is_deployed: builtins.bool = ...,
+        is_in_use: builtins.bool = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_uuid", b"build_uuid", "is_deployed", b"is_deployed"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["build_uuid", b"build_uuid", "is_deployed", b"is_deployed", "is_in_use", b"is_in_use"]) -> None: ...
 
 global___BuildDeploymentStatus = BuildDeploymentStatus