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

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/core/inner/di_configuration/dependency_wiring.py

@@ -27,29 +27,30 @@ def wire_dependencies() -> FrogmlContainer:
         )
 
     from frogml.core.clients import (
-        kube_deployment_captain,
-        model_version_manager,
+        administration,
+        alert_management,
+        alerts_registry,
         analytics,
-        system_secret,
+        audience,
+        automation_management,
         autoscaling,
-        build_orchestrator,
         batch_job_management,
-        instance_template,
-        feature_store,
+        build_orchestrator,
+        data_versioning,
         deployment,
-        user_application_instance,
-        jfrog_gateway,
-        alert_management,
+        feature_store,
+        file_versioning,
+        instance_template,
         integration_management,
-        model_management,
-        audience,
-        data_versioning,
+        jfrog_gateway,
+        kube_deployment_captain,
         logging_client,
-        automation_management,
-        file_versioning,
-        alerts_registry,
-        administration,
+        model_deployment_manager,
         model_group_management,
+        model_management,
+        model_version_manager,
+        system_secret,
+        user_application_instance,
     )
 
     container.wire(
@@ -77,6 +78,7 @@ def wire_dependencies() -> FrogmlContainer:
             model_version_manager,
             jfrog_gateway,
             model_group_management,
+            model_deployment_manager,
         ]
     )
 

{frogml-1.2.41.dist-info → frogml-1.2.47.dist-info}/METADATA

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

@@ -1,4 +1,4 @@
-frogml/__init__.py,sha256=Iha2EFEpm8UEhQZulMOPWPSHZKElxWm4XT9dGhtHKck,741
+frogml/__init__.py,sha256=nvwO4HCXHk0dxYeR0cagpgioYfR0AhrIPIVdCPaco5g,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,15 +167,24 @@ 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/account_secret_pb2.py,sha256=Owd4zpaBIh3BytZy3-2ioU8vioULkXzmVV6BD9FoCuo,2543
+frogml/_proto/qwak/admiral/secret/v0/account_secret_pb2.pyi,sha256=ONAvrjhRLkbRAfY-dulJb2S2w0Enm4vq2z7DZRmbq3U,2736
+frogml/_proto/qwak/admiral/secret/v0/account_secret_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
 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
 frogml/_proto/qwak/admiral/secret/v0/secret_service_pb2.py,sha256=V5_l2ipfgimZ-V3BawNz_h2u31e7LiF6f9XBnHZznXk,3425
 frogml/_proto/qwak/admiral/secret/v0/secret_service_pb2.pyi,sha256=LHBpsDVvWBFG-K5_ZQR_698sKKce7qwQqEd40yJwkaA,3343
 frogml/_proto/qwak/admiral/secret/v0/secret_service_pb2_grpc.py,sha256=95auDdnyLijxuVsxlU213MXYF595jjWvz1L5i-HVd1Y,6873
-frogml/_proto/qwak/admiral/secret/v0/system_secret_service_pb2.py,sha256=upoqMkZwa-9yNTwfNxc8DhyqSqLh8yJ30Fa-0uTfJRI,3422
-frogml/_proto/qwak/admiral/secret/v0/system_secret_service_pb2.pyi,sha256=lvlahGJyM8LKj3hwEz4rqoAc3dh54iwlrjBBQVjPDX8,3810
-frogml/_proto/qwak/admiral/secret/v0/system_secret_service_pb2_grpc.py,sha256=dGG6SxGjs4V-M6aKR9G5YPPcds-PNImvqdmY7l1Lp8k,6918
+frogml/_proto/qwak/admiral/secret/v0/system_secret_service_pb2.py,sha256=oXHMyMj9bWjGbMIRnEx5v0pvlArb3HYQwvxDzy5QXnQ,5594
+frogml/_proto/qwak/admiral/secret/v0/system_secret_service_pb2.pyi,sha256=-fyHIVVv5Aa7pbv1EEmzYpYsnwDKFlnEfF5XFQI8VI4,7653
+frogml/_proto/qwak/admiral/secret/v0/system_secret_service_pb2_grpc.py,sha256=Jl8KPDxhIcl3JwTuBi1Q-i9TCVaKI4FR4tCtIi9S19Y,13074
 frogml/_proto/qwak/admiral/user_application_instance/v0/user_application_instance_pb2.py,sha256=nPksL4ROLEgchP1YtL-BSAcmBvb3NcEdsNBxVInxDNI,8028
 frogml/_proto/qwak/admiral/user_application_instance/v0/user_application_instance_pb2.pyi,sha256=TVQAGe8Sil6DyMFJbWOioYhdqRYHp26WouKjMbCgDnk,20646
 frogml/_proto/qwak/admiral/user_application_instance/v0/user_application_instance_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
@@ -665,11 +677,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
@@ -860,7 +876,7 @@ frogml/core/inner/di_configuration/__init__.py,sha256=GQKLaDPpBQ78h0gbwxaW3JkKbN
 frogml/core/inner/di_configuration/account.py,sha256=YbQBmZ-roybEZJ1fmGk9TLQ5CEjLWcP9NuxS6WSSrbI,3016
 frogml/core/inner/di_configuration/config.yml,sha256=GUvaZMWIDIR_d7hFcPVG_kHdCwpERKH1AFDakG3vqI4,242
 frogml/core/inner/di_configuration/containers.py,sha256=8rEMnLSFvlfjEnfvbm9hOizCGWXX9F8S9QEYF5sX48E,1495
-frogml/core/inner/di_configuration/dependency_wiring.py,sha256=bbVfjcDYEiAqKfIf2l60XthmDjYBe8YIwutz8DZcglg,2714
+frogml/core/inner/di_configuration/dependency_wiring.py,sha256=sbgR4wJo08UDO9MQR6ub2lIq8HRzb9QBucvGVRRYQ4Y,2786
 frogml/core/inner/instance_template/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml/core/inner/instance_template/verify_template_id.py,sha256=SSVReUkx9t0FkXaSh0MKXOKdjd_55wE3NjAPOt6DGIM,1952
 frogml/core/inner/model_loggers_utils.py,sha256=mGBZQ1CbiEePaoftFEH_yfLlsxP8fcPcu83Cb8Ut8cM,2465
@@ -1143,9 +1159,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 +1174,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 +1196,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.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-1.2.47.dist-info/METADATA,sha256=q8fKfEVudZ2jPcKn3gO_I07sbIuuT3Dh_Ul-QPXzpcQ,5599
+frogml-1.2.47.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+frogml-1.2.47.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")

frogml_services_mock/mocks/environment_v0_service.py

@@ -0,0 +1,175 @@
+from typing import Optional
+
+import grpc
+
+from frogml._proto.qwak.administration.v0.environments.environment_pb2 import (
+    QwakEnvironment,
+)
+from frogml._proto.qwak.administration.v0.environments.environment_service_pb2 import (
+    CreateEnvironmentResponse,
+    GetEnvironmentApplicationUserCredentialsResponse,
+    GetEnvironmentResponse,
+    ListEnvironmentsResponse,
+    SetEnvironmentApplicationUserCredentialsResponse,
+    UpdateEnvironmentConfigurationResponse,
+    UpdateEnvironmentPersonalizationResponse,
+)
+from frogml._proto.qwak.administration.v0.environments.environment_service_pb2_grpc import (
+    EnvironmentServiceServicer,
+)
+
+
+class EnvironmentV0ServiceMock(EnvironmentServiceServicer):
+    def __init__(self):
+        super().__init__()
+        self.__create_environment_response: Optional[CreateEnvironmentResponse] = None
+        self.__create_environment_error: Optional[grpc.StatusCode] = None
+
+        self.__get_environment_response: Optional[GetEnvironmentResponse] = None
+        self.__get_environment_error: Optional[grpc.StatusCode] = None
+
+        self.__update_environment_configuration_response: Optional[
+            UpdateEnvironmentConfigurationResponse
+        ] = None
+        self.__update_environment_configuration_error: Optional[grpc.StatusCode] = None
+
+        self.__update_environment_personalization_response: Optional[
+            UpdateEnvironmentPersonalizationResponse
+        ] = None
+        self.__update_environment_personalization_error: Optional[grpc.StatusCode] = (
+            None
+        )
+
+        self.__list_environments_response: Optional[ListEnvironmentsResponse] = None
+        self.__list_environments_error: Optional[grpc.StatusCode] = None
+
+        self.__get_credentials_response: Optional[
+            GetEnvironmentApplicationUserCredentialsResponse
+        ] = None
+        self.__get_credentials_error: Optional[grpc.StatusCode] = None
+
+        self.__set_credentials_response: Optional[
+            SetEnvironmentApplicationUserCredentialsResponse
+        ] = None
+        self.__set_credentials_error: Optional[grpc.StatusCode] = None
+
+    def given_create_environment(
+        self,
+        response: Optional[CreateEnvironmentResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__create_environment_response = response
+        self.__create_environment_error = error_code
+
+    def given_get_environment(
+        self,
+        response: Optional[GetEnvironmentResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__get_environment_response = response
+        self.__get_environment_error = error_code
+
+    def given_update_environment_configuration(
+        self,
+        response: Optional[UpdateEnvironmentConfigurationResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__update_environment_configuration_response = response
+        self.__update_environment_configuration_error = error_code
+
+    def given_update_environment_personalization(
+        self,
+        response: Optional[UpdateEnvironmentPersonalizationResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__update_environment_personalization_response = response
+        self.__update_environment_personalization_error = error_code
+
+    def given_list_environments(
+        self,
+        response: Optional[ListEnvironmentsResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__list_environments_response = response
+        self.__list_environments_error = error_code
+
+    def given_get_credentials(
+        self,
+        response: Optional[GetEnvironmentApplicationUserCredentialsResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__get_credentials_response = response
+        self.__get_credentials_error = error_code
+
+    def given_set_credentials(
+        self,
+        response: Optional[SetEnvironmentApplicationUserCredentialsResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__set_credentials_response = response
+        self.__set_credentials_error = error_code
+
+    def CreateEnvironment(self, request, context):
+        if self.__create_environment_error:
+            context.set_code(self.__create_environment_error)
+            context.set_details("Failed to create environment")
+            return CreateEnvironmentResponse()
+        if self.__create_environment_response:
+            return self.__create_environment_response
+        return CreateEnvironmentResponse(environment=QwakEnvironment(id="mock-env-id"))
+
+    def GetEnvironment(self, request, context):
+        if self.__get_environment_error:
+            context.set_code(self.__get_environment_error)
+            context.set_details("Failed to get environment")
+            return GetEnvironmentResponse()
+        if self.__get_environment_response:
+            return self.__get_environment_response
+        return GetEnvironmentResponse(
+            environment=QwakEnvironment(id=request.environment_id)
+        )
+
+    def UpdateEnvironmentConfiguration(self, request, context):
+        if self.__update_environment_configuration_error:
+            context.set_code(self.__update_environment_configuration_error)
+            context.set_details("Failed to update environment configuration")
+            return UpdateEnvironmentConfigurationResponse()
+        if self.__update_environment_configuration_response:
+            return self.__update_environment_configuration_response
+        return UpdateEnvironmentConfigurationResponse()
+
+    def UpdateEnvironmentPersonalization(self, request, context):
+        if self.__update_environment_personalization_error:
+            context.set_code(self.__update_environment_personalization_error)
+            context.set_details("Failed to update environment personalization")
+            return UpdateEnvironmentPersonalizationResponse()
+        if self.__update_environment_personalization_response:
+            return self.__update_environment_personalization_response
+        return UpdateEnvironmentPersonalizationResponse()
+
+    def ListEnvironments(self, request, context):
+        if self.__list_environments_error:
+            context.set_code(self.__list_environments_error)
+            context.set_details("Failed to list environments")
+            return ListEnvironmentsResponse()
+        if self.__list_environments_response:
+            return self.__list_environments_response
+        return ListEnvironmentsResponse()
+
+    def GetEnvironmentApplicationUserCredentials(self, request, context):
+        if self.__get_credentials_error:
+            context.set_code(self.__get_credentials_error)
+            context.set_details("Failed to get credentials")
+            return GetEnvironmentApplicationUserCredentialsResponse()
+        if self.__get_credentials_response:
+            return self.__get_credentials_response
+        return GetEnvironmentApplicationUserCredentialsResponse()
+
+    def SetEnvironmentApplicationUserCredentials(self, request, context):
+        if self.__set_credentials_error:
+            context.set_code(self.__set_credentials_error)
+            context.set_details("Failed to set credentials")
+            return SetEnvironmentApplicationUserCredentialsResponse()
+        if self.__set_credentials_response:
+            return self.__set_credentials_response
+        return SetEnvironmentApplicationUserCredentialsResponse()

frogml_services_mock/mocks/environment_v1_service.py

@@ -0,0 +1,37 @@
+from typing import Optional
+
+import grpc
+
+from frogml._proto.qwak.administration.v0.environments.environment_pb2 import (
+    QwakEnvironment,
+)
+from frogml._proto.qwak.administration.v1.environments.environment_service_pb2 import (
+    CreateEnvironmentResponse,
+)
+from frogml._proto.qwak.administration.v1.environments.environment_service_pb2_grpc import (
+    EnvironmentServiceServicer,
+)
+
+
+class EnvironmentV1ServiceMock(EnvironmentServiceServicer):
+    def __init__(self):
+        super().__init__()
+        self.__create_environment_response: Optional[CreateEnvironmentResponse] = None
+        self.__create_environment_error: Optional[grpc.StatusCode] = None
+
+    def given_create_environment(
+        self,
+        response: Optional[CreateEnvironmentResponse] = None,
+        error_code: Optional[grpc.StatusCode] = None,
+    ):
+        self.__create_environment_response = response
+        self.__create_environment_error = error_code
+
+    def CreateEnvironment(self, request, context):
+        if self.__create_environment_error:
+            context.set_code(self.__create_environment_error)
+            context.set_details("Failed to create environment")
+            return CreateEnvironmentResponse()
+        if self.__create_environment_response:
+            return self.__create_environment_response
+        return CreateEnvironmentResponse(environment=QwakEnvironment(id="mock-env-id"))

frogml_services_mock/mocks/frogml_mocks.py

@@ -23,11 +23,14 @@ from frogml_services_mock.mocks.build_orchestrator_build_settings_api import (
 from frogml_services_mock.mocks.build_orchestrator_service_api import (
     BuildOrchestratorServiceApiMock,
 )
+from frogml_services_mock.mocks.cluster_v2_service import ClusterV2ServiceMock
 from frogml_services_mock.mocks.data_versioning_service import DataVersioningServiceMock
 from frogml_services_mock.mocks.deployment_management_service import (
     DeploymentManagementServiceMock,
 )
 from frogml_services_mock.mocks.ecosystem_service_api import EcoSystemServiceMock
+from frogml_services_mock.mocks.environment_v0_service import EnvironmentV0ServiceMock
+from frogml_services_mock.mocks.environment_v1_service import EnvironmentV1ServiceMock
 from frogml_services_mock.mocks.execution_management_service import (
     ExecutionManagementServiceMock,
 )
@@ -135,3 +138,6 @@ class FrogmlMocks:
     location_discovery_service: LocationDiscoveryServiceApiMock
     jfrog_tenant_info_service: JFrogTenantInfoServiceMock
     model_deployment_manager_mock: ModelDeploymentManagerMock
+    cluster_v2_service_mock: ClusterV2ServiceMock
+    environment_v0_service_mock: EnvironmentV0ServiceMock
+    environment_v1_service_mock: EnvironmentV1ServiceMock