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

frogml/_proto/qwak/administration/v1/environments/environment_service_pb2.pyi

@@ -0,0 +1,59 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import frogml._proto.qwak.administration.v0.environments.environment_pb2
+import frogml._proto.qwak.administration.v1.environments.environment_pb2
+import typing
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class CreateEnvironmentRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ENVIRONMENT_NAME_FIELD_NUMBER: builtins.int
+    CLUSTER_ID_FIELD_NUMBER: builtins.int
+    SPEC_FIELD_NUMBER: builtins.int
+    environment_name: builtins.str
+    """Environment name"""
+    cluster_id: builtins.str
+    """The JFrogML Cluster the environment is deployed in."""
+    @property
+    def spec(self) -> frogml._proto.qwak.administration.v1.environments.environment_pb2.EnvironmentRuntimeConfigSpec:
+        """Environment configuration"""
+
+    def __init__(
+        self,
+        *,
+        environment_name: builtins.str = ...,
+        cluster_id: builtins.str = ...,
+        spec: frogml._proto.qwak.administration.v1.environments.environment_pb2.EnvironmentRuntimeConfigSpec | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["spec", b"spec"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["cluster_id", b"cluster_id", "environment_name", b"environment_name", "spec", b"spec"]) -> None: ...
+
+global___CreateEnvironmentRequest = CreateEnvironmentRequest
+
+@typing.final
+class CreateEnvironmentResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ENVIRONMENT_FIELD_NUMBER: builtins.int
+    @property
+    def environment(self) -> frogml._proto.qwak.administration.v0.environments.environment_pb2.QwakEnvironment:
+        """Created environment"""
+
+    def __init__(
+        self,
+        *,
+        environment: frogml._proto.qwak.administration.v0.environments.environment_pb2.QwakEnvironment | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["environment", b"environment"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["environment", b"environment"]) -> None: ...
+
+global___CreateEnvironmentResponse = CreateEnvironmentResponse

frogml/_proto/qwak/administration/v1/environments/environment_service_pb2_grpc.py

@@ -0,0 +1,67 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from frogml._proto.qwak.administration.v1.environments import environment_service_pb2 as qwak_dot_administration_dot_v1_dot_environments_dot_environment__service__pb2
+
+
+class EnvironmentServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.CreateEnvironment = channel.unary_unary(
+                '/qwak.administration.v1.environment.EnvironmentService/CreateEnvironment',
+                request_serializer=qwak_dot_administration_dot_v1_dot_environments_dot_environment__service__pb2.CreateEnvironmentRequest.SerializeToString,
+                response_deserializer=qwak_dot_administration_dot_v1_dot_environments_dot_environment__service__pb2.CreateEnvironmentResponse.FromString,
+                )
+
+
+class EnvironmentServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def CreateEnvironment(self, request, context):
+        """Create an environment
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_EnvironmentServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'CreateEnvironment': grpc.unary_unary_rpc_method_handler(
+                    servicer.CreateEnvironment,
+                    request_deserializer=qwak_dot_administration_dot_v1_dot_environments_dot_environment__service__pb2.CreateEnvironmentRequest.FromString,
+                    response_serializer=qwak_dot_administration_dot_v1_dot_environments_dot_environment__service__pb2.CreateEnvironmentResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'qwak.administration.v1.environment.EnvironmentService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class EnvironmentService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def CreateEnvironment(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.administration.v1.environment.EnvironmentService/CreateEnvironment',
+            qwak_dot_administration_dot_v1_dot_environments_dot_environment__service__pb2.CreateEnvironmentRequest.SerializeToString,
+            qwak_dot_administration_dot_v1_dot_environments_dot_environment__service__pb2.CreateEnvironmentResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

frogml/core/clients/administration/cluster_v2/__init__.py

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

frogml/core/clients/administration/cluster_v2/client.py

@@ -0,0 +1,122 @@
+"""Cluster V2 API client.
+
+Provides wrapper methods for the Cluster V2 gRPC service with proper error handling.
+"""
+
+from frogml._proto.qwak.administration.cluster.v2.cluster_pb2 import (
+    ClusterSpec,
+)
+from frogml._proto.qwak.administration.cluster.v2.cluster_service_pb2 import (
+    CreateClusterCreationRequestRequest,
+    CreateClusterCreationRequestResponse,
+    DeleteClusterRequest,
+    DeleteClusterResponse,
+    GetClusterRequest,
+    GetClusterResponse,
+    GetClusterStateRequest,
+    GetClusterStateResponse,
+    ListClustersRequest,
+    ListClustersResponse,
+    UpdateClusterConfigurationRequest,
+    UpdateClusterConfigurationResponse,
+)
+from frogml._proto.qwak.administration.cluster.v2.cluster_service_pb2_grpc import (
+    ClusterServiceStub,
+)
+from dependency_injector.wiring import Provide
+from frogml.core.inner.di_configuration import FrogmlContainer
+from frogml.core.inner.tool.grpc.grpc_try_wrapping import grpc_try_catch_wrapper
+
+
+class ClusterV2Client:
+    """Client for interacting with the Cluster V2 API.
+
+    This client wraps the gRPC stub and provides methods with proper error handling
+    using the grpc_try_catch_wrapper decorator.
+    """
+
+    def __init__(self, grpc_channel=Provide[FrogmlContainer.core_grpc_channel]):
+        self.__cluster_service = ClusterServiceStub(grpc_channel)
+
+    @grpc_try_catch_wrapper("Failed to create cluster")
+    def create_cluster(
+        self, cluster_spec: ClusterSpec
+    ) -> CreateClusterCreationRequestResponse:
+        """Create a new cluster.
+
+        Args:
+            cluster_spec: The cluster specification.
+
+        Returns:
+            CreateClusterCreationRequestResponse with the request_id.
+        """
+        request = CreateClusterCreationRequestRequest(cluster_spec=cluster_spec)
+        return self.__cluster_service.CreateClusterCreationRequest(request)
+
+    @grpc_try_catch_wrapper("Failed to get cluster state for request {request_id}")
+    def get_cluster_state(self, request_id: str) -> GetClusterStateResponse:
+        """Get cluster state by request ID.
+
+        Args:
+            request_id: The request ID from create_cluster.
+
+        Returns:
+            GetClusterStateResponse with the cluster state.
+        """
+        request = GetClusterStateRequest(request_id=request_id)
+        return self.__cluster_service.GetClusterState(request)
+
+    @grpc_try_catch_wrapper("Failed to get cluster {cluster_id}")
+    def get_cluster(self, cluster_id: str) -> GetClusterResponse:
+        """Get cluster by ID.
+
+        Args:
+            cluster_id: The cluster ID.
+
+        Returns:
+            GetClusterResponse with the cluster state.
+        """
+        request = GetClusterRequest(cluster_id=cluster_id)
+        return self.__cluster_service.GetCluster(request)
+
+    @grpc_try_catch_wrapper("Failed to list clusters")
+    def list_clusters(self) -> ListClustersResponse:
+        """List all clusters.
+
+        Returns:
+            ListClustersResponse with list of clusters.
+        """
+        request = ListClustersRequest()
+        return self.__cluster_service.ListClusters(request)
+
+    @grpc_try_catch_wrapper("Failed to update cluster {cluster_id}")
+    def update_cluster(
+        self, cluster_id: str, cluster_spec: ClusterSpec
+    ) -> UpdateClusterConfigurationResponse:
+        """Update cluster configuration.
+
+        Args:
+            cluster_id: The cluster ID to update.
+            cluster_spec: The updated cluster specification.
+
+        Returns:
+            UpdateClusterConfigurationResponse with the request_id.
+        """
+        request = UpdateClusterConfigurationRequest(
+            cluster_id=cluster_id,
+            cluster_spec=cluster_spec,
+        )
+        return self.__cluster_service.UpdateClusterConfiguration(request)
+
+    @grpc_try_catch_wrapper("Failed to delete cluster {cluster_id}")
+    def delete_cluster(self, cluster_id: str) -> DeleteClusterResponse:
+        """Delete cluster by ID.
+
+        Args:
+            cluster_id: The cluster ID to delete.
+
+        Returns:
+            DeleteClusterResponse with the request_id.
+        """
+        request = DeleteClusterRequest(cluster_id=cluster_id)
+        return self.__cluster_service.DeleteCluster(request)

frogml/core/clients/administration/environment/client.py

@@ -4,17 +4,23 @@ from dependency_injector.wiring import Provide
 from frogml._proto.qwak.administration.v0.environments.environment_pb2 import (
     QwakEnvironmentStatus,
 )
+from frogml._proto.qwak.administration.v0.environments.configuration_pb2 import (
+    QwakEnvironmentConfiguration,
+)
 from frogml._proto.qwak.administration.v0.environments.environment_service_pb2 import (
     GetEnvironmentApplicationUserCredentialsRequest,
     GetEnvironmentApplicationUserCredentialsResponse,
     ListEnvironmentsRequest,
     ListEnvironmentsResponse,
+    UpdateEnvironmentConfigurationRequest,
+    UpdateEnvironmentConfigurationResponse,
 )
 from frogml._proto.qwak.administration.v0.environments.environment_service_pb2_grpc import (
     EnvironmentServiceStub,
 )
 from frogml.core.exceptions import FrogmlException
 from frogml.core.inner.di_configuration import FrogmlContainer
+from frogml.core.inner.tool.grpc.grpc_try_wrapping import grpc_try_catch_wrapper
 
 
 class EnvironmentClient:
@@ -75,3 +81,25 @@ class EnvironmentClient:
             raise FrogmlException(
                 f"Failed to get application user, error is {e.details()}"
             )
+
+    @grpc_try_catch_wrapper(
+        "Failed to update environment configuration for {environment_id}"
+    )
+    def update_environment_configuration(
+        self, environment_id: str, configuration: QwakEnvironmentConfiguration
+    ) -> UpdateEnvironmentConfigurationResponse:
+        """
+        Update environment configuration.
+
+        Args:
+            environment_id: The environment ID to update.
+            configuration: The new environment configuration.
+
+        Returns:
+            UpdateEnvironmentConfigurationResponse
+        """
+        request = UpdateEnvironmentConfigurationRequest(
+            environment_id=environment_id,
+            configuration=configuration,
+        )
+        return self._environment_service.UpdateEnvironmentConfiguration(request)

frogml/core/clients/administration/environment_v1/__init__.py

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

frogml/core/clients/administration/environment_v1/client.py

@@ -0,0 +1,53 @@
+"""Environment V1 API client.
+
+Provides wrapper methods for the Environment V1 gRPC service with proper error handling.
+"""
+
+from frogml._proto.qwak.administration.v1.environments.environment_pb2 import (
+    EnvironmentRuntimeConfigSpec,
+)
+from frogml._proto.qwak.administration.v1.environments.environment_service_pb2 import (
+    CreateEnvironmentRequest,
+    CreateEnvironmentResponse,
+)
+from frogml._proto.qwak.administration.v1.environments.environment_service_pb2_grpc import (
+    EnvironmentServiceStub,
+)
+from dependency_injector.wiring import Provide
+from frogml.core.inner.di_configuration import FrogmlContainer
+from frogml.core.inner.tool.grpc.grpc_try_wrapping import grpc_try_catch_wrapper
+
+
+class EnvironmentV1Client:
+    """Client for interacting with the Environment V1 API.
+
+    This client wraps the gRPC stub and provides methods with proper error handling
+    using the grpc_try_catch_wrapper decorator.
+    """
+
+    def __init__(self, grpc_channel=Provide[FrogmlContainer.core_grpc_channel]):
+        self.__environment_service = EnvironmentServiceStub(grpc_channel)
+
+    @grpc_try_catch_wrapper("Failed to create environment {environment_name}")
+    def create_environment(
+        self,
+        environment_name: str,
+        cluster_id: str,
+        spec: EnvironmentRuntimeConfigSpec,
+    ) -> CreateEnvironmentResponse:
+        """Create a new environment.
+
+        Args:
+            environment_name: The name of the environment.
+            cluster_id: The cluster ID the environment belongs to.
+            spec: The environment runtime configuration specification.
+
+        Returns:
+            CreateEnvironmentResponse with the created environment.
+        """
+        request = CreateEnvironmentRequest(
+            environment_name=environment_name,
+            cluster_id=cluster_id,
+            spec=spec,
+        )
+        return self.__environment_service.CreateEnvironment(request)

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

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

@@ -1,4 +1,4 @@
-frogml/__init__.py,sha256=sJOd_xEn0Cz_WXUSQWRDjwsaTMl7KajIC5Sop94NZe8,741
+frogml/__init__.py,sha256=jRkOZMd727GgKh3ZSEEs-cNVuwnimIo8NbArOnSrmp4,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
@@ -83,6 +83,9 @@ frogml/_proto/qwak/administration/authenticated_user/v1/details_pb2_grpc.py,sha2
 frogml/_proto/qwak/administration/cluster/v2/cluster_pb2.py,sha256=mn2N1ewXavzKj6fYOW7mtW3skdJHsIyDk0PoHGjV7xA,7710
 frogml/_proto/qwak/administration/cluster/v2/cluster_pb2.pyi,sha256=4dP_rJkV6nnxGkeJghFeq04mSaenDhKjrLsbk-ww6gA,17336
 frogml/_proto/qwak/administration/cluster/v2/cluster_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
+frogml/_proto/qwak/administration/cluster/v2/cluster_service_pb2.py,sha256=LO7i_xO2JCj3O5Jk1GonvuH47tmYqkcOxuANRNbGHh0,5388
+frogml/_proto/qwak/administration/cluster/v2/cluster_service_pb2.pyi,sha256=7Vn_OEVzdqDS2zcRez-IdsbftalVAkTI6IZZKxq3Pls,7344
+frogml/_proto/qwak/administration/cluster/v2/cluster_service_pb2_grpc.py,sha256=UDuRQ_AIKghmnp8z--FqG5mLlH26aVLBLQqj4jMSM74,12944
 frogml/_proto/qwak/administration/runtime_configuration/v0/container_registry_config_pb2.py,sha256=Qrxc5jKlR90Yi0sU7poq8NRUf4sZf0_TUTUab71XPis,2853
 frogml/_proto/qwak/administration/runtime_configuration/v0/container_registry_config_pb2.pyi,sha256=a_K7CSexEa3ewJUCsuX4ym0DUy4Qti0WGVC7TBN7NxE,2534
 frogml/_proto/qwak/administration/runtime_configuration/v0/container_registry_config_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
@@ -164,6 +167,12 @@ frogml/_proto/qwak/administration/v0/environments/personalization_pb2_grpc.py,sh
 frogml/_proto/qwak/administration/v0/users/user_pb2.py,sha256=VCv_1ojHeRH26hQmhzL9lomfDw1JRLseI75Le4Eo-6s,4795
 frogml/_proto/qwak/administration/v0/users/user_pb2.pyi,sha256=eUHlh367Qz0nQSvgxkkiOt3Qiy_1Rtm34_eaxUC2WKo,10514
 frogml/_proto/qwak/administration/v0/users/user_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
+frogml/_proto/qwak/administration/v1/environments/environment_pb2.py,sha256=CvQOXO8keOc2EJAW2rbP9976pF_dsyupnYyxe0pRkyo,2478
+frogml/_proto/qwak/administration/v1/environments/environment_pb2.pyi,sha256=0sxjj4w8G_AzSsEa7isLjaQPcssJKaxzHxBq5gknCSE,2015
+frogml/_proto/qwak/administration/v1/environments/environment_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
+frogml/_proto/qwak/administration/v1/environments/environment_service_pb2.py,sha256=eEKQjLsLzh8q0aJBOiO8bc7sANQRjdjF_fFifpmKtig,2867
+frogml/_proto/qwak/administration/v1/environments/environment_service_pb2.pyi,sha256=k8bq8klAQyw2Luy-2XKSq6V5b2b0pDaXQlLQAGffGsE,2320
+frogml/_proto/qwak/administration/v1/environments/environment_service_pb2_grpc.py,sha256=TuyAgZpEVHoR7QNjZhUtI6bWBmwxPO5nyVyizem1L-s,3163
 frogml/_proto/qwak/admiral/secret/v0/secret_pb2.py,sha256=U8KCZxLbs_wlrHyUr9ZMqsBz4_47yiqsrnuOMqh6Gj0,5180
 frogml/_proto/qwak/admiral/secret/v0/secret_pb2.pyi,sha256=3bLfqt8Ciki7n0jP_NiNcgtNWliBfrqG_HzsGSQMC3Q,10250
 frogml/_proto/qwak/admiral/secret/v0/secret_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
@@ -665,11 +674,15 @@ frogml/core/clients/administration/authenticated_user/__init__.py,sha256=47DEQpj
 frogml/core/clients/administration/authenticated_user/client.py,sha256=ouN4YxbZs4uYDI8O0DaJZxfUoXgqESrJCDRTlfuNk5s,1486
 frogml/core/clients/administration/authentication/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml/core/clients/administration/authentication/client.py,sha256=BzfCb5laJ3Vci5t3owrcpuVvF4o9bsiKtOFfYWM1Nno,1200
+frogml/core/clients/administration/cluster_v2/__init__.py,sha256=qsxlbOhYg9p5U-0RStC1CIJ86lEZ5UNIWuFPnL-8wGw,36
+frogml/core/clients/administration/cluster_v2/client.py,sha256=m4CTSYjxxSeed118VxCanDQivMLXg5o1jxyDmSeMHUM,4217
 frogml/core/clients/administration/eco_system/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml/core/clients/administration/eco_system/client.py,sha256=b2MYJVlbU4ehhwBWWkCwwl24xVw4K-_t7xMuGSMQ2uo,5379
 frogml/core/clients/administration/eco_system/eco_system_utils.py,sha256=iSyCJv7z65DI2c3qsREzBM_S8ZUGOG4ISjhrfL_t4_Y,521
 frogml/core/clients/administration/environment/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-frogml/core/clients/administration/environment/client.py,sha256=KHUsKFb2gKqpHFTU74vVSLGAPgHufbDS-hB4tbFvyE8,2739
+frogml/core/clients/administration/environment/client.py,sha256=YlfaKvoMXGAvcYj480PtXlAqc8GG6qSGv3KvPqtYh0o,3807
+frogml/core/clients/administration/environment_v1/__init__.py,sha256=15_J34QBVU9X0d9_A6stiwYc6oLwkvuCCviROwROPXo,40
+frogml/core/clients/administration/environment_v1/client.py,sha256=pEvE7dKv_QhYgOi-Aggc6oWAJq-80-Qw_Hsddn60upA,1924
 frogml/core/clients/administration/self_service/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml/core/clients/administration/self_service/client.py,sha256=znZgDkUzSLgtKx1yrUNU72_BiFtPbpCratD73B316bU,2648
 frogml/core/clients/alert_management/__init__.py,sha256=zAfCPJTj6oAMoj3vciLjRy8ULTTHCmmrdISuDCHASvk,43
@@ -1143,9 +1156,12 @@ frogml_services_mock/mocks/batch_job_manager_service.py,sha256=CYkNs1SkuTji0QHdv
 frogml_services_mock/mocks/build_orchestrator_build_api.py,sha256=_9ZmgkAH_yp0yWZlqnMPT9rlj9ctfsMIlIJxG_a2Jc4,5002
 frogml_services_mock/mocks/build_orchestrator_build_settings_api.py,sha256=c5_X7Q4EYP5krrLFdWbnml7_asdgLKB_6_VkwEOR8_A,1218
 frogml_services_mock/mocks/build_orchestrator_service_api.py,sha256=7mZzOahph29ZLGh_qQaE2QqF4Bs5YwTaFj8QJxd4TkA,5345
+frogml_services_mock/mocks/cluster_v2_service.py,sha256=r4A2A4UUIEpyGrBEI9JSMQc1bzpzqBU_Vu22MIKP2do,5690
 frogml_services_mock/mocks/data_versioning_service.py,sha256=8DzV5oxH7DXZAu_ZBiPEwW9m1AwbOlYOO1PFPjkq_Dk,2470
 frogml_services_mock/mocks/deployment_management_service.py,sha256=wofBAw-2NKToTiFYxHqjR3QDrvplpV8NWNZMiIX6U_Q,20583
 frogml_services_mock/mocks/ecosystem_service_api.py,sha256=saJYdT43nGVNyozWeDVc6HIXAsNvCdU5J1i-NNrnOr4,2089
+frogml_services_mock/mocks/environment_v0_service.py,sha256=xKKxKiD4_sqRfyNEpEdqZz-LQ7sJKop80JtvGTReI7s,7522
+frogml_services_mock/mocks/environment_v1_service.py,sha256=_5WPuVTchhopXrZyHSnh_TAprtTgTjt3Jkpprl0cHsk,1404
 frogml_services_mock/mocks/execution_management_service.py,sha256=pRX_DWzayMhwbw7zQVRx2blVpiubqVWpZng2Xn8ZgI0,1205
 frogml_services_mock/mocks/feature_store_data_sources_manager_api.py,sha256=YNd-0abshQH6bfrIEr1z0awIde_SYabfuZHV83LmcJ4,5249
 frogml_services_mock/mocks/feature_store_entities_manager_api.py,sha256=fF3hhJvbd7PDPyghvqtiJudISOR1GasP6MDs5X44Tic,3355
@@ -1155,7 +1171,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=16-x7m3pHl9C51SbuVYwDWmBWv4vzEAwEjfykQaSlfM,6336
+frogml_services_mock/mocks/frogml_mocks.py,sha256=eyJ76XwrRaSrBasdjyLGnlXWAmBXR1lJDJUqMBGLU90,6755
 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
@@ -1177,9 +1193,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=hciz4Q2NvjT84DE08sgxfVVDELfD4hitqgoDrZugcjg,19961
+frogml_services_mock/services_mock.py,sha256=cVydasU1hXNbvz6K_M57GIZJ285i0s2fcMATzH2HH6I,21378
 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.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-1.2.44.dist-info/METADATA,sha256=mVB2P67U5CNBghRdTmbGzuXM5r2Hysi6YgClWlmi6BE,5599
+frogml-1.2.44.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+frogml-1.2.44.dist-info/RECORD,,

frogml_services_mock/mocks/cluster_v2_service.py

@@ -0,0 +1,146 @@
+from typing import Optional
+
+import grpc
+
+from frogml._proto.qwak.administration.cluster.v2.cluster_pb2 import (
+    ClusterState,
+)
+from frogml._proto.qwak.administration.cluster.v2.cluster_service_pb2 import (
+    CreateClusterCreationRequestResponse,
+    DeleteClusterResponse,
+    GetClusterResponse,
+    GetClusterStateResponse,
+    ListClustersResponse,
+    UpdateClusterConfigurationResponse,
+)
+from frogml._proto.qwak.administration.cluster.v2.cluster_service_pb2_grpc import (
+    ClusterServiceServicer,
+)
+
+
+class ClusterV2ServiceMock(ClusterServiceServicer):
+    def __init__(self):
+        super().__init__()
+        self.__create_cluster_response: Optional[
+            CreateClusterCreationRequestResponse
+        ] = None
+        self.__create_cluster_error: Optional[grpc.StatusCode] = None
+
+        self.__get_cluster_state_response: Optional[GetClusterStateResponse] = None
+        self.__get_cluster_state_error: Optional[grpc.StatusCode] = None
+
+        self.__get_cluster_response: Optional[GetClusterResponse] = None
+        self.__get_cluster_error: Optional[grpc.StatusCode] = None
+
+        self.__list_clusters_response: Optional[ListClustersResponse] = None
+        self.__list_clusters_error: Optional[grpc.StatusCode] = None
+
+        self.__update_cluster_response: Optional[UpdateClusterConfigurationResponse] = (
+            None
+        )
+        self.__update_cluster_error: Optional[grpc.StatusCode] = None
+
+        self.__delete_cluster_response: Optional[DeleteClusterResponse] = None
+        self.__delete_cluster_error: Optional[grpc.StatusCode] = None
+
+    def given_create_cluster(
+        self,
+        response: Optional[CreateClusterCreationRequestResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__create_cluster_response = response
+        self.__create_cluster_error = error_code
+
+    def given_get_cluster_state(
+        self,
+        response: Optional[GetClusterStateResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__get_cluster_state_response = response
+        self.__get_cluster_state_error = error_code
+
+    def given_get_cluster(
+        self,
+        response: Optional[GetClusterResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__get_cluster_response = response
+        self.__get_cluster_error = error_code
+
+    def given_list_clusters(
+        self,
+        response: Optional[ListClustersResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__list_clusters_response = response
+        self.__list_clusters_error = error_code
+
+    def given_update_cluster(
+        self,
+        response: Optional[UpdateClusterConfigurationResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__update_cluster_response = response
+        self.__update_cluster_error = error_code
+
+    def given_delete_cluster(
+        self,
+        response: Optional[DeleteClusterResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__delete_cluster_response = response
+        self.__delete_cluster_error = error_code
+
+    def CreateClusterCreationRequest(self, request, context):
+        if self.__create_cluster_error:
+            context.set_code(self.__create_cluster_error)
+            context.set_details("Failed to create cluster")
+            return CreateClusterCreationRequestResponse()
+        if self.__create_cluster_response:
+            return self.__create_cluster_response
+        return CreateClusterCreationRequestResponse(request_id="mock-request-id")
+
+    def GetClusterState(self, request, context):
+        if self.__get_cluster_state_error:
+            context.set_code(self.__get_cluster_state_error)
+            context.set_details("Failed to get cluster state")
+            return GetClusterStateResponse()
+        if self.__get_cluster_state_response:
+            return self.__get_cluster_state_response
+        return GetClusterStateResponse(cluster_state=ClusterState())
+
+    def GetCluster(self, request, context):
+        if self.__get_cluster_error:
+            context.set_code(self.__get_cluster_error)
+            context.set_details("Failed to get cluster")
+            return GetClusterResponse()
+        if self.__get_cluster_response:
+            return self.__get_cluster_response
+        return GetClusterResponse(cluster_state=ClusterState())
+
+    def ListClusters(self, request, context):
+        if self.__list_clusters_error:
+            context.set_code(self.__list_clusters_error)
+            context.set_details("Failed to list clusters")
+            return ListClustersResponse()
+        if self.__list_clusters_response:
+            return self.__list_clusters_response
+        return ListClustersResponse()
+
+    def UpdateClusterConfiguration(self, request, context):
+        if self.__update_cluster_error:
+            context.set_code(self.__update_cluster_error)
+            context.set_details("Failed to update cluster")
+            return UpdateClusterConfigurationResponse()
+        if self.__update_cluster_response:
+            return self.__update_cluster_response
+        return UpdateClusterConfigurationResponse(request_id="mock-request-id")
+
+    def DeleteCluster(self, request, context):
+        if self.__delete_cluster_error:
+            context.set_code(self.__delete_cluster_error)
+            context.set_details("Failed to delete cluster")
+            return DeleteClusterResponse()
+        if self.__delete_cluster_response:
+            return self.__delete_cluster_response
+        return DeleteClusterResponse(request_id="mock-request-id")