frogml 1.2.40__py3-none-any.whl → 1.2.42__py3-none-any.whl

frogml/__init__.py

@@ -1,5 +1,5 @@
 __author__ = "jfrog"
-__version__ = "1.2.40"
+__version__ = "1.2.42"
 
 from frogml.sdk.model.decorators.api import api_decorator as api
 from frogml.sdk.model_loggers.model_logger import load_model, log_model

frogml/_proto/com/jfrog/ml/model/deployment/v1/auto_scaling_pb2.py

@@ -0,0 +1,43 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: com/jfrog/ml/model/deployment/v1/auto_scaling.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3com/jfrog/ml/model/deployment/v1/auto_scaling.proto\x12 com.jfrog.ml.model.deployment.v1\"\x8e\x02\n\x11\x41utoScalingConfig\x12\x19\n\x11min_replica_count\x18\x01 \x01(\x05\x12\x19\n\x11max_replica_count\x18\x02 \x01(\x05\x12\"\n\x18polling_interval_seconds\x18\x03 \x01(\x05H\x00\x12\"\n\x18\x63ool_down_period_seconds\x18\x04 \x01(\x05H\x01\x12;\n\x08triggers\x18\x05 \x03(\x0b\x32).com.jfrog.ml.model.deployment.v1.TriggerB\x1e\n\x1cpolling_interval_seconds_optB\x1e\n\x1c\x63ool_down_period_seconds_opt\"k\n\x07Trigger\x12P\n\x15scalar_scaling_metric\x18\x01 \x01(\x0b\x32/.com.jfrog.ml.model.deployment.v1.ScalarTriggerH\x00\x42\x0e\n\x0ctrigger_type\"\xd9\x01\n\rScalarTrigger\x12K\n\rmetric_source\x18\x01 \x01(\x0b\x32\x34.com.jfrog.ml.model.deployment.v1.ScalarMetricSource\x12K\n\x10\x61ggregation_type\x18\x02 \x01(\x0e\x32\x31.com.jfrog.ml.model.deployment.v1.AggregationType\x12\x11\n\tthreshold\x18\x03 \x01(\x05\x12\x1b\n\x13time_period_seconds\x18\x04 \x01(\x05\"\xcf\x02\n\x12ScalarMetricSource\x12\x46\n\x03\x63pu\x18\x01 \x01(\x0b\x32\x37.com.jfrog.ml.model.deployment.v1.ScalarMetricSourceCpuH\x00\x12S\n\nerror_rate\x18\x02 \x01(\x0b\x32=.com.jfrog.ml.model.deployment.v1.ScalarMetricSourceErrorRateH\x00\x12\x46\n\x03gpu\x18\x03 \x01(\x0b\x32\x37.com.jfrog.ml.model.deployment.v1.ScalarMetricSourceGpuH\x00\x12L\n\x06memory\x18\x04 \x01(\x0b\x32:.com.jfrog.ml.model.deployment.v1.ScalarMetricSourceMemoryH\x00\x42\x06\n\x04type\"\x17\n\x15ScalarMetricSourceCpu\"\x17\n\x15ScalarMetricSourceGpu\"\x1a\n\x18ScalarMetricSourceMemory\"E\n\x1bScalarMetricSourceErrorRate\x12\x14\n\nerror_code\x18\x04 \x01(\tH\x00\x42\x10\n\x0e\x65rror_code_opt*\x83\x02\n\x0f\x41ggregationType\x12\x1c\n\x18\x41GGREGATION_TYPE_INVALID\x10\x00\x12\x18\n\x14\x41GGREGATION_TYPE_MIN\x10\x01\x12\x18\n\x14\x41GGREGATION_TYPE_MAX\x10\x02\x12\x1c\n\x18\x41GGREGATION_TYPE_AVERAGE\x10\x03\x12\x18\n\x14\x41GGREGATION_TYPE_SUM\x10\x04\x12\x18\n\x14\x41GGREGATION_TYPE_P50\x10\x05\x12\x18\n\x14\x41GGREGATION_TYPE_P90\x10\x06\x12\x18\n\x14\x41GGREGATION_TYPE_P95\x10\x07\x12\x18\n\x14\x41GGREGATION_TYPE_P99\x10\x08\x42(\n$com.jfrog.ml.model.deployment.v1.apiP\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'com.jfrog.ml.model.deployment.v1.auto_scaling_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n$com.jfrog.ml.model.deployment.v1.apiP\001'
+  _globals['_AGGREGATIONTYPE']._serialized_start=1179
+  _globals['_AGGREGATIONTYPE']._serialized_end=1438
+  _globals['_AUTOSCALINGCONFIG']._serialized_start=90
+  _globals['_AUTOSCALINGCONFIG']._serialized_end=360
+  _globals['_TRIGGER']._serialized_start=362
+  _globals['_TRIGGER']._serialized_end=469
+  _globals['_SCALARTRIGGER']._serialized_start=472
+  _globals['_SCALARTRIGGER']._serialized_end=689
+  _globals['_SCALARMETRICSOURCE']._serialized_start=692
+  _globals['_SCALARMETRICSOURCE']._serialized_end=1027
+  _globals['_SCALARMETRICSOURCECPU']._serialized_start=1029
+  _globals['_SCALARMETRICSOURCECPU']._serialized_end=1052
+  _globals['_SCALARMETRICSOURCEGPU']._serialized_start=1054
+  _globals['_SCALARMETRICSOURCEGPU']._serialized_end=1077
+  _globals['_SCALARMETRICSOURCEMEMORY']._serialized_start=1079
+  _globals['_SCALARMETRICSOURCEMEMORY']._serialized_end=1105
+  _globals['_SCALARMETRICSOURCEERRORRATE']._serialized_start=1107
+  _globals['_SCALARMETRICSOURCEERRORRATE']._serialized_end=1176
+# @@protoc_insertion_point(module_scope)

frogml/_proto/com/jfrog/ml/model/deployment/v1/auto_scaling_pb2.pyi

@@ -0,0 +1,218 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+import sys
+import typing
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _AggregationType:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _AggregationTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_AggregationType.ValueType], builtins.type):
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    AGGREGATION_TYPE_INVALID: _AggregationType.ValueType  # 0
+    AGGREGATION_TYPE_MIN: _AggregationType.ValueType  # 1
+    AGGREGATION_TYPE_MAX: _AggregationType.ValueType  # 2
+    AGGREGATION_TYPE_AVERAGE: _AggregationType.ValueType  # 3
+    AGGREGATION_TYPE_SUM: _AggregationType.ValueType  # 4
+    AGGREGATION_TYPE_P50: _AggregationType.ValueType  # 5
+    AGGREGATION_TYPE_P90: _AggregationType.ValueType  # 6
+    AGGREGATION_TYPE_P95: _AggregationType.ValueType  # 7
+    AGGREGATION_TYPE_P99: _AggregationType.ValueType  # 8
+
+class AggregationType(_AggregationType, metaclass=_AggregationTypeEnumTypeWrapper): ...
+
+AGGREGATION_TYPE_INVALID: AggregationType.ValueType  # 0
+AGGREGATION_TYPE_MIN: AggregationType.ValueType  # 1
+AGGREGATION_TYPE_MAX: AggregationType.ValueType  # 2
+AGGREGATION_TYPE_AVERAGE: AggregationType.ValueType  # 3
+AGGREGATION_TYPE_SUM: AggregationType.ValueType  # 4
+AGGREGATION_TYPE_P50: AggregationType.ValueType  # 5
+AGGREGATION_TYPE_P90: AggregationType.ValueType  # 6
+AGGREGATION_TYPE_P95: AggregationType.ValueType  # 7
+AGGREGATION_TYPE_P99: AggregationType.ValueType  # 8
+global___AggregationType = AggregationType
+
+@typing.final
+class AutoScalingConfig(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MIN_REPLICA_COUNT_FIELD_NUMBER: builtins.int
+    MAX_REPLICA_COUNT_FIELD_NUMBER: builtins.int
+    POLLING_INTERVAL_SECONDS_FIELD_NUMBER: builtins.int
+    COOL_DOWN_PERIOD_SECONDS_FIELD_NUMBER: builtins.int
+    TRIGGERS_FIELD_NUMBER: builtins.int
+    min_replica_count: builtins.int
+    """The minimum number of replicas will scale the resource down to"""
+    max_replica_count: builtins.int
+    """The maximum number of replicas of the target resource"""
+    polling_interval_seconds: builtins.int
+    cool_down_period_seconds: builtins.int
+    @property
+    def triggers(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Trigger]:
+        """List of triggers based on which to scale up/down."""
+
+    def __init__(
+        self,
+        *,
+        min_replica_count: builtins.int = ...,
+        max_replica_count: builtins.int = ...,
+        polling_interval_seconds: builtins.int = ...,
+        cool_down_period_seconds: builtins.int = ...,
+        triggers: collections.abc.Iterable[global___Trigger] | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["cool_down_period_seconds", b"cool_down_period_seconds", "cool_down_period_seconds_opt", b"cool_down_period_seconds_opt", "polling_interval_seconds", b"polling_interval_seconds", "polling_interval_seconds_opt", b"polling_interval_seconds_opt"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["cool_down_period_seconds", b"cool_down_period_seconds", "cool_down_period_seconds_opt", b"cool_down_period_seconds_opt", "max_replica_count", b"max_replica_count", "min_replica_count", b"min_replica_count", "polling_interval_seconds", b"polling_interval_seconds", "polling_interval_seconds_opt", b"polling_interval_seconds_opt", "triggers", b"triggers"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["cool_down_period_seconds_opt", b"cool_down_period_seconds_opt"]) -> typing.Literal["cool_down_period_seconds"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["polling_interval_seconds_opt", b"polling_interval_seconds_opt"]) -> typing.Literal["polling_interval_seconds"] | None: ...
+
+global___AutoScalingConfig = AutoScalingConfig
+
+@typing.final
+class Trigger(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SCALAR_SCALING_METRIC_FIELD_NUMBER: builtins.int
+    @property
+    def scalar_scaling_metric(self) -> global___ScalarTrigger:
+        """A generic trigger that works by calculating aggregates on scalar time series:"""
+
+    def __init__(
+        self,
+        *,
+        scalar_scaling_metric: global___ScalarTrigger | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["scalar_scaling_metric", b"scalar_scaling_metric", "trigger_type", b"trigger_type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["scalar_scaling_metric", b"scalar_scaling_metric", "trigger_type", b"trigger_type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["trigger_type", b"trigger_type"]) -> typing.Literal["scalar_scaling_metric"] | None: ...
+
+global___Trigger = Trigger
+
+@typing.final
+class ScalarTrigger(google.protobuf.message.Message):
+    """A generic scalar trigger that works by calculating aggregates on scalar time series and firing based on a
+    threshold.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    METRIC_SOURCE_FIELD_NUMBER: builtins.int
+    AGGREGATION_TYPE_FIELD_NUMBER: builtins.int
+    THRESHOLD_FIELD_NUMBER: builtins.int
+    TIME_PERIOD_SECONDS_FIELD_NUMBER: builtins.int
+    aggregation_type: global___AggregationType.ValueType
+    """The type of the aggregation to run over the time series"""
+    threshold: builtins.int
+    """Value to trigger from"""
+    time_period_seconds: builtins.int
+    """The period over which to aggregate the metric."""
+    @property
+    def metric_source(self) -> global___ScalarMetricSource:
+        """The source of the metric"""
+
+    def __init__(
+        self,
+        *,
+        metric_source: global___ScalarMetricSource | None = ...,
+        aggregation_type: global___AggregationType.ValueType = ...,
+        threshold: builtins.int = ...,
+        time_period_seconds: builtins.int = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["metric_source", b"metric_source"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["aggregation_type", b"aggregation_type", "metric_source", b"metric_source", "threshold", b"threshold", "time_period_seconds", b"time_period_seconds"]) -> None: ...
+
+global___ScalarTrigger = ScalarTrigger
+
+@typing.final
+class ScalarMetricSource(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CPU_FIELD_NUMBER: builtins.int
+    ERROR_RATE_FIELD_NUMBER: builtins.int
+    GPU_FIELD_NUMBER: builtins.int
+    MEMORY_FIELD_NUMBER: builtins.int
+    @property
+    def cpu(self) -> global___ScalarMetricSourceCpu: ...
+    @property
+    def error_rate(self) -> global___ScalarMetricSourceErrorRate: ...
+    @property
+    def gpu(self) -> global___ScalarMetricSourceGpu: ...
+    @property
+    def memory(self) -> global___ScalarMetricSourceMemory: ...
+    def __init__(
+        self,
+        *,
+        cpu: global___ScalarMetricSourceCpu | None = ...,
+        error_rate: global___ScalarMetricSourceErrorRate | None = ...,
+        gpu: global___ScalarMetricSourceGpu | None = ...,
+        memory: global___ScalarMetricSourceMemory | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["cpu", b"cpu", "error_rate", b"error_rate", "gpu", b"gpu", "memory", b"memory", "type", b"type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["cpu", b"cpu", "error_rate", b"error_rate", "gpu", b"gpu", "memory", b"memory", "type", b"type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["type", b"type"]) -> typing.Literal["cpu", "error_rate", "gpu", "memory"] | None: ...
+
+global___ScalarMetricSource = ScalarMetricSource
+
+@typing.final
+class ScalarMetricSourceCpu(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ScalarMetricSourceCpu = ScalarMetricSourceCpu
+
+@typing.final
+class ScalarMetricSourceGpu(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ScalarMetricSourceGpu = ScalarMetricSourceGpu
+
+@typing.final
+class ScalarMetricSourceMemory(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ScalarMetricSourceMemory = ScalarMetricSourceMemory
+
+@typing.final
+class ScalarMetricSourceErrorRate(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ERROR_CODE_FIELD_NUMBER: builtins.int
+    error_code: builtins.str
+    def __init__(
+        self,
+        *,
+        error_code: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["error_code", b"error_code", "error_code_opt", b"error_code_opt"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["error_code", b"error_code", "error_code_opt", b"error_code_opt"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["error_code_opt", b"error_code_opt"]) -> typing.Literal["error_code"] | None: ...
+
+global___ScalarMetricSourceErrorRate = ScalarMetricSourceErrorRate

frogml/_proto/com/jfrog/ml/model/deployment/v1/auto_scaling_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

frogml/_proto/com/jfrog/ml/model/deployment/v1/deployment_pb2.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: com/jfrog/ml/model/deployment/v1/deployment.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+from frogml._proto.com.jfrog.ml.model.deployment.v1 import environment_variables_configuration_pb2 as com_dot_jfrog_dot_ml_dot_model_dot_deployment_dot_v1_dot_environment__variables__configuration__pb2
+from frogml._proto.com.jfrog.ml.model.deployment.v1 import model_artifact_identifier_pb2 as com_dot_jfrog_dot_ml_dot_model_dot_deployment_dot_v1_dot_model__artifact__identifier__pb2
+from frogml._proto.com.jfrog.ml.model.deployment.v1 import probes_configuration_pb2 as com_dot_jfrog_dot_ml_dot_model_dot_deployment_dot_v1_dot_probes__configuration__pb2
+from frogml._proto.com.jfrog.ml.model.deployment.v1 import realtime_deployment_pb2 as com_dot_jfrog_dot_ml_dot_model_dot_deployment_dot_v1_dot_realtime__deployment__pb2
+from frogml._proto.com.jfrog.ml.model.deployment.v1 import resource_configuration_pb2 as com_dot_jfrog_dot_ml_dot_model_dot_deployment_dot_v1_dot_resource__configuration__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n1com/jfrog/ml/model/deployment/v1/deployment.proto\x12 com.jfrog.ml.model.deployment.v1\x1a\x1fgoogle/protobuf/timestamp.proto\x1aJcom/jfrog/ml/model/deployment/v1/environment_variables_configuration.proto\x1a@com/jfrog/ml/model/deployment/v1/model_artifact_identifier.proto\x1a;com/jfrog/ml/model/deployment/v1/probes_configuration.proto\x1a:com/jfrog/ml/model/deployment/v1/realtime_deployment.proto\x1a=com/jfrog/ml/model/deployment/v1/resource_configuration.proto\"\xc6\x02\n\x1dMultipleEnvironmentDeployment\x12\\\n\x19model_artifact_identifier\x18\x01 \x01(\x0b\x32\x39.com.jfrog.ml.model.deployment.v1.ModelArtifactIdentifier\x12\x65\n\x0b\x64\x65ployments\x18\x02 \x03(\x0b\x32P.com.jfrog.ml.model.deployment.v1.MultipleEnvironmentDeployment.DeploymentsEntry\x1a`\n\x10\x44\x65ploymentsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12;\n\x05value\x18\x02 \x01(\x0b\x32,.com.jfrog.ml.model.deployment.v1.Deployment:\x02\x38\x01\"\x94\x03\n\nDeployment\x12W\n\x16resource_configuration\x18\x01 \x01(\x0b\x32\x37.com.jfrog.ml.model.deployment.v1.ResourceConfiguration\x12p\n#environment_variables_configuration\x18\x02 \x01(\x0b\x32\x43.com.jfrog.ml.model.deployment.v1.EnvironmentVariablesConfiguration\x12S\n\x14probes_configuration\x18\x03 \x01(\x0b\x32\x35.com.jfrog.ml.model.deployment.v1.ProbesConfiguration\x12S\n\x13realtime_deployment\x18\x04 \x01(\x0b\x32\x34.com.jfrog.ml.model.deployment.v1.RealtimeDeploymentH\x00\x42\x11\n\x0f\x64\x65ployment_type\"\xd8\x03\n\x15ModelDeploymentStatus\x12g\n\x1emodel_deployment_status_failed\x18\x01 \x01(\x0b\x32=.com.jfrog.ml.model.deployment.v1.ModelDeploymentStatusFailedH\x00\x12o\n\"model_deployment_status_successful\x18\x02 \x01(\x0b\x32\x41.com.jfrog.ml.model.deployment.v1.ModelDeploymentStatusSuccessfulH\x00\x12p\n#model_deployment_status_in_progress\x18\x03 \x01(\x0b\x32\x41.com.jfrog.ml.model.deployment.v1.ModelDeploymentStatusInProgressH\x00\x12i\n\x1fmodel_deployment_status_failing\x18\x04 \x01(\x0b\x32>.com.jfrog.ml.model.deployment.v1.ModelDeploymentStatusFailingH\x00\x42\x08\n\x06status\"\x1d\n\x1bModelDeploymentStatusFailed\"!\n\x1fModelDeploymentStatusSuccessful\"!\n\x1fModelDeploymentStatusInProgress\"\x1e\n\x1cModelDeploymentStatusFailing\"\xcd\x01\n\x12\x44\x65ploymentMetadata\x12\x15\n\rdeployment_id\x18\x01 \x01(\t\x12G\n\x06status\x18\x02 \x01(\x0b\x32\x37.com.jfrog.ml.model.deployment.v1.ModelDeploymentStatus\x12\x13\n\x0b\x65nvironment\x18\x03 \x01(\t\x12.\n\ncreated_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\ncreated_by\x18\x05 \x01(\t\"\xa0\x02\n\x1aModelDeploymentInformation\x12\\\n\x19model_artifact_identifier\x18\x01 \x01(\x0b\x32\x39.com.jfrog.ml.model.deployment.v1.ModelArtifactIdentifier\x12K\n\x15model_deployment_spec\x18\x02 \x01(\x0b\x32,.com.jfrog.ml.model.deployment.v1.Deployment\x12W\n\x19model_deployment_metadata\x18\x03 \x01(\x0b\x32\x34.com.jfrog.ml.model.deployment.v1.DeploymentMetadataB(\n$com.jfrog.ml.model.deployment.v1.apiP\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'com.jfrog.ml.model.deployment.v1.deployment_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n$com.jfrog.ml.model.deployment.v1.apiP\001'
+  _globals['_MULTIPLEENVIRONMENTDEPLOYMENT_DEPLOYMENTSENTRY']._options = None
+  _globals['_MULTIPLEENVIRONMENTDEPLOYMENT_DEPLOYMENTSENTRY']._serialized_options = b'8\001'
+  _globals['_MULTIPLEENVIRONMENTDEPLOYMENT']._serialized_start=447
+  _globals['_MULTIPLEENVIRONMENTDEPLOYMENT']._serialized_end=773
+  _globals['_MULTIPLEENVIRONMENTDEPLOYMENT_DEPLOYMENTSENTRY']._serialized_start=677
+  _globals['_MULTIPLEENVIRONMENTDEPLOYMENT_DEPLOYMENTSENTRY']._serialized_end=773
+  _globals['_DEPLOYMENT']._serialized_start=776
+  _globals['_DEPLOYMENT']._serialized_end=1180
+  _globals['_MODELDEPLOYMENTSTATUS']._serialized_start=1183
+  _globals['_MODELDEPLOYMENTSTATUS']._serialized_end=1655
+  _globals['_MODELDEPLOYMENTSTATUSFAILED']._serialized_start=1657
+  _globals['_MODELDEPLOYMENTSTATUSFAILED']._serialized_end=1686
+  _globals['_MODELDEPLOYMENTSTATUSSUCCESSFUL']._serialized_start=1688
+  _globals['_MODELDEPLOYMENTSTATUSSUCCESSFUL']._serialized_end=1721
+  _globals['_MODELDEPLOYMENTSTATUSINPROGRESS']._serialized_start=1723
+  _globals['_MODELDEPLOYMENTSTATUSINPROGRESS']._serialized_end=1756
+  _globals['_MODELDEPLOYMENTSTATUSFAILING']._serialized_start=1758
+  _globals['_MODELDEPLOYMENTSTATUSFAILING']._serialized_end=1788
+  _globals['_DEPLOYMENTMETADATA']._serialized_start=1791
+  _globals['_DEPLOYMENTMETADATA']._serialized_end=1996
+  _globals['_MODELDEPLOYMENTINFORMATION']._serialized_start=1999
+  _globals['_MODELDEPLOYMENTINFORMATION']._serialized_end=2287
+# @@protoc_insertion_point(module_scope)

frogml/_proto/com/jfrog/ml/model/deployment/v1/deployment_pb2.pyi

@@ -0,0 +1,258 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import collections.abc
+import frogml._proto.com.jfrog.ml.model.deployment.v1.environment_variables_configuration_pb2
+import frogml._proto.com.jfrog.ml.model.deployment.v1.model_artifact_identifier_pb2
+import frogml._proto.com.jfrog.ml.model.deployment.v1.probes_configuration_pb2
+import frogml._proto.com.jfrog.ml.model.deployment.v1.realtime_deployment_pb2
+import frogml._proto.com.jfrog.ml.model.deployment.v1.resource_configuration_pb2
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.message
+import google.protobuf.timestamp_pb2
+import typing
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class MultipleEnvironmentDeployment(google.protobuf.message.Message):
+    """Multi Environment Deployment."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
+    class DeploymentsEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+        @property
+        def value(self) -> global___Deployment: ...
+        def __init__(
+            self,
+            *,
+            key: builtins.str = ...,
+            value: global___Deployment | None = ...,
+        ) -> None: ...
+        def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
+
+    MODEL_ARTIFACT_IDENTIFIER_FIELD_NUMBER: builtins.int
+    DEPLOYMENTS_FIELD_NUMBER: builtins.int
+    @property
+    def model_artifact_identifier(self) -> frogml._proto.com.jfrog.ml.model.deployment.v1.model_artifact_identifier_pb2.ModelArtifactIdentifier:
+        """What should be deployed:"""
+
+    @property
+    def deployments(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___Deployment]:
+        """Where and how should it be deployed:
+        Multi environment deployment request. A mapping of Environment ID to deploy in, to Deployment
+        to deploy.
+        """
+
+    def __init__(
+        self,
+        *,
+        model_artifact_identifier: frogml._proto.com.jfrog.ml.model.deployment.v1.model_artifact_identifier_pb2.ModelArtifactIdentifier | None = ...,
+        deployments: collections.abc.Mapping[builtins.str, global___Deployment] | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["model_artifact_identifier", b"model_artifact_identifier"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["deployments", b"deployments", "model_artifact_identifier", b"model_artifact_identifier"]) -> None: ...
+
+global___MultipleEnvironmentDeployment = MultipleEnvironmentDeployment
+
+@typing.final
+class Deployment(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    RESOURCE_CONFIGURATION_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_VARIABLES_CONFIGURATION_FIELD_NUMBER: builtins.int
+    PROBES_CONFIGURATION_FIELD_NUMBER: builtins.int
+    REALTIME_DEPLOYMENT_FIELD_NUMBER: builtins.int
+    @property
+    def resource_configuration(self) -> frogml._proto.com.jfrog.ml.model.deployment.v1.resource_configuration_pb2.ResourceConfiguration:
+        """Resources of the deployment. CPU/RAM/ROM/GPU."""
+
+    @property
+    def environment_variables_configuration(self) -> frogml._proto.com.jfrog.ml.model.deployment.v1.environment_variables_configuration_pb2.EnvironmentVariablesConfiguration:
+        """Configuration of Environment Variables that should be available to the instances at runtime.
+        Additional Environment variables are going to be injected into the runtime container like the `HF_URL` and tokens
+        e.g. for JPD access. The Environment variables specified here are recessive and may be overwritten by the service.
+        """
+
+    @property
+    def probes_configuration(self) -> frogml._proto.com.jfrog.ml.model.deployment.v1.probes_configuration_pb2.ProbesConfiguration:
+        """Configuration of Probes that should be used to check the health of instances in the deployment."""
+
+    @property
+    def realtime_deployment(self) -> frogml._proto.com.jfrog.ml.model.deployment.v1.realtime_deployment_pb2.RealtimeDeployment:
+        """Deployment type for a Kubernetes deployment."""
+
+    def __init__(
+        self,
+        *,
+        resource_configuration: frogml._proto.com.jfrog.ml.model.deployment.v1.resource_configuration_pb2.ResourceConfiguration | None = ...,
+        environment_variables_configuration: frogml._proto.com.jfrog.ml.model.deployment.v1.environment_variables_configuration_pb2.EnvironmentVariablesConfiguration | None = ...,
+        probes_configuration: frogml._proto.com.jfrog.ml.model.deployment.v1.probes_configuration_pb2.ProbesConfiguration | None = ...,
+        realtime_deployment: frogml._proto.com.jfrog.ml.model.deployment.v1.realtime_deployment_pb2.RealtimeDeployment | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["deployment_type", b"deployment_type", "environment_variables_configuration", b"environment_variables_configuration", "probes_configuration", b"probes_configuration", "realtime_deployment", b"realtime_deployment", "resource_configuration", b"resource_configuration"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["deployment_type", b"deployment_type", "environment_variables_configuration", b"environment_variables_configuration", "probes_configuration", b"probes_configuration", "realtime_deployment", b"realtime_deployment", "resource_configuration", b"resource_configuration"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["deployment_type", b"deployment_type"]) -> typing.Literal["realtime_deployment"] | None: ...
+
+global___Deployment = Deployment
+
+@typing.final
+class ModelDeploymentStatus(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_DEPLOYMENT_STATUS_FAILED_FIELD_NUMBER: builtins.int
+    MODEL_DEPLOYMENT_STATUS_SUCCESSFUL_FIELD_NUMBER: builtins.int
+    MODEL_DEPLOYMENT_STATUS_IN_PROGRESS_FIELD_NUMBER: builtins.int
+    MODEL_DEPLOYMENT_STATUS_FAILING_FIELD_NUMBER: builtins.int
+    @property
+    def model_deployment_status_failed(self) -> global___ModelDeploymentStatusFailed:
+        """The deployment has failed."""
+
+    @property
+    def model_deployment_status_successful(self) -> global___ModelDeploymentStatusSuccessful:
+        """The deployment is successful."""
+
+    @property
+    def model_deployment_status_in_progress(self) -> global___ModelDeploymentStatusInProgress:
+        """The deployment is in progress."""
+
+    @property
+    def model_deployment_status_failing(self) -> global___ModelDeploymentStatusFailing:
+        """The deployment is failing and requires intervention"""
+
+    def __init__(
+        self,
+        *,
+        model_deployment_status_failed: global___ModelDeploymentStatusFailed | None = ...,
+        model_deployment_status_successful: global___ModelDeploymentStatusSuccessful | None = ...,
+        model_deployment_status_in_progress: global___ModelDeploymentStatusInProgress | None = ...,
+        model_deployment_status_failing: global___ModelDeploymentStatusFailing | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["model_deployment_status_failed", b"model_deployment_status_failed", "model_deployment_status_failing", b"model_deployment_status_failing", "model_deployment_status_in_progress", b"model_deployment_status_in_progress", "model_deployment_status_successful", b"model_deployment_status_successful", "status", b"status"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["model_deployment_status_failed", b"model_deployment_status_failed", "model_deployment_status_failing", b"model_deployment_status_failing", "model_deployment_status_in_progress", b"model_deployment_status_in_progress", "model_deployment_status_successful", b"model_deployment_status_successful", "status", b"status"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["status", b"status"]) -> typing.Literal["model_deployment_status_failed", "model_deployment_status_successful", "model_deployment_status_in_progress", "model_deployment_status_failing"] | None: ...
+
+global___ModelDeploymentStatus = ModelDeploymentStatus
+
+@typing.final
+class ModelDeploymentStatusFailed(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ModelDeploymentStatusFailed = ModelDeploymentStatusFailed
+
+@typing.final
+class ModelDeploymentStatusSuccessful(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ModelDeploymentStatusSuccessful = ModelDeploymentStatusSuccessful
+
+@typing.final
+class ModelDeploymentStatusInProgress(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ModelDeploymentStatusInProgress = ModelDeploymentStatusInProgress
+
+@typing.final
+class ModelDeploymentStatusFailing(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ModelDeploymentStatusFailing = ModelDeploymentStatusFailing
+
+@typing.final
+class DeploymentMetadata(google.protobuf.message.Message):
+    """Metadata about the deployment, such as status, environment, timestamp, and ID."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DEPLOYMENT_ID_FIELD_NUMBER: builtins.int
+    STATUS_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_FIELD_NUMBER: builtins.int
+    CREATED_AT_FIELD_NUMBER: builtins.int
+    CREATED_BY_FIELD_NUMBER: builtins.int
+    deployment_id: builtins.str
+    """The unique UUID of the deployment."""
+    environment: builtins.str
+    """The environment where the deployment is located."""
+    created_by: builtins.str
+    """The user who created the deployment."""
+    @property
+    def status(self) -> global___ModelDeploymentStatus:
+        """The status of the deployment."""
+
+    @property
+    def created_at(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """The timestamp of the deployments creation"""
+
+    def __init__(
+        self,
+        *,
+        deployment_id: builtins.str = ...,
+        status: global___ModelDeploymentStatus | None = ...,
+        environment: builtins.str = ...,
+        created_at: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        created_by: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["created_at", b"created_at", "status", b"status"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["created_at", b"created_at", "created_by", b"created_by", "deployment_id", b"deployment_id", "environment", b"environment", "status", b"status"]) -> None: ...
+
+global___DeploymentMetadata = DeploymentMetadata
+
+@typing.final
+class ModelDeploymentInformation(google.protobuf.message.Message):
+    """Detailed information about a model deployment, including the requested deployment."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_ARTIFACT_IDENTIFIER_FIELD_NUMBER: builtins.int
+    MODEL_DEPLOYMENT_SPEC_FIELD_NUMBER: builtins.int
+    MODEL_DEPLOYMENT_METADATA_FIELD_NUMBER: builtins.int
+    @property
+    def model_artifact_identifier(self) -> frogml._proto.com.jfrog.ml.model.deployment.v1.model_artifact_identifier_pb2.ModelArtifactIdentifier:
+        """The model artifact identifier of the deployment."""
+
+    @property
+    def model_deployment_spec(self) -> global___Deployment:
+        """The currently requested deployment."""
+
+    @property
+    def model_deployment_metadata(self) -> global___DeploymentMetadata:
+        """Metadata about the deployment."""
+
+    def __init__(
+        self,
+        *,
+        model_artifact_identifier: frogml._proto.com.jfrog.ml.model.deployment.v1.model_artifact_identifier_pb2.ModelArtifactIdentifier | None = ...,
+        model_deployment_spec: global___Deployment | None = ...,
+        model_deployment_metadata: global___DeploymentMetadata | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["model_artifact_identifier", b"model_artifact_identifier", "model_deployment_metadata", b"model_deployment_metadata", "model_deployment_spec", b"model_deployment_spec"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["model_artifact_identifier", b"model_artifact_identifier", "model_deployment_metadata", b"model_deployment_metadata", "model_deployment_spec", b"model_deployment_spec"]) -> None: ...
+
+global___ModelDeploymentInformation = ModelDeploymentInformation

frogml/_proto/com/jfrog/ml/model/deployment/v1/deployment_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+