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

frogml/core/clients/batch_job_management/client.py

@@ -71,7 +71,7 @@ from .results import (
     GetExecutionReportResult,
     StartExecutionResult,
 )
-from typing import List
+from typing import List, Optional
 
 CLIENT_TIMEOUT = 180  # Seconds
 
@@ -359,15 +359,21 @@ class BatchJobManagerClient:
         except grpc.RpcError as e:
             raise FrogmlException(f"Failed to cancel execution, error is {e}")
 
-    def get_execution_report(self, execution_id: str) -> GetExecutionReportResult:
+    def get_execution_report(
+        self,
+        execution_id: str,
+        model_id: Optional[str] = None,
+        model_group_name: Optional[str] = None,
+    ) -> GetExecutionReportResult:
         """
 
         Args:
-            execution_id: the batch execution id to get the current status of
+            execution_id: The batch execution id to get the report for.
+            model_id: The model ID. This is the display name(identifier), not the UUID.
+            model_group_name: The model group name.
 
         Returns:
-            A full report of all the events that occurred as part of the execution job
-
+            A full report of all the events that occurred as part of the execution job.
         """
         try:
             self.logging_client = (
@@ -383,7 +389,9 @@ class BatchJobManagerClient:
             try:
                 execution_log_response: ReadLogsResponse = (
                     self.logging_client.read_execution_models_logs(
-                        execution_id=execution_id
+                        execution_id=execution_id,
+                        model_id=model_id,
+                        model_group_name=model_group_name,
                     )
                 )
                 execution_logs = execution_log_response.log_line

frogml/core/clients/logging_client/client.py

@@ -49,15 +49,41 @@ class LoggingClient:
 
     def read_build_logs(
         self,
-        build_id=None,
-        before_offset=None,
-        after_offset=None,
-        max_number_of_results=None,
-        log_text_filter=None,
-    ):
+        build_id: Optional[str] = None,
+        before_offset: Optional[str] = None,
+        after_offset: Optional[str] = None,
+        max_number_of_results: Optional[int] = None,
+        log_text_filter: Optional[str] = None,
+        model_id: Optional[str] = None,
+        model_group_name: Optional[str] = None,
+    ) -> ReadLogsResponse:
+        """
+        Read logs for a remote build.
+
+        Args:
+            build_id: The build ID to fetch logs for
+            before_offset: Optional offset to read logs before this point
+            after_offset: Optional offset to read logs after this point
+            max_number_of_results: Optional maximum number of log lines to return
+            log_text_filter: Optional text filter to search for in logs
+            model_id: The model ID. This is the display name(identifier), not the UUID.
+            model_group_name: The model group name.
+
+        Returns:
+            ReadLogsResponse containing the log lines
+
+        Raises:
+            FrogmlException: If reading logs fails
+        """
         try:
+            remote_build_source = RemoteBuildSource(
+                build_id=build_id,
+                model_id=model_id,
+                model_group_name=model_group_name,
+            )
+
             response = self.read_logs(
-                source=LogSource(remote_build=RemoteBuildSource(build_id=build_id)),
+                source=LogSource(remote_build=remote_build_source),
                 before_offset=before_offset,
                 after_offset=after_offset,
                 log_text_filter=log_text_filter,
@@ -70,20 +96,44 @@ class LoggingClient:
 
     def read_model_runtime_logs(
         self,
-        build_id=None,
-        deployment_id=None,
-        before_offset=None,
-        after_offset=None,
-        max_number_of_results=None,
-        log_text_filter=None,
+        build_id: Optional[str] = None,
+        deployment_id: Optional[str] = None,
+        before_offset: Optional[str] = None,
+        after_offset: Optional[str] = None,
+        max_number_of_results: Optional[int] = None,
+        log_text_filter: Optional[str] = None,
+        model_id: Optional[str] = None,
+        model_group_name: Optional[str] = None,
     ) -> ReadLogsResponse:
+        """
+        Read logs for a model runtime (deployment).
+
+        Args:
+            build_id: The build ID to fetch logs for
+            deployment_id: The deployment ID to fetch logs for
+            before_offset: Optional offset to read logs before this point
+            after_offset: Optional offset to read logs after this point
+            max_number_of_results: Optional maximum number of log lines to return
+            log_text_filter: Optional text filter to search for in logs
+            model_id: The model ID. This is the display name(identifier), not the UUID.
+            model_group_name: The model group name.
+
+        Returns:
+            ReadLogsResponse containing the log lines
+
+        Raises:
+            FrogmlException: If reading logs fails
+        """
         try:
+            model_runtime_source = ModelRuntimeSource(
+                build_id=build_id,
+                deployment_id=deployment_id,
+                model_id=model_id,
+                model_group_name=model_group_name,
+            )
+
             response = self.read_logs(
-                source=LogSource(
-                    model_runtime=ModelRuntimeSource(
-                        build_id=build_id, deployment_id=deployment_id
-                    )
-                ),
+                source=LogSource(model_runtime=model_runtime_source),
                 before_offset=before_offset,
                 after_offset=after_offset,
                 log_text_filter=log_text_filter,
@@ -96,19 +146,41 @@ class LoggingClient:
 
     def read_execution_models_logs(
         self,
-        execution_id,
-        before_offset=None,
-        after_offset=None,
-        max_number_of_results=None,
-        log_text_filter=None,
+        execution_id: str,
+        before_offset: Optional[str] = None,
+        after_offset: Optional[str] = None,
+        max_number_of_results: Optional[int] = None,
+        log_text_filter: Optional[str] = None,
+        model_id: Optional[str] = None,
+        model_group_name: Optional[str] = None,
     ) -> ReadLogsResponse:
+        """
+        Read logs for a model inference execution (batch job).
+
+        Args:
+            execution_id: The execution/inference job ID to fetch logs for
+            before_offset: Optional offset to read logs before this point
+            after_offset: Optional offset to read logs after this point
+            max_number_of_results: Optional maximum number of log lines to return
+            log_text_filter: Optional text filter to search for in logs
+            model_id: The model ID. This is the display name(identifier), not the UUID.
+            model_group_name: The model group name.
+
+        Returns:
+            ReadLogsResponse containing the log lines
+
+        Raises:
+            FrogmlException: If reading logs fails
+        """
         try:
+            inference_execution_source = InferenceExecutionSource(
+                inference_job_id=execution_id,
+                model_id=model_id,
+                model_group_name=model_group_name,
+            )
+
             response = self.read_logs(
-                source=LogSource(
-                    inference_execution=InferenceExecutionSource(
-                        inference_job_id=execution_id
-                    )
-                ),
+                source=LogSource(inference_execution=inference_execution_source),
                 before_offset=before_offset,
                 after_offset=after_offset,
                 log_text_filter=log_text_filter,
@@ -122,12 +194,25 @@ class LoggingClient:
     @grpc_try_catch_wrapper("Failed to read logs request")
     def read_logs(
         self,
-        source,
-        before_offset,
-        after_offset,
-        max_number_of_results,
-        log_text_filter,
+        source: LogSource,
+        before_offset: Optional[str],
+        after_offset: Optional[str],
+        max_number_of_results: Optional[int],
+        log_text_filter: Optional[str],
     ) -> ReadLogsResponse:
+        """
+        Low-level method to read logs from any source.
+
+        Args:
+            source: The log source to read from
+            before_offset: Optional offset to read logs before this point
+            after_offset: Optional offset to read logs after this point
+            max_number_of_results: Optional maximum number of log lines to return
+            log_text_filter: Optional text filter to search for in logs
+
+        Returns:
+            ReadLogsResponse containing the log lines
+        """
         response: ReadLogsResponse = self._logging_service.ReadLogs(
             ReadLogsRequest(
                 source=source,

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.35.dist-info → frogml-1.2.41.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml
-Version: 1.2.35
+Version: 1.2.41
 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.35.dist-info → frogml-1.2.41.dist-info}/RECORD

@@ -1,4 +1,34 @@
-frogml/__init__.py,sha256=qCiP-lN6Bb32rwIiGrzBPQVv6qPplDfH17uU5IH8bIU,741
+frogml/__init__.py,sha256=Iha2EFEpm8UEhQZulMOPWPSHZKElxWm4XT9dGhtHKck,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
@@ -530,8 +560,8 @@ frogml/_proto/qwak/logging/log_line_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHD
 frogml/_proto/qwak/logging/log_reader_service_pb2.py,sha256=mGzDgASTP_oJlVcPkId5JxjhpUbh7O_CQ9xbcQRUOXI,2561
 frogml/_proto/qwak/logging/log_reader_service_pb2.pyi,sha256=KdIIgfFCw0I-51d2dEMZy3zVgkGw4tZNF26WcR5yNz4,3451
 frogml/_proto/qwak/logging/log_reader_service_pb2_grpc.py,sha256=5iwiKKEWLCgNl6KyUanuUFpsg-159HgiF6ugMvGgW9k,2833
-frogml/_proto/qwak/logging/log_source_pb2.py,sha256=KfVCEihMIVtJ2vVMR0jdiFCUsdY9RDJ7YDBVyzps-fk,6170
-frogml/_proto/qwak/logging/log_source_pb2.pyi,sha256=YJ3xGWEgG4dF9CvsF_X1dBITgHMxs7IdkGsWQ4G0uNg,18672
+frogml/_proto/qwak/logging/log_source_pb2.py,sha256=7wzXkpjxW6tEYJ-hZJ1s_7eB1G5m2K2_WEjL8NtyfqY,6971
+frogml/_proto/qwak/logging/log_source_pb2.pyi,sha256=9ekSIfto3LP-e2pyVwHjx_ewbDxaHpQf_siyT4zdP_4,20017
 frogml/_proto/qwak/logging/log_source_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
 frogml/_proto/qwak/model_descriptor/open_ai_descriptor_pb2.py,sha256=-4tcS6KSdpUK3knQ9YUyQTsMcHFaPTQ3A8kHPOLVPSQ,4203
 frogml/_proto/qwak/model_descriptor/open_ai_descriptor_pb2.pyi,sha256=7ZaPgYKgRhZvq9LMaj7jWlQxKhQ9fRUDK86SaJvMuu0,11801
@@ -542,9 +572,9 @@ frogml/_proto/qwak/model_group/model_group_pb2_grpc.py,sha256=YgDt3DtagEdZ7UtxcP
 frogml/_proto/qwak/model_group/model_group_repository_details_pb2.py,sha256=BxaVjDayd9h5Va2ZId3pNPeJFAqVXb8TJHIWmYsURN4,3806
 frogml/_proto/qwak/model_group/model_group_repository_details_pb2.pyi,sha256=3oCsYmmfvBjGkA1nRcx2cg11WNr40uuR3gAWXC7Wz2I,7245
 frogml/_proto/qwak/model_group/model_group_repository_details_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
-frogml/_proto/qwak/models/models_pb2.py,sha256=XF3-e8xW62ymbg7xGRCwc4fNsgXV2UZJ1lPgv47Ylkk,19046
-frogml/_proto/qwak/models/models_pb2.pyi,sha256=m2NGA7K3Jz3YFp8nOW4l_AB5kgjJUEDmr3xMUPYu3Dc,42638
-frogml/_proto/qwak/models/models_pb2_grpc.py,sha256=iOfvU_txYazEBCR2rlFlD3BXLseuSjKzXqCcwfI60h4,23627
+frogml/_proto/qwak/models/models_pb2.py,sha256=jtw5KSbMwX5q6NVhfA5eblpJIANVavURyYvmXYvFVr0,19427
+frogml/_proto/qwak/models/models_pb2.pyi,sha256=c8e6l8fxCXXJmlyV9GSZ7PTThelw79sCYiEpMKaMRZE,43564
+frogml/_proto/qwak/models/models_pb2_grpc.py,sha256=7HZ5RwIk9I0MOHMOvZhvPFq_UIKwVdjZ79N3h-dlk8o,25406
 frogml/_proto/qwak/models/models_query_pb2.py,sha256=ZDEx_w4JgAL1q2wNOq5qkXiO5DctBgVRWbjrpOmOzbo,2880
 frogml/_proto/qwak/models/models_query_pb2.pyi,sha256=JEWkO0TXbGD2uKElgZB6VNPfYoq99GQ3Nj3jHObBQ2s,7490
 frogml/_proto/qwak/models/models_query_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
@@ -656,7 +686,7 @@ frogml/core/clients/automation_management/client.py,sha256=Dk6SsZGMLaVDLLzvFBysL
 frogml/core/clients/autoscaling/__init__.py,sha256=A-zuZOWaZf16NjnpX38204m730c5Lde491KaXneGirE,38
 frogml/core/clients/autoscaling/client.py,sha256=d1BoUH4hBiqoSR6ZM2_Ds6H7gV9HZPM2IIYNDbjPGgY,1299
 frogml/core/clients/batch_job_management/__init__.py,sha256=zywxYf2JupkIWnGVB2C6ugZs5CrOdxTPrksD5P6uQu4,211
-frogml/core/clients/batch_job_management/client.py,sha256=yym_GRJlB2qWnFf58EK2i7iD4llSj3FwmjdmIEdBSRc,22621
+frogml/core/clients/batch_job_management/client.py,sha256=TBN_9MkHXW342PLRepUckuQmA_gLPPrq5VavkkDtnCQ,22979
 frogml/core/clients/batch_job_management/executions_config.py,sha256=tyQbNiIAo3U4t_vyQp8nY17U0oQIyuE5GLW5WqhDxjA,7045
 frogml/core/clients/batch_job_management/results.py,sha256=nrFGZ3y538FBhej94CYV4cwqQssK1aUQOLGubYdQVrc,1846
 frogml/core/clients/build_orchestrator/__init__.py,sha256=ReW7Lbjws06MbVlCTRdsymDZiS2CKuczXIZ1xnPKSdg,105
@@ -691,7 +721,9 @@ frogml/core/clients/kube_deployment_captain/client.py,sha256=la4V7WsVL1z4zuLQtPZ
 frogml/core/clients/location_discovery/__init__.py,sha256=sqGQ75YHFE6nvOcir38fykUUmAa6cFEIze8PJYgYWRc,44
 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=vq2Fv7UAEnrG910Npp1COxxWQ2_NWofbAStPkLhiz1w,4836
+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.35.dist-info/METADATA,sha256=VJC2cI-AIqs8Ut2FSG7oQQBYQsWpEWl39_C57crPn80,5599
-frogml-1.2.35.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-frogml-1.2.35.dist-info/RECORD,,
+frogml-1.2.41.dist-info/METADATA,sha256=-RAwoyHVI5VO_IY9sSKBZXVs9EJzKAd6qIHhXxKdZIc,5599
+frogml-1.2.41.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+frogml-1.2.41.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