qwak-core 0.1.37__py3-none-any.whl → 0.5.19__py3-none-any.whl

_qwak_proto/qwak/workspace/workspace_service_pb2_grpc.py

@@ -74,6 +74,16 @@ class WorkspaceManagementServiceStub(object):
                 request_serializer=qwak_dot_workspace_dot_workspace__service__pb2.GetDefaultWorkspaceValuesRequest.SerializeToString,
                 response_deserializer=qwak_dot_workspace_dot_workspace__service__pb2.GetDefaultWorkspaceValuesResponse.FromString,
                 )
+        self.MarkBuildAsCopied = channel.unary_unary(
+                '/qwak.workspace.management.WorkspaceManagementService/MarkBuildAsCopied',
+                request_serializer=qwak_dot_workspace_dot_workspace__service__pb2.MarkBuildAsCopiedRequest.SerializeToString,
+                response_deserializer=qwak_dot_workspace_dot_workspace__service__pb2.MarkBuildAsCopiedResponse.FromString,
+                )
+        self.CopyBuild = channel.unary_unary(
+                '/qwak.workspace.management.WorkspaceManagementService/CopyBuild',
+                request_serializer=qwak_dot_workspace_dot_workspace__service__pb2.CopyBuildRequest.SerializeToString,
+                response_deserializer=qwak_dot_workspace_dot_workspace__service__pb2.CopyBuildResponse.FromString,
+                )
 
 
 class WorkspaceManagementServiceServicer(object):
@@ -163,6 +173,18 @@ class WorkspaceManagementServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def MarkBuildAsCopied(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def CopyBuild(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
 
 def add_WorkspaceManagementServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -226,6 +248,16 @@ def add_WorkspaceManagementServiceServicer_to_server(servicer, server):
                     request_deserializer=qwak_dot_workspace_dot_workspace__service__pb2.GetDefaultWorkspaceValuesRequest.FromString,
                     response_serializer=qwak_dot_workspace_dot_workspace__service__pb2.GetDefaultWorkspaceValuesResponse.SerializeToString,
             ),
+            'MarkBuildAsCopied': grpc.unary_unary_rpc_method_handler(
+                    servicer.MarkBuildAsCopied,
+                    request_deserializer=qwak_dot_workspace_dot_workspace__service__pb2.MarkBuildAsCopiedRequest.FromString,
+                    response_serializer=qwak_dot_workspace_dot_workspace__service__pb2.MarkBuildAsCopiedResponse.SerializeToString,
+            ),
+            'CopyBuild': grpc.unary_unary_rpc_method_handler(
+                    servicer.CopyBuild,
+                    request_deserializer=qwak_dot_workspace_dot_workspace__service__pb2.CopyBuildRequest.FromString,
+                    response_serializer=qwak_dot_workspace_dot_workspace__service__pb2.CopyBuildResponse.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'qwak.workspace.management.WorkspaceManagementService', rpc_method_handlers)
@@ -439,3 +471,37 @@ class WorkspaceManagementService(object):
             qwak_dot_workspace_dot_workspace__service__pb2.GetDefaultWorkspaceValuesResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def MarkBuildAsCopied(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.workspace.management.WorkspaceManagementService/MarkBuildAsCopied',
+            qwak_dot_workspace_dot_workspace__service__pb2.MarkBuildAsCopiedRequest.SerializeToString,
+            qwak_dot_workspace_dot_workspace__service__pb2.MarkBuildAsCopiedResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def CopyBuild(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.workspace.management.WorkspaceManagementService/CopyBuild',
+            qwak_dot_workspace_dot_workspace__service__pb2.CopyBuildRequest.SerializeToString,
+            qwak_dot_workspace_dot_workspace__service__pb2.CopyBuildResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

qwak/__init__.py

@@ -1,8 +1,9 @@
 """Top-level package for qwak-core."""
 
 __author__ = "Qwak.ai"
-__version__ = "0.1.37"
+__version__ = "0.5.19"
 
+from qwak.inner.di_configuration import wire_dependencies
 from qwak.model.experiment_tracking import log_metric, log_param
 from qwak.model_loggers.artifact_logger import load_file, log_file
 from qwak.model_loggers.data_logger import load_data, log_data
@@ -12,6 +13,4 @@ from .model.decorators.api import api_decorator as api
 from .model.decorators.timer import qwak_timer
 from .qwak_client.client import QwakClient
 
-from qwak.inner.di_configuration import wire_dependencies
-
 _container = wire_dependencies()

qwak/automations/automation_executions.py

@@ -61,16 +61,20 @@ class ExecutionRunDetails:
     @staticmethod
     def from_proto(message: ExecutionRunDetailsMessage):
         return ExecutionRunDetails(
-            start_time=datetime.fromtimestamp(
-                message.start_time.seconds + message.start_time.nanos / 1e9
-            )
-            if message.start_time
-            else None,
-            end_time=datetime.fromtimestamp(
-                message.end_time.seconds + message.end_time.nanos / 1e9
-            )
-            if message.end_time
-            else None,
+            start_time=(
+                datetime.fromtimestamp(
+                    message.start_time.seconds + message.start_time.nanos / 1e9
+                )
+                if message.start_time
+                else None
+            ),
+            end_time=(
+                datetime.fromtimestamp(
+                    message.end_time.seconds + message.end_time.nanos / 1e9
+                )
+                if message.end_time
+                else None
+            ),
             status=proto_status_to_execution_status_map.get(message.status),
             task=message.task,
             error_details=message.error_details,

qwak/automations/automations.py

@@ -111,9 +111,11 @@ class MetricBasedTrigger(Trigger):
         )
         return MetricBasedTrigger(
             name=message.metric_based_trigger.name,
-            metric=metric.from_proto(message.metric_based_trigger.metric)
-            if metric
-            else None,
+            metric=(
+                metric.from_proto(message.metric_based_trigger.metric)
+                if metric
+                else None
+            ),
             threshold=message.metric_based_trigger.threshold,
             direction=map_proto_threshold_to_direction(
                 message.metric_based_trigger.threshold_direction
@@ -278,6 +280,8 @@ class Automation:
     create_audit: AutomationAudit = field(default_factory=AutomationAudit)
     on_error: Notification = field(default=None)
     on_success: Notification = field(default=None)
+    jfrog_token_id: str = field(default="")
+    platform_url: str = field(default="")
 
     def to_proto(self):
         return AutomationProto(
@@ -290,16 +294,18 @@ class Automation:
                 model_id=self.model_id,
                 action=self.action.to_proto(),
                 trigger=self.trigger.to_proto() if self.trigger is not None else None,
-                on_error=self.on_error.to_proto()
-                if self.on_error is not None
-                else None,
-                on_success=self.on_success.to_proto()
-                if self.on_success is not None
-                else None,
+                on_error=(
+                    self.on_error.to_proto() if self.on_error is not None else None
+                ),
+                on_success=(
+                    self.on_success.to_proto() if self.on_success is not None else None
+                ),
             ),
             qwak_environment_id=self.environment,
             create_audit=self.create_audit.to_proto(),
             is_deleted=self.is_deleted,
+            jfrog_token_id=self.jfrog_token_id,
+            platform_url=self.platform_url,
         )
 
     @staticmethod
@@ -326,22 +332,26 @@ class Automation:
             is_enabled=message.automation_spec.is_enabled,
             is_sdk_v1=message.automation_spec.is_sdk_v1,
             is_deleted=message.is_deleted,
-            action=action.from_proto(message.automation_spec.action)
-            if action
-            else None,
-            trigger=trigger.from_proto(message.automation_spec.trigger)
-            if trigger
-            else None,
+            action=(
+                action.from_proto(message.automation_spec.action) if action else None
+            ),
+            trigger=(
+                trigger.from_proto(message.automation_spec.trigger) if trigger else None
+            ),
             environment=message.qwak_environment_id,
             create_audit=AutomationAudit.from_proto(message.create_audit),
-            on_error=on_error_notification.from_proto(message.automation_spec.on_error)
-            if on_error_notification
-            else None,
-            on_success=on_success_notification.from_proto(
-                message.automation_spec.on_success
-            )
-            if on_success_notification
-            else None,
+            on_error=(
+                on_error_notification.from_proto(message.automation_spec.on_error)
+                if on_error_notification
+                else None
+            ),
+            on_success=(
+                on_success_notification.from_proto(message.automation_spec.on_success)
+                if on_success_notification
+                else None
+            ),
+            jfrog_token_id=message.jfrog_token_id,
+            platform_url=message.platform_url,
         )
 
     def __str__(self):

qwak/automations/build_and_deploy_action.py

@@ -10,12 +10,14 @@ from _qwak_proto.qwak.automation.v1.action_pb2 import (
     BuildMetricCondition,
     BuildSpec,
     CpuResources as CpuResourcesProto,
+    DeployedBuildMetricCondition as DeployedBuildMetricConditionProto,
     DeploymentCondition as DeploymentConditionProto,
     DeploymentSize,
     DeploymentSpec,
     GitModelSource,
     GpuResources as GpuResourceProto,
     GpuType,
+    PurchaseOption as PurchaseOptionProto,
     Resources as ResourceProto,
 )
 from _qwak_proto.qwak.automation.v1.auto_scaling_pb2 import (
@@ -30,6 +32,9 @@ from _qwak_proto.qwak.automation.v1.auto_scaling_pb2 import (
     METRIC_TYPE_CPU,
     METRIC_TYPE_LATENCY,
     METRIC_TYPE_MEMORY,
+    METRIC_TYPE_ERROR_RATE,
+    METRIC_TYPE_THROUGHPUT,
+    METRIC_TYPE_GPU,
     AutoScalingConfig as AutoScalingConfigProto,
     AutoScalingPrometheusTrigger as AutoScalingPrometheusTriggerProto,
     QuerySpec,
@@ -43,6 +48,7 @@ from _qwak_proto.qwak.user_application.common.v0.resources_pb2 import (
     GpuType as CommonGpuTypeProto,
     PodComputeResourceTemplateSpec,
 )
+from google.protobuf.wrappers_pb2 import BoolValue
 from qwak.automations.common import (
     Action,
     ThresholdDirection,
@@ -84,6 +90,35 @@ class AutoScaleTrigger(ABC):
         pass
 
 
+@dataclass
+class DeployedBuildMetric(DeploymentCondition):
+    metric_name: str = field(default="")
+    direction: ThresholdDirection = field(default=ThresholdDirection.ABOVE)
+    variation: str = field(default="default")
+
+    def to_proto(self):
+        return DeploymentConditionProto(
+            deployed_build_metric=DeployedBuildMetricConditionProto(
+                metric_name=self.metric_name,
+                threshold_direction=map_threshold_direction_to_proto(self.direction),
+                variation=self.variation,
+            )
+        )
+
+    @staticmethod
+    def from_proto(message: DeploymentConditionProto):
+        return DeployedBuildMetric(
+            metric_name=message.deployed_build_metric.metric_name,
+            direction=map_proto_threshold_to_direction(
+                message.deployed_build_metric.threshold_direction
+            ),
+            variation=message.deployed_build_metric.variation,
+        )
+
+    def __str__(self):
+        return f"Metric Name: \tDirection: {self.direction.name}"
+
+
 @dataclass
 class BuildMetric(DeploymentCondition):
     metric_name: str = field(default="")
@@ -275,6 +310,12 @@ class BuildSpecifications:
     resources: Resources = field(default_factory=CpuResources)
     dependency_file_path: str = field(default=None)
     env_vars: List[str] = field(default_factory=list)
+    git_ssh_token_secret: str = field(default=None)
+    gpu_compatible: bool = field(default=False)
+    push_image: bool = field(default=True)
+    purchase_option: str = field(default="spot")
+    provision_instance_timeout: int = field(default=120)
+    service_account_key_secret_name: str = field(default="")
 
     def to_proto(self):
         return BuildSpec(
@@ -283,6 +324,7 @@ class BuildSpecifications:
                 git_uri=self.git_uri,
                 git_credentials_secret_name=self.git_access_token_secret,
                 git_branch=self.git_branch,
+                git_ssh_secret_name=self.git_ssh_token_secret,
             ),
             main_dir=self.main_dir,
             tags=self.tags,
@@ -291,13 +333,33 @@ class BuildSpecifications:
             assumed_iam_role=self.assumed_iam_role,
             dependency_file_path=self.dependency_file_path,
             env_vars=self.env_vars,
-        )
+            gpu_compatible=self.gpu_compatible,
+            push_image=BoolValue(value=self.push_image),
+            purchase_option=self.__purchase_option_to_proto(),
+            provision_instance_timeout=self.provision_instance_timeout,
+            service_account_key_secret_name=self.service_account_key_secret_name,
+        )
+
+    def __purchase_option_to_proto(self):
+        purchase_option = self.purchase_option.lower()
+        if purchase_option == "spot":
+            return PurchaseOptionProto.SPOT
+        elif purchase_option == "ondemand":
+            return PurchaseOptionProto.ON_DEMAND
+        else:
+            raise ValueError(
+                f"Invalid purchase option {self.purchase_option}. Purchase options can be either spot or ondemand"
+            )
 
     @staticmethod
     def from_proto(build_spec: BuildSpec):
         resources = map_resources_name_to_class(
             build_spec.resource.WhichOneof("resource")
         )
+        push_image = True
+        if build_spec.HasField("push_image"):
+            push_image = build_spec.push_image.value
+
         return BuildSpecifications(
             parameters=build_spec.parameters,
             git_uri=build_spec.git_model_source.git_uri,
@@ -310,8 +372,27 @@ class BuildSpecifications:
             dependency_file_path=build_spec.dependency_file_path,
             assumed_iam_role=build_spec.assumed_iam_role,
             env_vars=build_spec.env_vars,
+            git_ssh_token_secret=build_spec.git_model_source.git_ssh_secret_name,
+            gpu_compatible=build_spec.gpu_compatible,
+            push_image=push_image,
+            purchase_option=BuildSpecifications.__purchase_option_from_proto(
+                build_spec.purchase_option
+            ),
+            service_account_key_secret_name=build_spec.service_account_key_secret_name,
+            provision_instance_timeout=build_spec.provision_instance_timeout,
         )
 
+    @staticmethod
+    def __purchase_option_from_proto(purchase_option: PurchaseOptionProto):
+        if purchase_option == PurchaseOptionProto.SPOT:
+            return "spot"
+        elif purchase_option == PurchaseOptionProto.ON_DEMAND:
+            return "ondemand"
+        else:
+            raise ValueError(
+                f"Invalid purchase option {purchase_option}. Purchase options can be either spot or ondemand"
+            )
+
     def __str__(self):
         result = f"Git Uri:\t{self.git_uri}\n"
         result += f"Git Branch:\t{self.git_branch}\n"
@@ -320,12 +401,25 @@ class BuildSpecifications:
             if self.git_access_token_secret
             else ""
         )
+        result += (
+            f"Git Ssh Secret:\t{self.git_ssh_token_secret}\n"
+            if self.git_ssh_token_secret
+            else ""
+        )
         result += f"Main Dir:\t{self.main_dir}\n" if self.main_dir != "main" else ""
+        result += f"Gpu Compatible:\t{self.gpu_compatible}\n"
+        result += f"Purchase Option:\t{self.purchase_option}\n"
+        result += f"Provision Instance Timeout:\t{self.provision_instance_timeout}\n"
         result += f"Parameters:\t{self.parameters}\n" if self.parameters else ""
         result += f"Tags:\t{self.tags}\n" if self.tags else ""
         result += (
             f"IAM Role:\t{self.assumed_iam_role}\n" if self.assumed_iam_role else ""
         )
+        result += (
+            f"Service Account Key Secret Name:\t{self.service_account_key_secret_name}\n"
+            if self.service_account_key_secret_name
+            else ""
+        )
         result += f"Base Image:\t{self.base_image}\n" if self.base_image else ""
         result += f"Resources:\n{self.resources}\n" if self.resources else ""
         result += f"Environment Variables:\n{self.env_vars}\n" if self.env_vars else ""
@@ -336,6 +430,9 @@ class MetricType(Enum):
     cpu = METRIC_TYPE_CPU
     latency = METRIC_TYPE_LATENCY
     memory = METRIC_TYPE_MEMORY
+    error_rate = METRIC_TYPE_ERROR_RATE
+    throughput = METRIC_TYPE_THROUGHPUT
+    gpu = METRIC_TYPE_GPU
 
 
 class AggregationType(Enum):
@@ -409,6 +506,7 @@ class DeploymentSpecifications:
     env_vars: List[str] = field(default_factory=list)
     deployment_timeout: int = field(default=0)
     instance: str = field(default="")
+    service_account_key_secret_name: str = field(default="")
 
     def to_proto(self):
         deployment_size = self.__get_deployment_size_proto()
@@ -423,6 +521,7 @@ class DeploymentSpecifications:
                 custom_iam_role_arn=self.custom_iam_role_arn,
                 auto_scaling_config=self.auto_scale_config.to_proto(),
                 deployment_process_timeout_limit=self.deployment_timeout,
+                service_account_key_secret_name=self.service_account_key_secret_name,
             ),
             environments=self.environments,
             env_vars=self.env_vars,
@@ -464,6 +563,11 @@ class DeploymentSpecifications:
             if self.custom_iam_role_arn
             else ""
         )
+        result += (
+            f"Custom Service Account Key Secret Name: {self.service_account_key_secret_name}\n"
+            if self.service_account_key_secret_name
+            else ""
+        )
         result += f"Auto scaling config: {self.auto_scale_config}"
         return result
 
@@ -507,9 +611,9 @@ class DeploymentSpecifications:
             cpu=self.cpu_fraction,
             memory_units=map_memory_units(memory=self.memory),
             memory_amount=get_memory_amount(self.memory),
-            gpu_resources=self.gpu_resources.to_gpu_proto()
-            if self.gpu_resources
-            else None,
+            gpu_resources=(
+                self.gpu_resources.to_gpu_proto() if self.gpu_resources else None
+            ),
         )
 
     @staticmethod
@@ -529,24 +633,29 @@ class DeploymentSpecifications:
             + map_common_memory_units_proto(
                 message.deployment_size.client_pod_compute_resources.cpu_resources.memory_units
             ),
-            gpu_resources=GpuResources.from_common_gpu_proto(
-                message.deployment_size.client_pod_compute_resources.gpu_resources
-            )
-            if (
-                message.deployment_size.client_pod_compute_resources.gpu_resources
-                and message.deployment_size.client_pod_compute_resources.gpu_resources.gpu_amount
-                > 0
-            )
-            else None,
-            instance=message.deployment_size.client_pod_compute_resources.template_spec.template_id
-            if message.deployment_size.client_pod_compute_resources.template_spec.template_id
-            else "",
+            gpu_resources=(
+                GpuResources.from_common_gpu_proto(
+                    message.deployment_size.client_pod_compute_resources.gpu_resources
+                )
+                if (
+                    message.deployment_size.client_pod_compute_resources.gpu_resources
+                    and message.deployment_size.client_pod_compute_resources.gpu_resources.gpu_amount
+                    > 0
+                )
+                else None
+            ),
+            instance=(
+                message.deployment_size.client_pod_compute_resources.template_spec.template_id
+                if message.deployment_size.client_pod_compute_resources.template_spec.template_id
+                else ""
+            ),
             auto_scale_config=AutoScalingConfig.from_proto(
                 message.advanced_options.auto_scaling_config
             ),
             environments=message.environments,
             env_vars=message.env_vars,
             deployment_timeout=message.advanced_options.deployment_process_timeout_limit,
+            service_account_key_secret_name=message.advanced_options.service_account_key_secret_name,
         )
 
     @staticmethod
@@ -562,17 +671,18 @@ class DeploymentSpecifications:
             cpu_fraction=message.deployment_size.cpu,
             memory=str(message.deployment_size.memory_amount)
             + map_memory_units_proto(message.deployment_size.memory_units),
-            gpu_resources=GpuResources.from_gpu_proto(
-                message.deployment_size.gpu_resources
-            )
-            if message.deployment_size.gpu_resources
-            else None,
+            gpu_resources=(
+                GpuResources.from_gpu_proto(message.deployment_size.gpu_resources)
+                if message.deployment_size.gpu_resources
+                else None
+            ),
             auto_scale_config=AutoScalingConfig.from_proto(
                 message.advanced_options.auto_scaling_config
             ),
             environments=message.environments,
             env_vars=message.env_vars,
             deployment_timeout=message.advanced_options.deployment_process_timeout_limit,
+            service_account_key_secret_name=message.advanced_options.service_account_key_secret_name,
         )
 
 
@@ -601,11 +711,13 @@ class QwakBuildDeploy(Action):
             deployment_spec=DeploymentSpecifications.from_proto(
                 message.build_deploy.deployment_spec
             ),
-            deployment_condition=deployment_condition.from_proto(
-                message.build_deploy.deployment_condition
-            )
-            if deployment_condition
-            else None,
+            deployment_condition=(
+                deployment_condition.from_proto(
+                    message.build_deploy.deployment_condition
+                )
+                if deployment_condition
+                else None
+            ),
         )
 
     def __str__(self):
@@ -619,10 +731,11 @@ class AutoScaleQuerySpec:
     time_period: int = field(default=None)
 
     def to_proto(self):
+        aggregation_type_lower: str = self.aggregation_type.lower()
         return QuerySpec(
             time_period=self.time_period,
             metric_type=MetricType[self.metric_type.lower()].value,
-            aggregation_type=AggregationType[self.aggregation_type.lower()].value,
+            aggregation_type=AggregationType[aggregation_type_lower].value,
         )
 
     @staticmethod
@@ -674,7 +787,10 @@ def map_resources_name_to_class(resource_name: str):
 
 
 def map_deployment_condition_name_to_class(deployment_condition_name: str):
-    mapping = {"build_metric": BuildMetric}
+    mapping = {
+        "build_metric": BuildMetric,
+        "deployed_build_metric": DeployedBuildMetric,
+    }
     return mapping.get(deployment_condition_name)
 
 
@@ -683,6 +799,9 @@ def map_auto_scaling_metric_type_proto_to_name(metric_type):
         METRIC_TYPE_CPU: "cpu",
         METRIC_TYPE_LATENCY: "latency",
         METRIC_TYPE_MEMORY: "memory",
+        METRIC_TYPE_ERROR_RATE: "error_rate",
+        METRIC_TYPE_THROUGHPUT: "throughput",
+        METRIC_TYPE_GPU: "gpu",
     }
     return mapping.get(metric_type)
 

qwak/automations/common.py

@@ -75,11 +75,17 @@ def get_memory_amount(memory: str):
 class ThresholdDirection(Enum):
     ABOVE = 1
     BELOW = 2
+    EQUALS = 3
+    BELOW_OR_EQUALS = 4
+    ABOVE_OR_EQUALS = 5
 
 
 threshold_to_proto_mapping = {
     ThresholdDirection.ABOVE: MetricThresholdDirection.ABOVE,
     ThresholdDirection.BELOW: MetricThresholdDirection.BELOW,
+    ThresholdDirection.EQUALS: MetricThresholdDirection.EQUALS,
+    ThresholdDirection.BELOW_OR_EQUALS: MetricThresholdDirection.BELOW_OR_EQUALS,
+    ThresholdDirection.ABOVE_OR_EQUALS: MetricThresholdDirection.ABOVE_OR_EQUALS,
 }
 
 proto_threshold_to_threshold = {v: k for k, v in threshold_to_proto_mapping.items()}

qwak/clients/administration/eco_system/client.py

@@ -5,10 +5,10 @@ from _qwak_proto.qwak.ecosystem.v0.ecosystem_pb2 import (
     EnvironmentDetails,
     UserContextAccountDetails,
     UserContextEnvironmentDetails,
+    AuthenticatedUserContext,
 )
 from _qwak_proto.qwak.ecosystem.v0.ecosystem_runtime_service_pb2 import (
     GetAuthenticatedUserContextRequest,
-    GetAuthenticatedUserContextResponse,
     GetCloudCredentialsRequest,
     GetCloudCredentialsResponse,
 )
@@ -42,7 +42,7 @@ class EcosystemClient:
                 f"Failed to get cloud credentials, error is {e.details()}"
             )
 
-    def get_authenticated_user_context(self) -> GetAuthenticatedUserContextResponse:
+    def get_authenticated_user_context(self) -> AuthenticatedUserContext:
         try:
             return self._ecosystem_service.GetAuthenticatedUserContext(
                 GetAuthenticatedUserContextRequest()

qwak/clients/administration/eco_system/eco_system_utils.py

@@ -0,0 +1,15 @@
+from qwak.clients.administration.eco_system.client import EcosystemClient
+
+
+class EcosystemUtils:
+    def __init__(self):
+        self._client = EcosystemClient()
+
+    def get_user_context(self):
+        return self._client.get_authenticated_user_context().user
+
+    def get_default_environment_id(self) -> str:
+        return self.get_user_context().account_details.default_environment_id
+
+    def get_current_environment_name(self) -> str:
+        return self.get_user_context().environment_details.name.lower()

qwak/clients/administration/environment/client.py

@@ -7,6 +7,8 @@ from _qwak_proto.qwak.administration.v0.environments.environment_service_pb2 imp
     GetEnvironmentApplicationUserCredentialsResponse,
     ListEnvironmentsRequest,
     ListEnvironmentsResponse,
+    GetEnvironmentResponse,
+    GetEnvironmentRequest,
 )
 from _qwak_proto.qwak.administration.v0.environments.environment_service_pb2_grpc import (
     EnvironmentServiceStub,
@@ -74,3 +76,17 @@ class EnvironmentClient:
             raise QwakException(
                 f"Failed to get application user, error is {e.details()}"
             )
+
+    def get_environment(self, environment_id: str) -> GetEnvironmentResponse:
+        """
+        Get Environment by id
+        :param environment_id:
+        Return: GetEnvironmentResponse
+        """
+        request = GetEnvironmentRequest(environment_id=environment_id)
+        try:
+            return self._environment_service.GetEnvironment(request)
+        except grpc.RpcError as e:
+            raise QwakException(
+                f"Failed to get environment {environment_id}, error is {e}"
+            )