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

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

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: com/jfrog/ml/model/deployment/v1/resource_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/resource_configuration.proto\x12 com.jfrog.ml.model.deployment.v1\"U\n\x15ResourceConfiguration\x12 \n\x16template_specification\x18\x01 \x01(\tH\x00\x42\x1a\n\x18resource_descriptor_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.resource_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['_RESOURCECONFIGURATION']._serialized_start=99
+  _globals['_RESOURCECONFIGURATION']._serialized_end=184
+# @@protoc_insertion_point(module_scope)

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

@@ -0,0 +1,29 @@
+"""
+@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 ResourceConfiguration(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TEMPLATE_SPECIFICATION_FIELD_NUMBER: builtins.int
+    template_specification: builtins.str
+    """A legacy field for describing a compute resources allocation using templated resource sizes."""
+    def __init__(
+        self,
+        *,
+        template_specification: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["resource_descriptor_type", b"resource_descriptor_type", "template_specification", b"template_specification"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["resource_descriptor_type", b"resource_descriptor_type", "template_specification", b"template_specification"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["resource_descriptor_type", b"resource_descriptor_type"]) -> typing.Literal["template_specification"] | None: ...
+
+global___ResourceConfiguration = ResourceConfiguration

frogml/_proto/com/jfrog/ml/model/deployment/v1/resource_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/core/clients/model_deployment_manager/__init__.py

@@ -0,0 +1 @@
+from .client import ModelDeploymentManagerClient

frogml/core/clients/model_deployment_manager/client.py

@@ -0,0 +1,103 @@
+from dependency_injector.wiring import Provide
+from frogml._proto.com.jfrog.ml.model.deployment.v1.deployment_pb2 import (
+    Deployment,
+    ModelDeploymentInformation,
+    MultipleEnvironmentDeployment,
+)
+from frogml._proto.com.jfrog.ml.model.deployment.v1.deployment_service_pb2 import (
+    DeploymentIds,
+    DeploymentIdToDeploymentSpecMap,
+    DeployModelRequest,
+    EditModelDeploymentRequest,
+    GetModelDeploymentResponse,
+    ListModelDeploymentsRequest,
+    ListModelDeploymentsResponse,
+    UndeployModelRequest,
+)
+from frogml._proto.com.jfrog.ml.model.deployment.v1.deployment_service_pb2_grpc import (
+    ModelDeploymentServiceStub,
+)
+from frogml._proto.com.jfrog.ml.model.deployment.v1.model_artifact_identifier_pb2 import (
+    ModelArtifactIdentifier,
+    ModelBasedArtifactIdentifier,
+)
+from frogml._proto.com.jfrog.ml.model.deployment.v1.model_deployment_brief_pb2 import (
+    ModelDeploymentBrief,
+)
+from frogml._proto.com.jfrog.ml.model.deployment.v1.model_deployment_filter_pb2 import (
+    ModelDeploymentFilter,
+)
+from frogml._proto.qwak.deployment.deployment_service_pb2 import GetDeploymentRequest
+from frogml.core.inner.di_configuration import FrogmlContainer
+from frogml.core.inner.tool.grpc.grpc_try_wrapping import grpc_try_catch_wrapper
+
+
+class ModelDeploymentManagerClient:
+    def __init__(self, grpc_channel=Provide[FrogmlContainer.core_grpc_channel]):
+        self.__client = ModelDeploymentServiceStub(grpc_channel)
+
+    @grpc_try_catch_wrapper("Failed to deploy LLM for model {model_id}")
+    def deploy_llm(
+        self,
+        model_id: str,
+        model_group_id: str,
+        image_path: str,
+        environment_deployment_configuration: dict[str, Deployment],
+    ) -> None:
+        llm_model_identifier = ModelBasedArtifactIdentifier(
+            model_group_id=model_group_id,
+            model_id=model_id,
+            image_path=image_path,
+        )
+        model_artifact_identifier = ModelArtifactIdentifier(
+            model_based_artifact_id=llm_model_identifier
+        )
+        multiple_environment_deployment = MultipleEnvironmentDeployment(
+            model_artifact_identifier=model_artifact_identifier,
+            deployments=environment_deployment_configuration,
+        )
+        request = DeployModelRequest(
+            multiple_environment_deployment=multiple_environment_deployment
+        )
+
+        self.__client.DeployModel(request)
+
+    @grpc_try_catch_wrapper("Failed to undeploy")
+    def undeploy(self, deployment_ids: list[str]) -> None:
+        deployment_ids_proto = DeploymentIds(deployment_id=deployment_ids)
+        request = UndeployModelRequest(deployment_ids=deployment_ids_proto)
+        self.__client.UndeployModel(request)
+
+    @grpc_try_catch_wrapper("Failed to edit deployments")
+    def edit_deployments(
+        self, deployment_id_to_spec_map: dict[str, Deployment]
+    ) -> None:
+        deployments_map = DeploymentIdToDeploymentSpecMap(
+            deployments=deployment_id_to_spec_map
+        )
+        request = EditModelDeploymentRequest(
+            deployment_id_to_deployment_spec_map=deployments_map
+        )
+        self.__client.EditModelDeployment(request)
+
+    @grpc_try_catch_wrapper("Failed to list deployments")
+    def list_deployments(
+        self,
+        model_deployment_filter: ModelDeploymentFilter = None,
+    ) -> list[ModelDeploymentBrief]:
+        if model_deployment_filter is None:
+            model_deployment_filter = ModelDeploymentFilter()
+
+        request = ListModelDeploymentsRequest(
+            model_deployment_filter=model_deployment_filter
+        )
+        response: ListModelDeploymentsResponse = self.__client.ListModelDeployments(
+            request
+        )
+        return list(response.model_deployment_brief)
+
+    @grpc_try_catch_wrapper("Failed to get deployment with ID {deployment_id}")
+    def get_deployment(self, deployment_id: str) -> ModelDeploymentInformation:
+        request = GetDeploymentRequest(deployment_id=deployment_id)
+        response: GetModelDeploymentResponse = self.__client.GetModelDeployment(request)
+        return response.model_deployment_information

{frogml-1.2.40.dist-info → frogml-1.2.42.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml
-Version: 1.2.40
+Version: 1.2.42
 Summary: frogml contains the necessary objects and communication tools for using the JFrog ml Platform
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model

{frogml-1.2.40.dist-info → frogml-1.2.42.dist-info}/RECORD

@@ -1,4 +1,34 @@
-frogml/__init__.py,sha256=PjA51CtynH2yUX2k_lzDfUDiPya6uCO3uTncfZ3d6cM,741
+frogml/__init__.py,sha256=sJOd_xEn0Cz_WXUSQWRDjwsaTMl7KajIC5Sop94NZe8,741
+frogml/_proto/com/jfrog/ml/model/deployment/v1/auto_scaling_pb2.py,sha256=-SD6RYgn5DFBLlwTpH9YkV8aW_zYymBI68Y-12Fn0I0,4193
+frogml/_proto/com/jfrog/ml/model/deployment/v1/auto_scaling_pb2.pyi,sha256=s0rdjIYw_dOfmuentpri79ls5dC3NqoUq__-XeavaXE,9688
+frogml/_proto/com/jfrog/ml/model/deployment/v1/auto_scaling_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
+frogml/_proto/com/jfrog/ml/model/deployment/v1/deployment_pb2.py,sha256=c1cvdGhas7PyjFa_qWgp8cQBT9FJJatDXpLFaAVqbIo,6441
+frogml/_proto/com/jfrog/ml/model/deployment/v1/deployment_pb2.pyi,sha256=NMwN-k3r1Y3Z_2yrUvfTQvO0rywEMIsyXTpkroNy6X8,12950
+frogml/_proto/com/jfrog/ml/model/deployment/v1/deployment_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
+frogml/_proto/com/jfrog/ml/model/deployment/v1/deployment_service_pb2.py,sha256=MtpCW3ZPBYrpIhMC0XMJRCGDVpI-Ak_jeKQ9DGqYaUQ,6286
+frogml/_proto/com/jfrog/ml/model/deployment/v1/deployment_service_pb2.pyi,sha256=srnR6yTrjo8tnqAmnhxXEbf5EKD-dxMnfXZ-rABtmaY,9569
+frogml/_proto/com/jfrog/ml/model/deployment/v1/deployment_service_pb2_grpc.py,sha256=qvF_o89cqJ9Jz_nZs3n72p_pY2Bo75Sq99ObFsx4Nsw,11599
+frogml/_proto/com/jfrog/ml/model/deployment/v1/environment_variables_configuration_pb2.py,sha256=RPbECbR8DOWfq37aSYm4oWJhlVd-IWKGYEIhjmJ5VlU,2422
+frogml/_proto/com/jfrog/ml/model/deployment/v1/environment_variables_configuration_pb2.pyi,sha256=z3FDspmH0Bfd15x_KnzPDRvBdVbxCPxIz7ckC5wqJPM,3860
+frogml/_proto/com/jfrog/ml/model/deployment/v1/environment_variables_configuration_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
+frogml/_proto/com/jfrog/ml/model/deployment/v1/model_artifact_identifier_pb2.py,sha256=7TCaIZwY3cYX5JKUA1NR8WmkRH0CtLipi4kiaTjeL9E,2331
+frogml/_proto/com/jfrog/ml/model/deployment/v1/model_artifact_identifier_pb2.pyi,sha256=4FFrif0OZFR18xXKuEpZC7sibeKNMMXM49Y-J7VXyGA,4529
+frogml/_proto/com/jfrog/ml/model/deployment/v1/model_artifact_identifier_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
+frogml/_proto/com/jfrog/ml/model/deployment/v1/model_deployment_brief_pb2.py,sha256=r1nMzNPG0MktLtRPTFFif66qEHL0VGHuAoiM2ffXVlk,2306
+frogml/_proto/com/jfrog/ml/model/deployment/v1/model_deployment_brief_pb2.pyi,sha256=H-XA4wfh7t8PLYtfbPINIbWea43ubn8tbFdPN9IjLKs,2628
+frogml/_proto/com/jfrog/ml/model/deployment/v1/model_deployment_brief_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
+frogml/_proto/com/jfrog/ml/model/deployment/v1/model_deployment_filter_pb2.py,sha256=6qhYmErXArD2rOoKeqnlvjwC-6dJo8k0HUcTTMW4oQQ,3889
+frogml/_proto/com/jfrog/ml/model/deployment/v1/model_deployment_filter_pb2.pyi,sha256=gdxw9GSXh-Msfe0ifVSmER-7kHtcU3u0RNCjb93YY-Y,9804
+frogml/_proto/com/jfrog/ml/model/deployment/v1/model_deployment_filter_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
+frogml/_proto/com/jfrog/ml/model/deployment/v1/probes_configuration_pb2.py,sha256=xn_xW0y6JJ0TjyLaJBDVE7UwHO22tQr3nphz7xLGaSM,1833
+frogml/_proto/com/jfrog/ml/model/deployment/v1/probes_configuration_pb2.pyi,sha256=t_EkUDqnMHgXr0ft9IXyfYbIR4vXR3yGzVp-cG0P0G4,2578
+frogml/_proto/com/jfrog/ml/model/deployment/v1/probes_configuration_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
+frogml/_proto/com/jfrog/ml/model/deployment/v1/realtime_deployment_pb2.py,sha256=PhvTR0wld0M7fQFbmQ2vWplyHIDThaM_ZmVHNqcQBCc,1799
+frogml/_proto/com/jfrog/ml/model/deployment/v1/realtime_deployment_pb2.pyi,sha256=tGb_ZZJuwMWmfKBiONnhwxREuREmDpV0x4Ya4DQaqos,2234
+frogml/_proto/com/jfrog/ml/model/deployment/v1/realtime_deployment_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
+frogml/_proto/com/jfrog/ml/model/deployment/v1/resource_configuration_pb2.py,sha256=d3oCCHJywlvPD91GiTB_RveXoUhHyyJOkNJ5d5tGl58,1435
+frogml/_proto/com/jfrog/ml/model/deployment/v1/resource_configuration_pb2.pyi,sha256=FRhxd3AhNEICHElc8iv0gSnRH2zqmIw7Yd98tLWaR78,1253
+frogml/_proto/com/jfrog/ml/model/deployment/v1/resource_configuration_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
 frogml/_proto/jfml/hosting_gateway/v1/build_upload_url_pb2.py,sha256=pY7-QarLJWbL9uNmZ69RfE7IPVzwpVIhtd3A0wztSNY,1942
 frogml/_proto/jfml/hosting_gateway/v1/build_upload_url_pb2.pyi,sha256=505I4uUrCmv2hsUw2H5eT-nlkNOCQOWodpvEkH2fHIo,3073
 frogml/_proto/jfml/hosting_gateway/v1/build_upload_url_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
@@ -692,6 +722,8 @@ frogml/core/clients/location_discovery/__init__.py,sha256=sqGQ75YHFE6nvOcir38fyk
 frogml/core/clients/location_discovery/client.py,sha256=ZcBbbLqzZ9E7oLyamBTLTpp611LldhfFOSkzWroyIQg,2895
 frogml/core/clients/logging_client/__init__.py,sha256=1OCHnigQBYThBwGbxCreYA0BgP0HcuLFzNEWd3Yxh-c,34
 frogml/core/clients/logging_client/client.py,sha256=ZTRRPrcM2cqrZX9T3DvIr0YICrhDK91P4CDP-ei_ZjQ,8599
+frogml/core/clients/model_deployment_manager/__init__.py,sha256=Kp6FTZzsXjvfoCLz8kurr7AIypRGIIdUek7_SBCt7f4,49
+frogml/core/clients/model_deployment_manager/client.py,sha256=VLPgE3WDZOOg9y_mYZSKKYqVSi866vJipvAi5aOlDng,4135
 frogml/core/clients/model_group_management/__init__.py,sha256=f-nExP4vGWoWMUrRstY8QdtPqXnbsl2Ux0KfFkvGBkg,47
 frogml/core/clients/model_group_management/client.py,sha256=F8YCbfyphcljb8WxpeY7H7wnPpksyaKQ2ZOKi03RU6M,2016
 frogml/core/clients/model_management/__init__.py,sha256=vjWVP8MjmK4_A70WOgJqa6x24AeLK-ABjGJtogGzw9w,43
@@ -1123,7 +1155,7 @@ frogml_services_mock/mocks/features_operator_v3_service.py,sha256=v6XdUpiI5T5mog
 frogml_services_mock/mocks/features_set_state_service_api.py,sha256=YaF3X12gCFDKFYFUAH3ZsK2jjbxU1aYYyExwe-F1Zdg,2293
 frogml_services_mock/mocks/feedback_service.py,sha256=yHItKZnlrN75F7D2rIGVrffipRaBCyMIvxK5ZmMVckE,1142
 frogml_services_mock/mocks/file_versioning_service.py,sha256=PF_XyipSF5I8SfCNDYv298p3lMCizgDTZORlSuo4RYQ,2609
-frogml_services_mock/mocks/frogml_mocks.py,sha256=KpIMhgwmboLM08vo6SYep9Vj_tnGL2GiBD2UGwGJjt0,6143
+frogml_services_mock/mocks/frogml_mocks.py,sha256=16-x7m3pHl9C51SbuVYwDWmBWv4vzEAwEjfykQaSlfM,6336
 frogml_services_mock/mocks/fs_offline_serving_service.py,sha256=rzxL_G8t3PIM-IFuchqkHuUUzlOZfiP5A5IMSeSQynA,2096
 frogml_services_mock/mocks/instance_template_management_service.py,sha256=AZ-QCzrx4d4g5E4XIPqFNhOJPBwb4S90mfVzsn6oNe4,4766
 frogml_services_mock/mocks/integration_management_service.py,sha256=cw45A8EadycTyq3Okx1sgcuWZ-Lz2iPMRin6f4e_0Ws,2775
@@ -1133,6 +1165,7 @@ frogml_services_mock/mocks/job_registry_service_api.py,sha256=gbytsg5mc5yjOYE_LP
 frogml_services_mock/mocks/kube_captain_service_api.py,sha256=0zRthbKSWeIl1qj16tqTg4jiTbUZoJRyPpEOXiESK2A,1598
 frogml_services_mock/mocks/location_discovery_service_api.py,sha256=i_alD-qEZ1SsIFgQew9gibGN47MrPhTfMqnxapJhTbo,3685
 frogml_services_mock/mocks/logging_service.py,sha256=-1FrXhGO5_NNUW7pzajTwGSf1soQC1feLuMwkOAXD7A,7389
+frogml_services_mock/mocks/model_deployment_manager_service_mock.py,sha256=83BtdT3FLhewmEjvqMLw7hrrKOxp_lLe_hlDIZ72ukI,9743
 frogml_services_mock/mocks/model_group_management_service.py,sha256=tEtBQPExusWI1NiigqEToSDIAforCP1DpVCGXiqfppc,1540
 frogml_services_mock/mocks/model_management_service.py,sha256=3byRc-8Fb5WnXTrWkHj1-S6RA2yfqK1yzX-7fqq_ZNk,4169
 frogml_services_mock/mocks/model_version_manager_service.py,sha256=XzRnAnn0VMY5hW8jP1ngWF-6OYIq8OgatZQ79JyxeMQ,3759
@@ -1144,9 +1177,9 @@ frogml_services_mock/mocks/system_secret_service.py,sha256=ceOR1nfQOGW4ZBcry4Q_l
 frogml_services_mock/mocks/user_application_instance_service_api.py,sha256=Es3s9MnutY0Qsd_FOq_ctKvT-3wHn7-fxah1fvr5Jv4,4100
 frogml_services_mock/mocks/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_services_mock/mocks/utils/exception_handlers.py,sha256=k_8mez3cwjNjKE9yGQRJUuK95qNQyk_slotIF08IIEE,308
-frogml_services_mock/services_mock.py,sha256=y1_h189Ldu5MdVUW4nj-WMBtvUfkM_aKse7UfAb6Rxk,19419
+frogml_services_mock/services_mock.py,sha256=hciz4Q2NvjT84DE08sgxfVVDELfD4hitqgoDrZugcjg,19961
 frogml_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
-frogml-1.2.40.dist-info/METADATA,sha256=gPjSmQKhg_oYhCMNfkKU7i5mTcFMOr6CAiOGCl6wD5w,5599
-frogml-1.2.40.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-frogml-1.2.40.dist-info/RECORD,,
+frogml-1.2.42.dist-info/METADATA,sha256=fTkHE_LzeDdRqF9rzohysGaAPkGhd9tYbv1o5WK2KiM,5599
+frogml-1.2.42.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+frogml-1.2.42.dist-info/RECORD,,

frogml_services_mock/mocks/frogml_mocks.py

@@ -56,7 +56,9 @@ from frogml_services_mock.mocks.integration_management_service import (
 from frogml_services_mock.mocks.internal_build_orchestrator_service import (
     InternalBuildOrchestratorServiceMock,
 )
-from frogml_services_mock.mocks.jfrog_tenant_info_service_mock import JFrogTenantInfoServiceMock
+from frogml_services_mock.mocks.jfrog_tenant_info_service_mock import (
+    JFrogTenantInfoServiceMock,
+)
 from frogml_services_mock.mocks.job_registry_service_api import (
     JobRegistryServiceApiMock,
 )
@@ -67,15 +69,20 @@ from frogml_services_mock.mocks.location_discovery_service_api import (
     LocationDiscoveryServiceApiMock,
 )
 from frogml_services_mock.mocks.logging_service import LoggingServiceApiMock
-from frogml_services_mock.mocks.model_group_management_service import ModelGroupManagementServiceMock
+from frogml_services_mock.mocks.model_deployment_manager_service_mock import (
+    ModelDeploymentManagerMock,
+)
+from frogml_services_mock.mocks.model_group_management_service import (
+    ModelGroupManagementServiceMock,
+)
 from frogml_services_mock.mocks.model_management_service import (
     ModelsManagementServiceMock,
 )
 from frogml_services_mock.mocks.model_version_manager_service import (
     ModelVersionManagerServiceMock,
 )
-from frogml_services_mock.mocks.repository_service_mock import RepositoryServiceMock
 from frogml_services_mock.mocks.project_manager_service import ProjectManagerServiceMock
+from frogml_services_mock.mocks.repository_service_mock import RepositoryServiceMock
 from frogml_services_mock.mocks.secret_service import SecretServiceMock
 from frogml_services_mock.mocks.self_service_user_service import (
     SelfServiceUserServiceMock,
@@ -127,3 +134,4 @@ class FrogmlMocks:
     repository_service: RepositoryServiceMock
     location_discovery_service: LocationDiscoveryServiceApiMock
     jfrog_tenant_info_service: JFrogTenantInfoServiceMock
+    model_deployment_manager_mock: ModelDeploymentManagerMock

frogml_services_mock/mocks/model_deployment_manager_service_mock.py

@@ -0,0 +1,255 @@
+import uuid
+from datetime import datetime, timezone
+from typing import Callable, Optional
+
+import grpc
+from frogml._proto.com.jfrog.ml.model.deployment.v1.deployment_pb2 import (
+    Deployment,
+    DeploymentMetadata,
+    ModelDeploymentInformation,
+    ModelDeploymentStatus,
+    ModelDeploymentStatusSuccessful,
+    MultipleEnvironmentDeployment,
+)
+from frogml._proto.com.jfrog.ml.model.deployment.v1.deployment_service_pb2 import (
+    DeployModelRequest,
+    DeployModelResponse,
+    EditModelDeploymentRequest,
+    EditModelDeploymentResponse,
+    GetModelDeploymentRequest,
+    GetModelDeploymentResponse,
+    ListModelDeploymentsRequest,
+    ListModelDeploymentsResponse,
+    UndeployModelRequest,
+    UndeployModelResponse,
+)
+from frogml._proto.com.jfrog.ml.model.deployment.v1.deployment_service_pb2_grpc import (
+    ModelDeploymentServiceServicer,
+)
+from frogml._proto.com.jfrog.ml.model.deployment.v1.model_artifact_identifier_pb2 import (
+    ModelArtifactIdentifier,
+    ModelBasedArtifactIdentifier,
+)
+from frogml._proto.com.jfrog.ml.model.deployment.v1.model_deployment_brief_pb2 import (
+    ModelDeploymentBrief,
+)
+from google.protobuf.timestamp_pb2 import Timestamp
+from pydantic import BaseModel, ConfigDict
+
+
+class DeploymentInformation(BaseModel):
+    model_config = ConfigDict(arbitrary_types_allowed=True)
+
+    deployment_id: uuid.UUID
+    model_id: str
+    model_group_id: str
+    path: str
+    environment_id: uuid.UUID
+    deployment_config: Deployment
+    created_at: datetime
+
+
+class ModelDeploymentManagerMock(ModelDeploymentServiceServicer):
+    def __init__(self):
+        self.__deployment_id_to_spec: dict[str, DeploymentInformation] = {}
+
+    @staticmethod
+    def __get_model_identifier_details(
+        model_artifact_identifier: ModelArtifactIdentifier,
+    ) -> tuple[str, str, str]:
+        identifier_type = model_artifact_identifier.WhichOneof("identifier_type")
+
+        if identifier_type == "model_based_artifact_id":
+            artifact = model_artifact_identifier.model_based_artifact_id
+            return (
+                artifact.model_id,
+                artifact.model_group_id,
+                artifact.image_path,
+            )
+        elif identifier_type == "custom_model_artifact_id":
+            artifact = model_artifact_identifier.custom_model_artifact_id
+            return (
+                artifact.model_id,
+                artifact.model_group_id,
+                artifact.build_id,
+            )
+
+        raise Exception("No Model Artifact Identifier found")
+
+    def DeployModel(
+        self, request: DeployModelRequest, context: grpc.ServicerContext
+    ) -> DeployModelResponse:
+        multiple_environment_deployment: MultipleEnvironmentDeployment = (
+            request.multiple_environment_deployment
+        )
+        model_artifact_identifier: ModelArtifactIdentifier = (
+            multiple_environment_deployment.model_artifact_identifier
+        )
+        model_id, model_group_id, path = self.__get_model_identifier_details(
+            model_artifact_identifier
+        )
+
+        self.__deployment_id_to_spec.update(
+            {
+                (deployment_id := str(uuid.uuid4())): DeploymentInformation(
+                    deployment_id=uuid.UUID(deployment_id),
+                    model_id=model_id,
+                    model_group_id=model_group_id,
+                    path=path,
+                    environment_id=uuid.UUID(env_id),
+                    deployment_config=deployment,
+                    created_at=datetime.now(tz=timezone.utc),
+                )
+                for env_id, deployment in multiple_environment_deployment.deployments.items()
+            }
+        )
+
+        return DeployModelResponse()
+
+    def UndeployModel(
+        self, request: UndeployModelRequest, context: grpc.ServicerContext
+    ) -> UndeployModelResponse:
+        deployment_ids = list(request.deployment_ids.deployment_id)
+        for deployment_id in deployment_ids:
+            self.__deployment_id_to_spec.pop(deployment_id)
+
+        return UndeployModelResponse()
+
+    def EditModelDeployment(
+        self, request: EditModelDeploymentRequest, context: grpc.ServicerContext
+    ) -> EditModelDeploymentResponse:
+        deployment_map = dict(request.deployment_id_to_deployment_spec_map.deployments)
+        for deployment_id, new_spec in deployment_map.items():
+            if deployment_id not in self.__deployment_id_to_spec:
+                raise Exception("Deployment id does not exist")
+            self.__deployment_id_to_spec[deployment_id].deployment_config = new_spec
+
+        return EditModelDeploymentResponse()
+
+    @staticmethod
+    def __build_filter_predicates(
+        request: ListModelDeploymentsRequest,
+    ) -> list[Callable[[DeploymentInformation], bool]]:
+        """Build a list of filter predicates from the request."""
+        predicates: list[Callable[[DeploymentInformation], bool]] = []
+
+        if not request.model_deployment_filter.HasField(
+            "simple_model_deployment_filter"
+        ):
+            return predicates
+
+        simple_filter = request.model_deployment_filter.simple_model_deployment_filter
+
+        if simple_filter.HasField("model_identifier_filter"):
+            model_group_id = simple_filter.model_identifier_filter.model_group_id
+            model_id = simple_filter.model_identifier_filter.model_id
+            predicates.append(
+                lambda d, model_group_id=model_group_id, model_id=model_id: d.model_group_id
+                == model_group_id
+                and d.model_id == model_id
+            )
+
+        if simple_filter.HasField("model_group_ids"):
+            allowed_model_group_ids = set(simple_filter.model_group_ids.model_group_id)
+            predicates.append(
+                lambda d, allowed=allowed_model_group_ids: d.model_group_id in allowed
+            )
+
+        if simple_filter.HasField("model_ids"):
+            allowed_model_ids = set(simple_filter.model_ids.model_id)
+            predicates.append(
+                lambda d, allowed=allowed_model_ids: d.model_id in allowed
+            )
+
+        if simple_filter.HasField("environment_ids"):
+            allowed_environment_ids = set(simple_filter.environment_ids.environment_id)
+            predicates.append(
+                lambda d, allowed=allowed_environment_ids: str(d.environment_id)
+                in allowed
+            )
+
+        return predicates
+
+    @staticmethod
+    def _deployment_to_brief(
+        deployment_info: DeploymentInformation,
+    ) -> ModelDeploymentBrief:
+        """Convert a DeploymentInformation to a ModelDeploymentBrief."""
+        timestamp = Timestamp()
+        timestamp.FromDatetime(deployment_info.created_at)
+
+        return ModelDeploymentBrief(
+            status=ModelDeploymentStatus(
+                model_deployment_status_successful=ModelDeploymentStatusSuccessful()
+            ),
+            environment=str(deployment_info.environment_id),
+            created_at=timestamp,
+            deployment_id=str(deployment_info.deployment_id),
+            model_artifact_identifier=ModelArtifactIdentifier(
+                model_based_artifact_id=ModelBasedArtifactIdentifier(
+                    model_id=deployment_info.model_id,
+                    model_group_id=deployment_info.model_group_id,
+                    image_path=deployment_info.path,
+                )
+            ),
+        )
+
+    def ListModelDeployments(
+        self, request: ListModelDeploymentsRequest, context: grpc.ServicerContext
+    ) -> ListModelDeploymentsResponse:
+        predicates: list[Callable[[DeploymentInformation], bool]] = (
+            self.__build_filter_predicates(request)
+        )
+
+        result = [
+            self._deployment_to_brief(deployment)
+            for deployment in self.__deployment_id_to_spec.values()
+            if all(predicate(deployment) for predicate in predicates)
+        ]
+
+        return ListModelDeploymentsResponse(model_deployment_brief=result)
+
+    def GetModelDeployment(
+        self, request: GetModelDeploymentRequest, context: grpc.ServicerContext
+    ) -> GetModelDeploymentResponse:
+        deployment_id = request.deployment_id
+        deployment_info = self.__deployment_id_to_spec.get(deployment_id)
+
+        if deployment_info is None:
+            return GetModelDeploymentResponse()
+
+        timestamp = Timestamp()
+        timestamp.FromDatetime(deployment_info.created_at)
+
+        model_artifact_identifier = ModelArtifactIdentifier(
+            model_based_artifact_id=ModelBasedArtifactIdentifier(
+                model_id=deployment_info.model_id,
+                model_group_id=deployment_info.model_group_id,
+                image_path=deployment_info.path,
+            )
+        )
+
+        model_deployment_info = ModelDeploymentInformation(
+            model_artifact_identifier=model_artifact_identifier,
+            model_deployment_spec=deployment_info.deployment_config,
+            model_deployment_metadata=DeploymentMetadata(
+                deployment_id=str(deployment_info.deployment_id),
+                status=ModelDeploymentStatus(
+                    model_deployment_status_successful=ModelDeploymentStatusSuccessful()
+                ),
+                environment=str(deployment_info.environment_id),
+                created_at=timestamp,
+            ),
+        )
+
+        return GetModelDeploymentResponse(
+            model_deployment_information=model_deployment_info
+        )
+
+    def get_deployment_by_id(
+        self, deployment_id: str
+    ) -> Optional[DeploymentInformation]:
+        return self.__deployment_id_to_spec.get(deployment_id)
+
+    def get_all_deployments(self) -> list[DeploymentInformation]:
+        return list(self.__deployment_id_to_spec.values())

frogml_services_mock/services_mock.py

@@ -3,7 +3,9 @@ from typing import Any, Generator, List, Tuple
 
 import grpc
 import pytest
-
+from frogml._proto.com.jfrog.ml.model.deployment.v1.deployment_service_pb2_grpc import (
+    add_ModelDeploymentServiceServicer_to_server,
+)
 from frogml._proto.jfml.model_version.v1.model_version_manager_service_pb2_grpc import (
     add_ModelVersionManagerServiceServicer_to_server,
 )
@@ -189,6 +191,9 @@ from frogml_services_mock.mocks.location_discovery_service_api import (
     LocationDiscoveryServiceApiMock,
 )
 from frogml_services_mock.mocks.logging_service import LoggingServiceApiMock
+from frogml_services_mock.mocks.model_deployment_manager_service_mock import (
+    ModelDeploymentManagerMock,
+)
 from frogml_services_mock.mocks.model_group_management_service import (
     ModelGroupManagementServiceMock,
 )
@@ -248,13 +253,18 @@ def frogml_container():
         kube_deployment_captain,
         location_discovery,
         logging_client,
+        model_deployment_manager,
         model_group_management,
         model_management,
         model_version_manager,
         secret_service,
         user_application_instance,
     )
-    from frogml.core.clients.administration import eco_system, self_service, authentication
+    from frogml.core.clients.administration import (
+        authentication,
+        eco_system,
+        self_service,
+    )
     from frogml.core.clients.integration_management import integration_manager_client
     from frogml.core.clients.system_secret import system_secret_client
 
@@ -287,6 +297,7 @@ def frogml_container():
             model_version_manager,
             jfrog_gateway,
             location_discovery,
+            model_deployment_manager,
         ]
     )
 
@@ -501,6 +512,11 @@ def attach_servicers(free_port, server):
                 JFrogTenantInfoServiceMock,
                 add_JFrogTenantInfoServiceServicer_to_server,
             ),
+            (
+                "model_deployment_manager_mock",
+                ModelDeploymentManagerMock,
+                add_ModelDeploymentServiceServicer_to_server,
+            ),
             ("port", free_port, None),
         ],
     )