frogml 1.2.35__py3-none-any.whl → 1.2.41__py3-none-any.whl

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

@@ -0,0 +1,99 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import typing
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class ModelArtifactIdentifier(google.protobuf.message.Message):
+    """Identifier for the Model Group, Model and image to deploy."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_BASED_ARTIFACT_ID_FIELD_NUMBER: builtins.int
+    CUSTOM_MODEL_ARTIFACT_ID_FIELD_NUMBER: builtins.int
+    @property
+    def model_based_artifact_id(self) -> global___ModelBasedArtifactIdentifier:
+        """A Model based identifier, which contains the Model Group ID, Model ID and image path."""
+
+    @property
+    def custom_model_artifact_id(self) -> global___CustomModelArtifactIdentifier:
+        """A Custom Model based identifier, which contains the Model Group ID, Model ID and Build ID."""
+
+    def __init__(
+        self,
+        *,
+        model_based_artifact_id: global___ModelBasedArtifactIdentifier | None = ...,
+        custom_model_artifact_id: global___CustomModelArtifactIdentifier | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["custom_model_artifact_id", b"custom_model_artifact_id", "identifier_type", b"identifier_type", "model_based_artifact_id", b"model_based_artifact_id"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["custom_model_artifact_id", b"custom_model_artifact_id", "identifier_type", b"identifier_type", "model_based_artifact_id", b"model_based_artifact_id"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["identifier_type", b"identifier_type"]) -> typing.Literal["model_based_artifact_id", "custom_model_artifact_id"] | None: ...
+
+global___ModelArtifactIdentifier = ModelArtifactIdentifier
+
+@typing.final
+class ModelBasedArtifactIdentifier(google.protobuf.message.Message):
+    """An identifier based on Large Language Models"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_GROUP_ID_FIELD_NUMBER: builtins.int
+    MODEL_ID_FIELD_NUMBER: builtins.int
+    IMAGE_PATH_FIELD_NUMBER: builtins.int
+    JFROG_PROJECT_KEY_FIELD_NUMBER: builtins.int
+    model_group_id: builtins.str
+    """The model group ID, used for permission checking and for organisation e.g. "my-model-group" """
+    model_id: builtins.str
+    """The model ID, which will be used as a routing key for the model e.g. "credit_risk" """
+    image_path: builtins.str
+    """The image path to deploy, without the registry, model group and model prefixes."""
+    jfrog_project_key: builtins.str
+    """The JFrog Project Key where the model image is stored e.g. "my-jfrog-project" """
+    def __init__(
+        self,
+        *,
+        model_group_id: builtins.str = ...,
+        model_id: builtins.str = ...,
+        image_path: builtins.str = ...,
+        jfrog_project_key: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["image_path", b"image_path", "jfrog_project_key", b"jfrog_project_key", "model_group_id", b"model_group_id", "model_id", b"model_id"]) -> None: ...
+
+global___ModelBasedArtifactIdentifier = ModelBasedArtifactIdentifier
+
+@typing.final
+class CustomModelArtifactIdentifier(google.protobuf.message.Message):
+    """An identifier based on Custom Models"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_GROUP_ID_FIELD_NUMBER: builtins.int
+    MODEL_ID_FIELD_NUMBER: builtins.int
+    BUILD_ID_FIELD_NUMBER: builtins.int
+    JFROG_PROJECT_KEY_FIELD_NUMBER: builtins.int
+    model_group_id: builtins.str
+    """The model group ID, used for permission checking and for organisation e.g. "my-model-group" """
+    model_id: builtins.str
+    """The model ID, which will be used as a routing key for the model e.g. "credit_risk" """
+    build_id: builtins.str
+    """The build ID to deploy."""
+    jfrog_project_key: builtins.str
+    """The JFrog Project Key where the model image is stored e.g. "my-jfrog-project" """
+    def __init__(
+        self,
+        *,
+        model_group_id: builtins.str = ...,
+        model_id: builtins.str = ...,
+        build_id: builtins.str = ...,
+        jfrog_project_key: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["build_id", b"build_id", "jfrog_project_key", b"jfrog_project_key", "model_group_id", b"model_group_id", "model_id", b"model_id"]) -> None: ...
+
+global___CustomModelArtifactIdentifier = CustomModelArtifactIdentifier

frogml/_proto/com/jfrog/ml/model/deployment/v1/model_artifact_identifier_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/model_deployment_brief_pb2.py

@@ -0,0 +1,30 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: com/jfrog/ml/model/deployment/v1/model_deployment_brief.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 deployment_pb2 as com_dot_jfrog_dot_ml_dot_model_dot_deployment_dot_v1_dot_deployment__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
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n=com/jfrog/ml/model/deployment/v1/model_deployment_brief.proto\x12 com.jfrog.ml.model.deployment.v1\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x31\x63om/jfrog/ml/model/deployment/v1/deployment.proto\x1a@com/jfrog/ml/model/deployment/v1/model_artifact_identifier.proto\"\x99\x02\n\x14ModelDeploymentBrief\x12G\n\x06status\x18\x01 \x01(\x0b\x32\x37.com.jfrog.ml.model.deployment.v1.ModelDeploymentStatus\x12\x13\n\x0b\x65nvironment\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x15\n\rdeployment_id\x18\x04 \x01(\t\x12\\\n\x19model_artifact_identifier\x18\x05 \x01(\x0b\x32\x39.com.jfrog.ml.model.deployment.v1.ModelArtifactIdentifierB(\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.model_deployment_brief_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['_MODELDEPLOYMENTBRIEF']._serialized_start=250
+  _globals['_MODELDEPLOYMENTBRIEF']._serialized_end=531
+# @@protoc_insertion_point(module_scope)

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

@@ -0,0 +1,55 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import frogml._proto.com.jfrog.ml.model.deployment.v1.deployment_pb2
+import frogml._proto.com.jfrog.ml.model.deployment.v1.model_artifact_identifier_pb2
+import google.protobuf.descriptor
+import google.protobuf.message
+import google.protobuf.timestamp_pb2
+import typing
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class ModelDeploymentBrief(google.protobuf.message.Message):
+    """Represents a **brief** overview of a model deployment containing only essential information at a glance."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    STATUS_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_FIELD_NUMBER: builtins.int
+    CREATED_AT_FIELD_NUMBER: builtins.int
+    DEPLOYMENT_ID_FIELD_NUMBER: builtins.int
+    MODEL_ARTIFACT_IDENTIFIER_FIELD_NUMBER: builtins.int
+    environment: builtins.str
+    """The environment where the deployment is located."""
+    deployment_id: builtins.str
+    """The unique UUID of the deployment."""
+    @property
+    def status(self) -> frogml._proto.com.jfrog.ml.model.deployment.v1.deployment_pb2.ModelDeploymentStatus:
+        """The status of the deployment."""
+
+    @property
+    def created_at(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """The timestamp of the deployment."""
+
+    @property
+    def model_artifact_identifier(self) -> frogml._proto.com.jfrog.ml.model.deployment.v1.model_artifact_identifier_pb2.ModelArtifactIdentifier:
+        """The model artifact identifier associated with this deployment, containing the model group ID, model ID, and either image path or build ID."""
+
+    def __init__(
+        self,
+        *,
+        status: frogml._proto.com.jfrog.ml.model.deployment.v1.deployment_pb2.ModelDeploymentStatus | None = ...,
+        environment: builtins.str = ...,
+        created_at: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        deployment_id: builtins.str = ...,
+        model_artifact_identifier: frogml._proto.com.jfrog.ml.model.deployment.v1.model_artifact_identifier_pb2.ModelArtifactIdentifier | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["created_at", b"created_at", "model_artifact_identifier", b"model_artifact_identifier", "status", b"status"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["created_at", b"created_at", "deployment_id", b"deployment_id", "environment", b"environment", "model_artifact_identifier", b"model_artifact_identifier", "status", b"status"]) -> None: ...
+
+global___ModelDeploymentBrief = ModelDeploymentBrief

frogml/_proto/com/jfrog/ml/model/deployment/v1/model_deployment_brief_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/model_deployment_filter_pb2.py

@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: com/jfrog/ml/model/deployment/v1/model_deployment_filter.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 frogml._proto.com.jfrog.ml.model.deployment.v1 import deployment_pb2 as com_dot_jfrog_dot_ml_dot_model_dot_deployment_dot_v1_dot_deployment__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n>com/jfrog/ml/model/deployment/v1/model_deployment_filter.proto\x12 com.jfrog.ml.model.deployment.v1\x1a\x31\x63om/jfrog/ml/model/deployment/v1/deployment.proto\"\x9b\x01\n\x15ModelDeploymentFilter\x12g\n\x1esimple_model_deployment_filter\x18\x01 \x01(\x0b\x32=.com.jfrog.ml.model.deployment.v1.SimpleModelDeploymentFilterH\x00\x42\x19\n\x17model_deployment_filter\"\xb8\x04\n\x1bSimpleModelDeploymentFilter\x12Z\n\x17model_identifier_filter\x18\x01 \x01(\x0b\x32\x37.com.jfrog.ml.model.deployment.v1.ModelIdentifierFilterH\x00\x12J\n\x0fmodel_group_ids\x18\x02 \x01(\x0b\x32/.com.jfrog.ml.model.deployment.v1.ModelGroupIdsH\x01\x12?\n\tmodel_ids\x18\x03 \x01(\x0b\x32*.com.jfrog.ml.model.deployment.v1.ModelIdsH\x02\x12^\n\x19model_deployment_statuses\x18\x04 \x01(\x0b\x32\x39.com.jfrog.ml.model.deployment.v1.ModelDeploymentStatusesH\x03\x12K\n\x0f\x65nvironment_ids\x18\x05 \x01(\x0b\x32\x30.com.jfrog.ml.model.deployment.v1.EnvironmentIdsH\x04\x42\x1d\n\x1bmodel_identifier_filter_optB\x1b\n\x19model_group_id_filter_optB\x15\n\x13model_id_filter_optB\x13\n\x11status_filter_optB\x1b\n\x19\x65nvironment_id_filter_opt\"w\n\x1bModelDeploymentStatusFilter\x12X\n\x17model_deployment_status\x18\x01 \x03(\x0b\x32\x37.com.jfrog.ml.model.deployment.v1.ModelDeploymentStatus\"A\n\x15ModelIdentifierFilter\x12\x16\n\x0emodel_group_id\x18\x01 \x01(\t\x12\x10\n\x08model_id\x18\x02 \x01(\t\"\'\n\rModelGroupIds\x12\x16\n\x0emodel_group_id\x18\x01 \x03(\t\"\x1c\n\x08ModelIds\x12\x10\n\x08model_id\x18\x01 \x03(\t\"s\n\x17ModelDeploymentStatuses\x12X\n\x17model_deployment_status\x18\x01 \x03(\x0b\x32\x37.com.jfrog.ml.model.deployment.v1.ModelDeploymentStatus\"(\n\x0e\x45nvironmentIds\x12\x16\n\x0e\x65nvironment_id\x18\x01 \x03(\tB(\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.model_deployment_filter_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['_MODELDEPLOYMENTFILTER']._serialized_start=152
+  _globals['_MODELDEPLOYMENTFILTER']._serialized_end=307
+  _globals['_SIMPLEMODELDEPLOYMENTFILTER']._serialized_start=310
+  _globals['_SIMPLEMODELDEPLOYMENTFILTER']._serialized_end=878
+  _globals['_MODELDEPLOYMENTSTATUSFILTER']._serialized_start=880
+  _globals['_MODELDEPLOYMENTSTATUSFILTER']._serialized_end=999
+  _globals['_MODELIDENTIFIERFILTER']._serialized_start=1001
+  _globals['_MODELIDENTIFIERFILTER']._serialized_end=1066
+  _globals['_MODELGROUPIDS']._serialized_start=1068
+  _globals['_MODELGROUPIDS']._serialized_end=1107
+  _globals['_MODELIDS']._serialized_start=1109
+  _globals['_MODELIDS']._serialized_end=1137
+  _globals['_MODELDEPLOYMENTSTATUSES']._serialized_start=1139
+  _globals['_MODELDEPLOYMENTSTATUSES']._serialized_end=1254
+  _globals['_ENVIRONMENTIDS']._serialized_start=1256
+  _globals['_ENVIRONMENTIDS']._serialized_end=1296
+# @@protoc_insertion_point(module_scope)

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

@@ -0,0 +1,204 @@
+"""
+@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.deployment_pb2
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.message
+import typing
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class ModelDeploymentFilter(google.protobuf.message.Message):
+    """A filter to apply on Model Deployment lists. Currently only a simple filter is supported that doesnt allow for
+    complex filter expressions. Such a filter may be added here in the future.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SIMPLE_MODEL_DEPLOYMENT_FILTER_FIELD_NUMBER: builtins.int
+    @property
+    def simple_model_deployment_filter(self) -> global___SimpleModelDeploymentFilter:
+        """A simple filter to apply on Model Deployment lists."""
+
+    def __init__(
+        self,
+        *,
+        simple_model_deployment_filter: global___SimpleModelDeploymentFilter | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["model_deployment_filter", b"model_deployment_filter", "simple_model_deployment_filter", b"simple_model_deployment_filter"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["model_deployment_filter", b"model_deployment_filter", "simple_model_deployment_filter", b"simple_model_deployment_filter"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["model_deployment_filter", b"model_deployment_filter"]) -> typing.Literal["simple_model_deployment_filter"] | None: ...
+
+global___ModelDeploymentFilter = ModelDeploymentFilter
+
+@typing.final
+class SimpleModelDeploymentFilter(google.protobuf.message.Message):
+    """A filter to apply on Model Deployment lists.
+    The relationship between each filter element is an "AND". i.e. all filters must match for a given Deployment to be
+    included by the filter.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_IDENTIFIER_FILTER_FIELD_NUMBER: builtins.int
+    MODEL_GROUP_IDS_FIELD_NUMBER: builtins.int
+    MODEL_IDS_FIELD_NUMBER: builtins.int
+    MODEL_DEPLOYMENT_STATUSES_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_IDS_FIELD_NUMBER: builtins.int
+    @property
+    def model_identifier_filter(self) -> global___ModelIdentifierFilter: ...
+    @property
+    def model_group_ids(self) -> global___ModelGroupIds: ...
+    @property
+    def model_ids(self) -> global___ModelIds: ...
+    @property
+    def model_deployment_statuses(self) -> global___ModelDeploymentStatuses: ...
+    @property
+    def environment_ids(self) -> global___EnvironmentIds: ...
+    def __init__(
+        self,
+        *,
+        model_identifier_filter: global___ModelIdentifierFilter | None = ...,
+        model_group_ids: global___ModelGroupIds | None = ...,
+        model_ids: global___ModelIds | None = ...,
+        model_deployment_statuses: global___ModelDeploymentStatuses | None = ...,
+        environment_ids: global___EnvironmentIds | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["environment_id_filter_opt", b"environment_id_filter_opt", "environment_ids", b"environment_ids", "model_deployment_statuses", b"model_deployment_statuses", "model_group_id_filter_opt", b"model_group_id_filter_opt", "model_group_ids", b"model_group_ids", "model_id_filter_opt", b"model_id_filter_opt", "model_identifier_filter", b"model_identifier_filter", "model_identifier_filter_opt", b"model_identifier_filter_opt", "model_ids", b"model_ids", "status_filter_opt", b"status_filter_opt"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["environment_id_filter_opt", b"environment_id_filter_opt", "environment_ids", b"environment_ids", "model_deployment_statuses", b"model_deployment_statuses", "model_group_id_filter_opt", b"model_group_id_filter_opt", "model_group_ids", b"model_group_ids", "model_id_filter_opt", b"model_id_filter_opt", "model_identifier_filter", b"model_identifier_filter", "model_identifier_filter_opt", b"model_identifier_filter_opt", "model_ids", b"model_ids", "status_filter_opt", b"status_filter_opt"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["environment_id_filter_opt", b"environment_id_filter_opt"]) -> typing.Literal["environment_ids"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["model_group_id_filter_opt", b"model_group_id_filter_opt"]) -> typing.Literal["model_group_ids"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["model_id_filter_opt", b"model_id_filter_opt"]) -> typing.Literal["model_ids"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["model_identifier_filter_opt", b"model_identifier_filter_opt"]) -> typing.Literal["model_identifier_filter"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["status_filter_opt", b"status_filter_opt"]) -> typing.Literal["model_deployment_statuses"] | None: ...
+
+global___SimpleModelDeploymentFilter = SimpleModelDeploymentFilter
+
+@typing.final
+class ModelDeploymentStatusFilter(google.protobuf.message.Message):
+    """A filter that can match a specific Model Deployment status.
+    Any deployment whose status matches one of the given statuses will be included (logical OR between statuses).
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_DEPLOYMENT_STATUS_FIELD_NUMBER: builtins.int
+    @property
+    def model_deployment_status(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[com.jfrog.ml.model.deployment.v1.deployment_pb2.ModelDeploymentStatus]:
+        """The Model Deployment statuses to include in the filter.
+        Only Deployments with any of the given statuses will be included.
+        """
+
+    def __init__(
+        self,
+        *,
+        model_deployment_status: collections.abc.Iterable[com.jfrog.ml.model.deployment.v1.deployment_pb2.ModelDeploymentStatus] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["model_deployment_status", b"model_deployment_status"]) -> None: ...
+
+global___ModelDeploymentStatusFilter = ModelDeploymentStatusFilter
+
+@typing.final
+class ModelIdentifierFilter(google.protobuf.message.Message):
+    """A filter that can match a specific Model group and Model ID."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_GROUP_ID_FIELD_NUMBER: builtins.int
+    MODEL_ID_FIELD_NUMBER: builtins.int
+    model_group_id: builtins.str
+    """The Model group ID to include in the filter e.g. "my-Model-group" """
+    model_id: builtins.str
+    """The Model ID to include in the filter by e.g. "credit_risk" """
+    def __init__(
+        self,
+        *,
+        model_group_id: builtins.str = ...,
+        model_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["model_group_id", b"model_group_id", "model_id", b"model_id"]) -> None: ...
+
+global___ModelIdentifierFilter = ModelIdentifierFilter
+
+@typing.final
+class ModelGroupIds(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_GROUP_ID_FIELD_NUMBER: builtins.int
+    @property
+    def model_group_id(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """List of model group IDs to filter by."""
+
+    def __init__(
+        self,
+        *,
+        model_group_id: collections.abc.Iterable[builtins.str] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["model_group_id", b"model_group_id"]) -> None: ...
+
+global___ModelGroupIds = ModelGroupIds
+
+@typing.final
+class ModelIds(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_ID_FIELD_NUMBER: builtins.int
+    @property
+    def model_id(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """List of model IDs to filter by."""
+
+    def __init__(
+        self,
+        *,
+        model_id: collections.abc.Iterable[builtins.str] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["model_id", b"model_id"]) -> None: ...
+
+global___ModelIds = ModelIds
+
+@typing.final
+class ModelDeploymentStatuses(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_DEPLOYMENT_STATUS_FIELD_NUMBER: builtins.int
+    @property
+    def model_deployment_status(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[com.jfrog.ml.model.deployment.v1.deployment_pb2.ModelDeploymentStatus]:
+        """List of deployment statuses to filter by."""
+
+    def __init__(
+        self,
+        *,
+        model_deployment_status: collections.abc.Iterable[com.jfrog.ml.model.deployment.v1.deployment_pb2.ModelDeploymentStatus] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["model_deployment_status", b"model_deployment_status"]) -> None: ...
+
+global___ModelDeploymentStatuses = ModelDeploymentStatuses
+
+@typing.final
+class EnvironmentIds(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
+    @property
+    def environment_id(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """List of environment IDs to filter by."""
+
+    def __init__(
+        self,
+        *,
+        environment_id: collections.abc.Iterable[builtins.str] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["environment_id", b"environment_id"]) -> None: ...
+
+global___EnvironmentIds = EnvironmentIds

frogml/_proto/com/jfrog/ml/model/deployment/v1/model_deployment_filter_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/probes_configuration_pb2.py

@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: com/jfrog/ml/model/deployment/v1/probes_configuration.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'\n;com/jfrog/ml/model/deployment/v1/probes_configuration.proto\x12 com.jfrog.ml.model.deployment.v1\"\xdf\x01\n\x13ProbesConfiguration\x12M\n\rstartup_probe\x18\x01 \x01(\x0b\x32\x34.com.jfrog.ml.model.deployment.v1.ProbeConfigurationH\x00\x12N\n\x0eliveness_probe\x18\x02 \x01(\x0b\x32\x34.com.jfrog.ml.model.deployment.v1.ProbeConfigurationH\x01\x42\x13\n\x11startup_probe_optB\x14\n\x12liveness_probe_opt\"A\n\x12ProbeConfiguration\x12\x17\n\x0ftimeout_seconds\x18\x01 \x01(\x05\x12\x12\n\nprobe_path\x18\x02 \x01(\tB(\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.probes_configuration_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['_PROBESCONFIGURATION']._serialized_start=98
+  _globals['_PROBESCONFIGURATION']._serialized_end=321
+  _globals['_PROBECONFIGURATION']._serialized_start=323
+  _globals['_PROBECONFIGURATION']._serialized_end=388
+# @@protoc_insertion_point(module_scope)

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

@@ -0,0 +1,60 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import typing
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class ProbesConfiguration(google.protobuf.message.Message):
+    """Configuration of probes for monitoring and (for attempting) to restore health of containers"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    STARTUP_PROBE_FIELD_NUMBER: builtins.int
+    LIVENESS_PROBE_FIELD_NUMBER: builtins.int
+    @property
+    def startup_probe(self) -> global___ProbeConfiguration: ...
+    @property
+    def liveness_probe(self) -> global___ProbeConfiguration: ...
+    def __init__(
+        self,
+        *,
+        startup_probe: global___ProbeConfiguration | None = ...,
+        liveness_probe: global___ProbeConfiguration | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["liveness_probe", b"liveness_probe", "liveness_probe_opt", b"liveness_probe_opt", "startup_probe", b"startup_probe", "startup_probe_opt", b"startup_probe_opt"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["liveness_probe", b"liveness_probe", "liveness_probe_opt", b"liveness_probe_opt", "startup_probe", b"startup_probe", "startup_probe_opt", b"startup_probe_opt"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["liveness_probe_opt", b"liveness_probe_opt"]) -> typing.Literal["liveness_probe"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["startup_probe_opt", b"startup_probe_opt"]) -> typing.Literal["startup_probe"] | None: ...
+
+global___ProbesConfiguration = ProbesConfiguration
+
+@typing.final
+class ProbeConfiguration(google.protobuf.message.Message):
+    """Configuration of a specific probe."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TIMEOUT_SECONDS_FIELD_NUMBER: builtins.int
+    PROBE_PATH_FIELD_NUMBER: builtins.int
+    timeout_seconds: builtins.int
+    """Timeout in seconds after which a failing probe should trigger a failure."""
+    probe_path: builtins.str
+    """The probe path, such as /health, to be used as an http endpoint in the container."""
+    def __init__(
+        self,
+        *,
+        timeout_seconds: builtins.int = ...,
+        probe_path: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["probe_path", b"probe_path", "timeout_seconds", b"timeout_seconds"]) -> None: ...
+
+global___ProbeConfiguration = ProbeConfiguration

frogml/_proto/com/jfrog/ml/model/deployment/v1/probes_configuration_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/realtime_deployment_pb2.py

@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: com/jfrog/ml/model/deployment/v1/realtime_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 frogml._proto.com.jfrog.ml.model.deployment.v1 import auto_scaling_pb2 as com_dot_jfrog_dot_ml_dot_model_dot_deployment_dot_v1_dot_auto__scaling__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n:com/jfrog/ml/model/deployment/v1/realtime_deployment.proto\x12 com.jfrog.ml.model.deployment.v1\x1a\x33\x63om/jfrog/ml/model/deployment/v1/auto_scaling.proto\"\xb6\x01\n\x12RealtimeDeployment\x12\x18\n\x10\x65xposed_api_path\x18\x01 \x01(\t\x12\x17\n\rreplica_count\x18\x02 \x01(\x05H\x00\x12R\n\x13\x61uto_scaling_config\x18\x03 \x01(\x0b\x32\x33.com.jfrog.ml.model.deployment.v1.AutoScalingConfigH\x00\x42\x19\n\x17horizontal_scaling_typeB(\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.realtime_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['_REALTIMEDEPLOYMENT']._serialized_start=150
+  _globals['_REALTIMEDEPLOYMENT']._serialized_end=332
+# @@protoc_insertion_point(module_scope)

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

@@ -0,0 +1,47 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import frogml._proto.com.jfrog.ml.model.deployment.v1.auto_scaling_pb2
+import google.protobuf.descriptor
+import google.protobuf.message
+import typing
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class RealtimeDeployment(google.protobuf.message.Message):
+    """A Realtime Deployment is exposing the model as a real-time service, which can be used to serve predictions using
+    REST requests.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    EXPOSED_API_PATH_FIELD_NUMBER: builtins.int
+    REPLICA_COUNT_FIELD_NUMBER: builtins.int
+    AUTO_SCALING_CONFIG_FIELD_NUMBER: builtins.int
+    exposed_api_path: builtins.str
+    """A single specific API endpoint e.g. `/predict` for Custom Models, and `/v1/chat/completions` for vLLM.
+    This API endpoint is the only one that will be exposed.
+    This May be deprecated when AI Gateway is developed.
+    """
+    replica_count: builtins.int
+    """Fixed number of replicas"""
+    @property
+    def auto_scaling_config(self) -> frogml._proto.com.jfrog.ml.model.deployment.v1.auto_scaling_pb2.AutoScalingConfig:
+        """Automatic horizontal scaling based on the auto scaling configuration."""
+
+    def __init__(
+        self,
+        *,
+        exposed_api_path: builtins.str = ...,
+        replica_count: builtins.int = ...,
+        auto_scaling_config: frogml._proto.com.jfrog.ml.model.deployment.v1.auto_scaling_pb2.AutoScalingConfig | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["auto_scaling_config", b"auto_scaling_config", "horizontal_scaling_type", b"horizontal_scaling_type", "replica_count", b"replica_count"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["auto_scaling_config", b"auto_scaling_config", "exposed_api_path", b"exposed_api_path", "horizontal_scaling_type", b"horizontal_scaling_type", "replica_count", b"replica_count"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["horizontal_scaling_type", b"horizontal_scaling_type"]) -> typing.Literal["replica_count", "auto_scaling_config"] | None: ...
+
+global___RealtimeDeployment = RealtimeDeployment

frogml/_proto/com/jfrog/ml/model/deployment/v1/realtime_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
+