qwak-sdk 0.5.62__py3-none-any.whl → 0.5.64__py3-none-any.whl

qwak_sdk/__init__.py

@@ -1,6 +1,6 @@
 # fmt: off
 __author__ = '''Qwak.ai'''
-__version__ = '0.5.62'
+__version__ = '0.5.64'
 
 from qwak.inner import wire_dependencies
 

qwak_sdk/commands/models/build/ui.py

@@ -110,6 +110,11 @@ from qwak_sdk.inner.tools.config_handler import config_handler
     type=str,
     help="[REMOTE BUILD] Custom IAM Role ARN.",
 )
+@click.option(
+    "--service-account-key-secret-name",
+    type=str,
+    help="Custom service account for Gcp.",
+)
 @click.option(
     "--cache/--no-cache",
     default=None,

qwak_sdk/commands/models/deployments/deploy/_logic/deploy_config.py

@@ -246,7 +246,7 @@ class DeployConfig(YamlConfigMixin, QwakConfigBase):
     @dataclass
     class AdvancedOptions:
         iam_role_arn: str = field(default=None)
-        purchase_option: str = field(default=PurchaseOption.SPOT.value)
+        purchase_option: str = field(default="")
         service_account_key_secret_name: str = field(default=None)
 
     model_id: str = field(default="")

qwak_sdk/commands/models/deployments/deploy/_logic/deployment_message_helpers.py

@@ -11,7 +11,6 @@ from _qwak_proto.qwak.deployment.deployment_pb2 import (
     ServingStrategy,
 )
 from qwak.clients.administration.eco_system.client import EcosystemClient
-from qwak.clients.deployment.client import DeploymentManagementClient
 from qwak.clients.instance_template.client import InstanceTemplateManagementClient
 from qwak.exceptions import QwakException
 
@@ -40,12 +39,11 @@ UNKNOWN_SERVING_STRATEGY = (
 def get_env_to_deployment_message(
     deploy_config: DeployConfig,
     kube_deployment_type: KubeDeploymentType,
-    deployment_client: DeploymentManagementClient,
     ecosystem_client: EcosystemClient,
     instance_template_client: InstanceTemplateManagementClient,
 ) -> Dict[str, EnvironmentDeploymentMessage]:
     deployment_size = deployment_size_from_deploy_config(
-        deploy_config, instance_template_client
+        deploy_config, instance_template_client, ecosystem_client
     )
     advanced_deployment_options = get_advanced_deployment_options_from_deploy_config(
         deploy_config, kube_deployment_type
@@ -60,13 +58,8 @@ def get_env_to_deployment_message(
     )
 
     if kube_deployment_type == KubeDeploymentType.ONLINE:
-        model_traffic = deployment_client.get_model_traffic_config(
-            deploy_config.model_id
-        )
-
         env_to_serving_strategy = create_realtime_serving_strategy_from_deploy_config(
             deploy_config,
-            model_traffic,
             environment_name_to_config,
         )
         environment_variables = dictify_params(deploy_config.realtime.env_vars)

qwak_sdk/commands/models/deployments/deploy/_logic/deployment_size_mapper.py

@@ -7,8 +7,11 @@ from _qwak_proto.qwak.user_application.common.v0.resources_pb2 import (
     GpuResources,
     PodComputeResourceTemplateSpec,
 )
+from qwak.clients.administration.eco_system.client import EcosystemClient
 from qwak.clients.instance_template.client import InstanceTemplateManagementClient
+from qwak.exceptions import QwakException
 from qwak.inner.instance_template.verify_template_id import verify_template_id
+from qwak.inner.provider import Provider
 
 from qwak_sdk.commands.models.deployments.deploy._logic.deploy_config import (
     DeployConfig,
@@ -18,14 +21,37 @@ from qwak_sdk.commands.models.deployments.deploy._logic.deploy_config import (
 def deployment_size_from_deploy_config(
     deploy_config: DeployConfig,
     instance_template_client: InstanceTemplateManagementClient,
+    ecosystem_client: EcosystemClient,
 ) -> DeploymentSize:
     if deploy_config.resources.instance_size:
         deploy_config.resources.instance_size = (
             deploy_config.resources.instance_size.lower()
         )
-        verify_template_id(
-            deploy_config.resources.instance_size, instance_template_client
-        )
+        account_details = ecosystem_client.get_account_details()
+        if deploy_config.environments:
+            environments_config = list(
+                ecosystem_client.get_environments_names_to_details(
+                    deploy_config.environments
+                ).values()
+            )
+        else:
+            environments_config = [
+                account_details.environment_by_id[
+                    account_details.default_environment_id
+                ]
+            ]
+        for environment_config in environments_config:
+            provider = __get_provider(environment_config)
+            try:
+                verify_template_id(
+                    deploy_config.resources.instance_size,
+                    instance_template_client,
+                    provider=provider,
+                )
+            except QwakException as e:
+                raise QwakException(
+                    f"Error with template {deploy_config.resources.instance_size} for environment {environment_config.name}: {e.message}"
+                )
 
         return DeploymentSize(
             number_of_pods=deploy_config.resources.pods,
@@ -56,3 +82,15 @@ def deployment_size_from_deploy_config(
                 )
             ),
         )
+
+
+def __get_provider(environment_config):
+    provider = None
+    cloud_type = environment_config.configuration.cloud_configuration.WhichOneof(
+        "configuration"
+    )
+    if cloud_type == "aws_cloud_configuration":
+        provider = Provider.AWS
+    elif cloud_type == "gcp_cloud_configuration":
+        provider = Provider.GCP
+    return provider

qwak_sdk/commands/models/deployments/deploy/_logic/local_deployment.py

@@ -0,0 +1,193 @@
+import base64
+
+from _qwak_proto.qwak.ecosystem.v0.ecosystem_runtime_service_pb2 import (
+    GetCloudCredentialsParameters,
+    GetCloudCredentialsRequest,
+    PermissionSet,
+    PullModelsContainerRegistry,
+)
+from google.protobuf.duration_pb2 import Duration
+from qwak.clients.administration.eco_system.client import EcosystemClient
+from qwak.clients.build_orchestrator import BuildOrchestratorClient
+from qwak.exceptions import QwakException
+from rich.console import Console
+from rich.layout import Layout
+from rich.panel import Panel
+from rich.progress import Progress
+
+from qwak_sdk.commands.models.deployments.deploy._logic.deploy_config import (
+    DeployConfig,
+)
+
+tasks = {}
+
+
+def local_deploy(config: DeployConfig):
+    try:
+        import docker
+    except ImportError:
+        raise QwakException(
+            "Error: 'docker' package is required to for local model deployment."
+        )
+
+    console = Console()
+    console.print(Panel(f"Deploying model {config.model_id} locally", style="green"))
+
+    client = docker.from_env()
+    build_image = _get_build_image(config)
+
+    if not client.images.list(name=build_image):
+        console.print(
+            "Pulling serving image from Qwak's remote repository. Might take a few minutes...",
+            style="yellow",
+        )
+        _docker_login(client)
+        _image_pull(client, build_image)
+
+    try:
+        container = start_local_qwak_container(build_image, client, config)
+        container.reload()
+        host_port = container.ports.get("5000/tcp")[0]["HostPort"]
+
+        layout = example_usage_layout(host_port)
+        console.print(layout)
+
+        for line in container.logs(stream=True):
+            console.print(str(line.strip().decode()), style="blue")
+
+    except KeyboardInterrupt:
+        console.print(Panel("Stopping container...", style="red"))
+        container.stop()
+        container.remove()
+
+
+def example_usage_layout(host_port: str):
+    layout = Layout(size=6)
+    layout.split_row(
+        Layout(name="left"),
+        Layout(name="right"),
+    )
+    layout["left"].update(
+        Panel(
+            f"[bold]cURL usage:[bold]\n\n"
+            f"curl --location\n"
+            f"--request POST http://localhost:{host_port}/predict\n"
+            f"--header 'Content-Type: application/json'\n"
+            f"--data '...'",
+            style="green",
+        )
+    )
+    layout["right"].update(
+        Panel(
+            f"[bold]Python client usage:[bold]\n\n"
+            f"from qwak_inference import RealtimeClient\n\nfeature_vector=[...]\n"
+            f"client = RealTimeClient(model_api='http://localhost:{host_port}/predict')\n"
+            f"client.predict(feature_vector)",
+            style="green",
+        )
+    )
+    return layout
+
+
+def start_local_qwak_container(build_image: str, docker_client, config: DeployConfig):
+    return docker_client.containers.run(
+        build_image,
+        command=["bentoml", "serve-gunicorn", "./"],
+        entrypoint="./docker-entrypoint.sh",
+        environment=[
+            "BENTOML_GUNICORN_WORKERS=2",
+            "QWAK_DEBUG_MODE=True",
+            f"QWAK_MODEL_ID={config.model_id}",
+            "BENTOML_DO_NOT_TRACK=True",
+            "PYTHONPATH=/qwak/model_dir:/qwak/model_dir/main:$PYTHONPATH",
+            f"QWAK_BUILD_ID={config.build_id}",
+            "BUNDLE_PATH=/home/bentoml/bundle",
+            "BENTOML_HOME=/home/bentoml/",
+        ],
+        stream=True,
+        detach=True,
+        publish_all_ports=True,
+    )
+
+
+def _get_build_image(config: DeployConfig):
+    return (
+        BuildOrchestratorClient().get_build(config.build_id).build.build_destined_image
+    )
+
+
+def _show_progress(line, progress):
+    if line["status"] == "Downloading":
+        idx = f'[red][Download {line["id"]}]'
+    elif line["status"] == "Extracting":
+        idx = f'[green][Extract {line["id"]}]'
+    else:
+        # skip other statuses
+        return
+
+    if idx not in tasks.keys():
+        tasks[idx] = progress.add_task(f"{idx}", total=line["progressDetail"]["total"])
+    else:
+        progress.update(tasks[idx], completed=line["progressDetail"]["current"])
+
+
+def _image_pull(docker_client, image_name: str):
+    with Progress() as progress:
+        resp = docker_client.api.pull(image_name, stream=True, decode=True)
+        for line in resp:
+            _show_progress(line, progress)
+
+
+def _docker_login(docker_client):
+    try:
+        import boto3
+    except ImportError:
+        raise QwakException(
+            "Error: The 'boto3' package is necessary for local model deployment. Install it directly or by running"
+            " 'pip install qwak[local-deployment]'"
+        )
+
+    # TODO: block if not an AWS based container registry
+    credentials = _get_aws_credentials()
+
+    aws_credentials = credentials.cloud_credentials.aws_temporary_credentials
+    ecr_client = boto3.Session(
+        aws_access_key_id=aws_credentials.access_key_id,
+        aws_secret_access_key=aws_credentials.secret_access_key,
+        aws_session_token=aws_credentials.session_token,
+        region_name=aws_credentials.region,
+    ).client("ecr")
+
+    ecr_credentials = ecr_client.get_authorization_token()["authorizationData"][0]
+
+    try:
+        docker_client.login(
+            username="AWS",
+            registry=ecr_credentials["proxyEndpoint"],
+            password=base64.b64decode(ecr_credentials["authorizationToken"])
+            .replace(b"AWS:", b"")
+            .decode("utf-8"),
+        )
+
+    except Exception as e:
+        raise QwakException(f"Failed to login to Qwak's container registry: {e}")
+
+
+def _get_aws_credentials():
+    try:
+        eco_client = EcosystemClient()
+        credentials = eco_client.get_cloud_credentials(
+            request=GetCloudCredentialsRequest(
+                parameters=GetCloudCredentialsParameters(
+                    duration=Duration(seconds=60 * 60, nanos=0),  # 6 hours
+                    permission_set=PermissionSet(
+                        pull_models_container_registry=PullModelsContainerRegistry()
+                    ),
+                )
+            )
+        )
+        return credentials
+    except Exception as e:
+        raise QwakException(
+            f"Failed to get credentials to pull image from Qwak's container registry: {e}."
+        )

qwak_sdk/commands/models/deployments/deploy/batch/_logic/deploy_executor.py

@@ -14,7 +14,6 @@ class BatchDeployExecutor(BaseDeployExecutor):
         env_deployment_messages = get_env_to_deployment_message(
             self.config,
             KubeDeploymentType.BATCH,
-            self.deploy_client,
             self.ecosystem_client,
             self.instance_template_client,
         )

qwak_sdk/commands/models/deployments/deploy/batch/ui.py

@@ -64,7 +64,7 @@ from qwak_sdk.inner.tools.config_handler import config_handler
 @click.option(
     "--service-account-key-secret-name",
     type=str,
-    help="Custom service accound for Gcp.",
+    help="Custom service account for Gcp.",
 )
 @click.option(
     "--sync",

qwak_sdk/commands/models/deployments/deploy/realtime/_logic/deploy_executor.py

@@ -14,7 +14,6 @@ class RealtimeDeployExecutor(BaseDeployExecutor):
         env_deployment_messages = get_env_to_deployment_message(
             self.config,
             KubeDeploymentType.ONLINE,
-            self.deploy_client,
             self.ecosystem_client,
             self.instance_template_client,
         )

qwak_sdk/commands/models/deployments/deploy/realtime/_logic/serving_strategy_mapper.py

@@ -7,24 +7,14 @@ from _qwak_proto.qwak.deployment.deployment_pb2 import (
     RealTimeConfig,
     ServingStrategy,
     TrafficConfig,
-    Variation,
-)
-from _qwak_proto.qwak.deployment.deployment_service_pb2 import (
-    GetDeploymentStatusResponse,
 )
 from _qwak_proto.qwak.ecosystem.v0.ecosystem_pb2 import UserContextEnvironmentDetails
 from qwak.exceptions import QwakException
 
 from qwak_sdk.commands.audience._logic.config.v1.audience_config import AudienceConfig
-from qwak_sdk.commands.models._logic.variations import (
-    create_variation_from_variation_config,
-)
 from qwak_sdk.commands.models.deployments.deploy._logic.deploy_config import (
     DeployConfig,
 )
-from qwak_sdk.commands.models.deployments.deploy._logic.variations import (
-    get_variations_for_deploy,
-)
 
 
 def create_realtime_serving_strategy(
@@ -32,7 +22,6 @@ def create_realtime_serving_strategy(
     audiences: List[AudienceConfig],
     fallback_variation: str,
     variation_name: str,
-    variations: List[Variation],
     variation_protected_state: bool = False,
 ) -> ServingStrategy:
     return ServingStrategy(
@@ -41,7 +30,6 @@ def create_realtime_serving_strategy(
                 auto_scaling_config=auto_scaling,
                 traffic_config=TrafficConfig(
                     selected_variation_name=variation_name,
-                    variations=variations,
                     audience_routes_entries=[
                         audience.to_audience_route_entry(index)
                         for index, audience in enumerate(audiences)
@@ -56,7 +44,6 @@ def create_realtime_serving_strategy(
 
 def create_realtime_serving_strategy_from_deploy_config(
     deploy_config: DeployConfig,
-    model_traffic: GetDeploymentStatusResponse,
     environment_name_to_config: Dict[str, UserContextEnvironmentDetails],
 ) -> Dict[str, ServingStrategy]:
     serving_strategies = {}
@@ -70,34 +57,12 @@ def create_realtime_serving_strategy_from_deploy_config(
         else None
     )
     for env_name, env_config in environment_name_to_config.items():
-        env_variations_response = model_traffic.environment_to_model_traffic.get(
-            env_config.id
-        )
-        existing_variation = (
-            env_variations_response.variations if env_variations_response else []
-        )
         try:
-            variations = []
-            if not deploy_config.realtime.audiences:
-                variations = get_variations_for_deploy(
-                    variation_name=variation_name,
-                    existing_variations=existing_variation,
-                    requested_variations=list(
-                        map(
-                            create_variation_from_variation_config,
-                            deploy_config.realtime.variations,
-                        )
-                    ),
-                    environment_name=env_name,
-                    variation_protected_state=variation_protected_state,
-                )
-
             serving_strategies[env_config.id] = create_realtime_serving_strategy(
                 auto_scaling,
                 deploy_config.realtime.audiences,
                 fallback_variation,
                 variation_name,
-                variations,
                 variation_protected_state,
             )
 

qwak_sdk/commands/models/deployments/deploy/realtime/ui.py

@@ -14,6 +14,9 @@ from qwak_sdk.commands.models.deployments.deploy._logic.deploy_config import (
 from qwak_sdk.commands.models.deployments.deploy._logic.deployment_response_handler import (
     client_deployment,
 )
+from qwak_sdk.commands.models.deployments.deploy._logic.local_deployment import (
+    local_deploy,
+)
 from qwak_sdk.commands.models.deployments.deploy.realtime._logic.deploy_executor import (
     RealtimeDeployExecutor,
 )
@@ -87,7 +90,7 @@ logger = get_qwak_logger()
 @click.option(
     "--service-account-key-secret-name",
     type=str,
-    help="Custom service accound for Gcp.",
+    help="Custom service account for Gcp.",
 )
 @click.option(
     "--max-batch-size",
@@ -113,6 +116,12 @@ logger = get_qwak_logger()
     default=False,
     help="Waiting for deployments to be ready",
 )
+@click.option(
+    "--local",
+    is_flag=True,
+    default=False,
+    help="Deploy the model container locally, for development purposes only",
+)
 @click.option(
     "-v",
     "--verbose",
@@ -170,13 +179,20 @@ logger = get_qwak_logger()
     help="Whether the deployment variation is protected. Default is false",
 )
 def realtime(
-    verbose: bool, from_file: Optional[str], out_conf: bool, sync: bool, **kwargs
+    verbose: bool,
+    from_file: Optional[str],
+    out_conf: bool,
+    sync: bool,
+    local: bool,
+    **kwargs,
 ):
     set_qwak_logger_stdout_verbosity_level(verbose + 1)
-    deploy_realtime(from_file, out_conf, sync, **kwargs)
+    deploy_realtime(from_file, out_conf, sync, local, **kwargs)
 
 
-def deploy_realtime(from_file: Optional[str], out_conf: bool, sync: bool, **kwargs):
+def deploy_realtime(
+    from_file: Optional[str], out_conf: bool, sync: bool, local: bool, **kwargs
+):
     config: DeployConfig = config_handler(
         config=DeployConfig,
         from_file=from_file,
@@ -184,6 +200,10 @@ def deploy_realtime(from_file: Optional[str], out_conf: bool, sync: bool, **kwar
         sections=("realtime", "autoscaling"),
         **kwargs,
     )
-    if not out_conf:
+
+    if local:
+        local_deploy(config)
+
+    elif not out_conf:
         deploy_executor = RealtimeDeployExecutor(config)
         client_deployment(deploy=deploy_executor, sync=sync)

qwak_sdk/commands/models/deployments/deploy/streaming/_logic/deploy_executor.py

@@ -14,7 +14,6 @@ class StreamDeployExecutor(BaseDeployExecutor):
         env_deployment_messages = get_env_to_deployment_message(
             self.config,
             KubeDeploymentType.STREAM,
-            self.deploy_client,
             self.ecosystem_client,
             self.instance_template_client,
         )

qwak_sdk/commands/models/deployments/undeploy/_logic/request_undeploy.py

@@ -8,7 +8,6 @@ from _qwak_proto.qwak.deployment.deployment_pb2 import (
     EnvironmentUndeploymentMessage,
     ModelDeploymentStatus,
     TrafficConfig,
-    Variation,
 )
 from _qwak_proto.qwak.ecosystem.v0.ecosystem_pb2 import EnvironmentDetails
 from qwak.clients.administration.eco_system.client import EcosystemClient
@@ -16,15 +15,9 @@ from qwak.clients.deployment.client import DeploymentManagementClient
 from qwak.exceptions import QwakException
 from qwak.tools.logger.logger import get_qwak_logger
 
-from qwak_sdk.commands.models._logic.variations import (
-    create_variation_from_variation_config,
-)
 from qwak_sdk.commands.models.deployments.deploy._logic.deploy_config import (
     DeployConfig,
 )
-from qwak_sdk.commands.models.deployments.undeploy._logic.variations import (
-    validate_variations_for_undeploy,
-)
 from qwak_sdk.tools.utils import qwak_spinner
 
 NO_DEPLOYED_VARIATIONS_ERROR_MSG = (
@@ -45,12 +38,10 @@ def get_deployed_variation_name(existing_variations_names: Set[str]) -> str:
 
 def get_environment_undeploy_message(
     audiences: List[AudienceRoutesEntry],
-    env_name: str,
     existing_variations_names: Set[str],
     fallback_variation: str,
     model_id: str,
     model_uuid: str,
-    requested_variations: List[Variation],
     variation_name: str,
 ):
     if not variation_name and len(existing_variations_names) == 1:
@@ -58,20 +49,12 @@ def get_environment_undeploy_message(
             existing_variations_names=existing_variations_names
         )
 
-    if not audiences:
-        validate_variations_for_undeploy(
-            variation_name,
-            existing_variations_names,
-            requested_variations,
-            env_name,
-        )
     return EnvironmentUndeploymentMessage(
         model_id=model_id,
         model_uuid=model_uuid,
         hosting_service_type=DeploymentHostingServiceType.KUBE_DEPLOYMENT,
         traffic_config=TrafficConfig(
             selected_variation_name=variation_name,
-            variations=requested_variations,
             audience_routes_entries=audiences,
             fallback_variation=fallback_variation,
         ),
@@ -84,7 +67,6 @@ def get_env_to_undeploy_message(
     env_id_to_deployment_details: Dict[str, EnvironmentDeploymentDetailsMessage],
     env_name_to_env_details: Dict[str, EnvironmentDetails],
     model_id: str,
-    requested_variations: List[Variation],
     variation_name: str,
     fallback_variation: str,
 ) -> Dict[str, EnvironmentUndeploymentMessage]:
@@ -111,12 +93,10 @@ def get_env_to_undeploy_message(
                 env_details.id
             ] = get_environment_undeploy_message(
                 audiences,
-                env_name,
                 existing_variations_names,
                 fallback_variation,
                 model_id,
                 model_uuid,
-                requested_variations,
                 variation_name,
             )
         except QwakException as e:
@@ -138,12 +118,6 @@ def undeploy(
         audience.to_audience_route_entry(index)
         for index, audience in enumerate(config.realtime.audiences)
     ]
-    requested_variations = list(
-        map(
-            create_variation_from_variation_config,
-            config.realtime.variations if config.realtime else [],
-        )
-    )
 
     if not model_uuid:
         raise QwakException("missing argument model uuid")
@@ -164,7 +138,6 @@ def undeploy(
         env_id_to_deployment_details,
         env_name_to_env_details,
         model_id,
-        requested_variations,
         config.realtime.variation_name,
         config.realtime.fallback_variation,
     )

qwak_sdk/commands/models/runtime/runtime_commands_group.py

@@ -1,7 +1,6 @@
 import click
 
 from qwak_sdk.commands.models.runtime.logs.ui import runtime_logs
-from qwak_sdk.commands.models.runtime.traffic_update.ui import runtime_traffic_update
 from qwak_sdk.commands.models.runtime.update.ui import runtime_update
 
 
@@ -15,5 +14,4 @@ def runtime_commands_group():
 
 
 runtime_commands_group.add_command(runtime_logs)
-runtime_commands_group.add_command(runtime_traffic_update)
 runtime_commands_group.add_command(runtime_update)

{qwak_sdk-0.5.62.dist-info → qwak_sdk-0.5.64.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: qwak-sdk
-Version: 0.5.62
+Version: 0.5.64
 Summary: Qwak SDK and CLI for qwak models
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model
@@ -19,7 +19,8 @@ Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Provides-Extra: batch
 Provides-Extra: feedback
-Requires-Dist: boto3 (>=1.24.116,<2.0.0) ; extra == "batch" or extra == "feedback"
+Provides-Extra: local-deployment
+Requires-Dist: boto3 (>=1.24.116,<2.0.0) ; extra == "batch" or extra == "feedback" or extra == "local-deployment"
 Requires-Dist: cookiecutter
 Requires-Dist: croniter (==1.4.1)
 Requires-Dist: gitpython (>=2.1.0)
@@ -28,8 +29,9 @@ Requires-Dist: pandas (<1.4) ; (python_full_version >= "3.7.1" and python_versio
 Requires-Dist: pandas (>=1.4.3,<2.0.0) ; (python_version >= "3.8" and python_version < "3.10") and (extra == "batch" or extra == "feedback")
 Requires-Dist: pyarrow (>=6.0.0,<11.0.0) ; extra == "batch"
 Requires-Dist: python-json-logger (>=2.0.2)
-Requires-Dist: qwak-core (==0.3.129)
+Requires-Dist: qwak-core (==0.3.163)
 Requires-Dist: qwak-inference (>=0.1,<0.2)
+Requires-Dist: rich (>=13.0.0)
 Requires-Dist: tabulate (>=0.8.0)
 Requires-Dist: yaspin (>=2.0.0)
 Project-URL: Home page, https://www.qwak.com/

{qwak_sdk-0.5.62.dist-info → qwak_sdk-0.5.64.dist-info}/RECORD

@@ -1,4 +1,4 @@
-qwak_sdk/__init__.py,sha256=IfOiNHJPhkmbzOw9hp1kPZZDP7sKoxJVXBO78h1I4Dk,135
+qwak_sdk/__init__.py,sha256=lNU63Fap7GcBaP4WD3Kn5M3EOnMoAoKbJPYJRjJuWHA,135
 qwak_sdk/cli.py,sha256=FIK1dUNxR57ypb-CeD7fKSJnPJ02lrjR9G4aj2qMLPU,2458
 qwak_sdk/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/_logic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -99,8 +99,6 @@ qwak_sdk/commands/feature_store/resume/ui.py,sha256=nI87xvA30qNQVJnT67lYJgwKGBtv
 qwak_sdk/commands/feature_store/trigger/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/feature_store/trigger/ui.py,sha256=1VJBqzxMwUWcytmZL0ymf9QlYa7oX6RDLmKsLmYVW_4,1015
 qwak_sdk/commands/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-qwak_sdk/commands/models/_logic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-qwak_sdk/commands/models/_logic/variations.py,sha256=fpe6kM8HWP7zkRl0AhZa6Ffg_LE6uiSSow0Wf5HAJxA,2050
 qwak_sdk/commands/models/build/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/models/build/_logic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/models/build/_logic/build_steps.py,sha256=Iqc3XDWhJ4LSlUiqfwP2Y9aUDpqab3WWVq-xXt-Uvo0,1533
@@ -123,7 +121,7 @@ qwak_sdk/commands/models/build/_logic/util/__init__.py,sha256=47DEQpj8HBSa-_TImW
 qwak_sdk/commands/models/build/_logic/util/protobuf_factory.py,sha256=ar_oY38w_x0sxgVF7EBs5h7gchNsDntvtKK5sSYxb24,1686
 qwak_sdk/commands/models/build/_logic/util/step_decorator.py,sha256=HLZyCGdqe3Ir7SaPWp1YNRHJpjXG-e-bbAvnOFysAVM,1913
 qwak_sdk/commands/models/build/_logic/util/text.py,sha256=tH-v19Mt8l90sMVxku5XRtrderT0qdRqJ-jLijqannA,188
-qwak_sdk/commands/models/build/ui.py,sha256=vJ5dY1RbtDY6MIWesWhNaCaWA89An7IEIaYp3nVcAMg,8851
+qwak_sdk/commands/models/build/ui.py,sha256=MX4OpBHh3X83JXc0FFsx4PP_4IZVVGe2bBS1Z5VK36o,8967
 qwak_sdk/commands/models/builds/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/models/builds/builds_commands_group.py,sha256=0nSfTY8TracXG61rFboQWUTXJisHO6dgtJKeijy6ru8,491
 qwak_sdk/commands/models/builds/cancel/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -145,34 +143,33 @@ qwak_sdk/commands/models/deployments/deploy/__init__.py,sha256=47DEQpj8HBSa-_TIm
 qwak_sdk/commands/models/deployments/deploy/_logic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/models/deployments/deploy/_logic/advance_deployment_options_handler.py,sha256=gCo3WuP-IDuA5Sg7WbFmp1hYT5qyXyf6pUD3tZGyZXs,1251
 qwak_sdk/commands/models/deployments/deploy/_logic/base_deploy_executor.py,sha256=HYKUG5s4ZrpFwoxJYn3EuTYU71lmM0cFOOjbHqd9ajY,2706
-qwak_sdk/commands/models/deployments/deploy/_logic/deploy_config.py,sha256=n7jSImcKw28xvXHoSRrtxKiGbxpqhovKv6sa7_-LYoQ,8979
+qwak_sdk/commands/models/deployments/deploy/_logic/deploy_config.py,sha256=4kcjUeUqkuGf9Zab2t6Z05Spp8VgJszRpe1rjeTZ64w,8956
 qwak_sdk/commands/models/deployments/deploy/_logic/deployment.py,sha256=zBTjeeEVtTnTwUMwUfGTI3Ip9AgqKDvHAZ6_Wi22XVo,13221
-qwak_sdk/commands/models/deployments/deploy/_logic/deployment_message_helpers.py,sha256=sSKktrGVAwfbpKwQUW1xexhiNVBkPizaCj89OmLE-8E,4852
+qwak_sdk/commands/models/deployments/deploy/_logic/deployment_message_helpers.py,sha256=wkRiQWlnQr6tjWb4wfN00Lb4Dy98QydE-Zd4j9-3zDw,4608
 qwak_sdk/commands/models/deployments/deploy/_logic/deployment_response_handler.py,sha256=cHa2iF_2A8e1wx2a4UZuTjQ5IHgPvua1xziFJwzP_6s,5837
-qwak_sdk/commands/models/deployments/deploy/_logic/deployment_size_mapper.py,sha256=Vm-vrNUMb2X6RLpLkX8jzA1_gl2o7k16w1UMVOzunIA,2165
+qwak_sdk/commands/models/deployments/deploy/_logic/deployment_size_mapper.py,sha256=sfECrq-6Emt-qY6K3J-hf8-PYw8TP4EgpjynTPstDUE,3655
 qwak_sdk/commands/models/deployments/deploy/_logic/get_latest_successful_build.py,sha256=fsahuSvgGzVjkZGaTH9YyNSOtRWBHYlyAU3wTriyCgI,978
-qwak_sdk/commands/models/deployments/deploy/_logic/variations.py,sha256=n3OGHTYgjnmLYbafZbAOaudaGbeKjw8jbX4MhZX47fE,3256
+qwak_sdk/commands/models/deployments/deploy/_logic/local_deployment.py,sha256=n42Hc-FnA3ODKok-_Nmnbjj1aPGW2P-DJ2dGDSZpSrE,6270
 qwak_sdk/commands/models/deployments/deploy/batch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/models/deployments/deploy/batch/_logic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/models/deployments/deploy/batch/_logic/advanced_deployment_mapper.py,sha256=bwvPIoKCNaJmaNsPYL_dhK04FCRZvlj_jkwhCZM-eks,599
-qwak_sdk/commands/models/deployments/deploy/batch/_logic/deploy_executor.py,sha256=oOKPotp37qdLupYau5Fv2voki4SAIFitHMn_U4ZaJP8,946
-qwak_sdk/commands/models/deployments/deploy/batch/ui.py,sha256=LxEbrPty1CYg_ohpMFih3u3h0p65zNdbp3U3QkVXYcw,3542
+qwak_sdk/commands/models/deployments/deploy/batch/_logic/deploy_executor.py,sha256=7ND4JlV0xT8v_0kEHBseS9dHtpuCs_8D1CC2L_nVJDc,914
+qwak_sdk/commands/models/deployments/deploy/batch/ui.py,sha256=Sek4sZ-XiFAJ7-bCakMX1BlOB7zrOLYHF1QyRo6LETw,3542
 qwak_sdk/commands/models/deployments/deploy/deploy_commands_group.py,sha256=LnfdnfeSTCO90SrJfgXf21xiVRry6Y2sfj_JfdBQu_U,500
 qwak_sdk/commands/models/deployments/deploy/realtime/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/models/deployments/deploy/realtime/_logic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/models/deployments/deploy/realtime/_logic/advanced_deployment_mapper.py,sha256=LDJR2XzSG3QkHaAY4cSeDOb6e1-Aw-p5ZbRiLy2BXsU,1006
-qwak_sdk/commands/models/deployments/deploy/realtime/_logic/deploy_executor.py,sha256=KClooUmWbt4Iym7_PC8jg8oHCFRWNvno-16-aFWwutA,950
-qwak_sdk/commands/models/deployments/deploy/realtime/_logic/serving_strategy_mapper.py,sha256=4UWpT6RMHvYCxnAxOcaSrvkEIz7v4oGCA-3Iz2KGNxs,3943
-qwak_sdk/commands/models/deployments/deploy/realtime/ui.py,sha256=UbuneiX9xWYYq9X7cG1MKlC6yhwZisQ_5fe63vyJcCw,5153
+qwak_sdk/commands/models/deployments/deploy/realtime/_logic/deploy_executor.py,sha256=XuuwuOBS2FtT0mEhngwo1z6teKT3GyHvWvXAJXMZG9s,918
+qwak_sdk/commands/models/deployments/deploy/realtime/_logic/serving_strategy_mapper.py,sha256=297a6ncG-lS7kscZRy88Fg11qn3_vgB6rTYeq7kXH24,2580
+qwak_sdk/commands/models/deployments/deploy/realtime/ui.py,sha256=hiVte3GkLZp3Iio6TsOs6B1RmkGmsslCXyABGZPiT5A,5509
 qwak_sdk/commands/models/deployments/deploy/streaming/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/models/deployments/deploy/streaming/_logic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-qwak_sdk/commands/models/deployments/deploy/streaming/_logic/deploy_executor.py,sha256=RRkNzFcPhfj70pjFJE5DQDMxDi-gJOiaH5C9ofHEQUA,948
+qwak_sdk/commands/models/deployments/deploy/streaming/_logic/deploy_executor.py,sha256=tfYZ3Qdk4yFBelkASYamn60JKWZRlFeFEfdXrMs8N9A,916
 qwak_sdk/commands/models/deployments/deploy/streaming/_logic/serving_strategy_mapper.py,sha256=97m93afiwmrNL9EQ3tE4JmrBYfcBnKZB9ipcvzrIWDk,1500
 qwak_sdk/commands/models/deployments/deploy/streaming/ui.py,sha256=9tPMnVnLNX3lYyBd4BAh-YThDFz99oyvrCRd_tDZufU,5796
 qwak_sdk/commands/models/deployments/undeploy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/models/deployments/undeploy/_logic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-qwak_sdk/commands/models/deployments/undeploy/_logic/request_undeploy.py,sha256=e-9lHQ2AbTyogUHxppPVli5pugl6ml4c1zAuFoLQTpw,9481
-qwak_sdk/commands/models/deployments/undeploy/_logic/variations.py,sha256=qo4qUB78voUhtM1YIuTz1OSBaLRS3xw97bwxgqRuLgI,3036
+qwak_sdk/commands/models/deployments/undeploy/_logic/request_undeploy.py,sha256=s9YneVS_RgBMTGODAxLxK5qqgQ9bdGsO91xoNS76CdY,8623
 qwak_sdk/commands/models/deployments/undeploy/ui.py,sha256=1XYBouPgbJWOJCNR1UU1UOVnQS10iKsUqeNkvIBhHY8,2008
 qwak_sdk/commands/models/describe/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/models/describe/_logic.py,sha256=LN1r3zrXLHf83U5ate7ED7VdoZuWUqcHwLZPMMZXRfs,5589
@@ -252,12 +249,7 @@ qwak_sdk/commands/models/models_command_group.py,sha256=xrI_XwHUaahnH0P2iNkUps34
 qwak_sdk/commands/models/runtime/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/models/runtime/logs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/models/runtime/logs/ui.py,sha256=2wxk4eqwKqSr83Wsn-4wHRvug3hQLmHeoyJ4zBAsW4U,1577
-qwak_sdk/commands/models/runtime/runtime_commands_group.py,sha256=q8h9CNNXFdWW5ay-MFlTkG9MnwfLrkPDuI-BWsyjuos,566
-qwak_sdk/commands/models/runtime/traffic_update/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-qwak_sdk/commands/models/runtime/traffic_update/_logic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-qwak_sdk/commands/models/runtime/traffic_update/_logic/execute_runtime_update_traffic.py,sha256=el3MlANMcmbfxNCVTeqfT6x9DB_F4Ns6O-M2Am9F_uI,1811
-qwak_sdk/commands/models/runtime/traffic_update/_logic/variations.py,sha256=UhVjyW4jeS7558ItpgffvBa7b8Pl2Jbb7SIEL0kgukU,3160
-qwak_sdk/commands/models/runtime/traffic_update/ui.py,sha256=gCuPpRNYtB98Glr5FExj5nLfe8iwKF7xLAUzxqJ4tb8,1212
+qwak_sdk/commands/models/runtime/runtime_commands_group.py,sha256=nuyPkGSbJ7RD6HNif6ygfR4yPXtPiDEEJ7UaVn04Wm8,421
 qwak_sdk/commands/models/runtime/update/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/models/runtime/update/_logic.py,sha256=vXgx16OTwx-8iucop-fU5o-BTWo5Iqiv-8aZjyeZ3Yg,393
 qwak_sdk/commands/models/runtime/update/ui.py,sha256=24GX7V05zK0HwLdNIBMaf1zJ9CC2BcrWjQsgXAJviy8,622
@@ -326,7 +318,7 @@ qwak_sdk/tools/colors.py,sha256=7pui_GGjC4uZKYFsIyXaJjYsjLxJVHb4OrfTgr93hqo,287
 qwak_sdk/tools/files.py,sha256=AyKJTOy7NhvP3SrqwIw_lxYNCOy1CvLgMmSJpWZ0OKM,2257
 qwak_sdk/tools/log_handling.py,sha256=Aa1EmxUPCX8YWiZRutUvnqPv6K_z1zoGMwIWsEv24mM,6327
 qwak_sdk/tools/utils.py,sha256=SHmU4r_m2ABZyFYMC03P17GvltPbYbmB39hvalIZEtI,1168
-qwak_sdk-0.5.62.dist-info/entry_points.txt,sha256=vSl0ELYDyj640oMM57u0AjBP87wtLYxCcGOendhEx80,47
-qwak_sdk-0.5.62.dist-info/WHEEL,sha256=vVCvjcmxuUltf8cYhJ0sJMRDLr1XsPuxEId8YDzbyCY,88
-qwak_sdk-0.5.62.dist-info/METADATA,sha256=ZsEQFkftScEaa23YHo2LCvVGkB_1FixMFyvZPxvFob0,1889
-qwak_sdk-0.5.62.dist-info/RECORD,,
+qwak_sdk-0.5.64.dist-info/entry_points.txt,sha256=vSl0ELYDyj640oMM57u0AjBP87wtLYxCcGOendhEx80,47
+qwak_sdk-0.5.64.dist-info/WHEEL,sha256=vVCvjcmxuUltf8cYhJ0sJMRDLr1XsPuxEId8YDzbyCY,88
+qwak_sdk-0.5.64.dist-info/METADATA,sha256=9eccdkFDg9jVxb-G3KZOWYoUwQtBJKKEp-bRBOjiSVE,1984
+qwak_sdk-0.5.64.dist-info/RECORD,,

qwak_sdk/commands/models/_logic/variations.py

@@ -1,55 +0,0 @@
-from __future__ import annotations
-
-from typing import List
-
-from _qwak_proto.qwak.deployment.deployment_pb2 import TrafficSpec, Variation
-from qwak.exceptions import QwakException
-
-from qwak_sdk.commands.models.deployments.deploy._logic.deploy_config import (
-    DeployConfig,
-)
-
-
-def validate_percentages(variations: List[Variation]):
-    for variation in variations:
-        if not (0 <= variation.traffic.percentage <= 100):
-            raise QwakException(
-                f"The variation '{variation.name}' contains invalid value '{variation.traffic.percentage}'. "
-                f"Value must be between 0 and 100."
-            )
-
-    non_shadow_variations = list(filter(lambda v: not v.traffic.is_shadow, variations))
-    non_shadow_variations_percentage_sum = sum(
-        map(lambda v: v.traffic.percentage, non_shadow_variations)
-    )
-    if non_shadow_variations and non_shadow_variations_percentage_sum != 100:
-        raise QwakException(
-            "The variations do not sum to 100 percent. Please go over the variations config file."
-        )
-
-
-def validate_shadow_variation(variations: List[Variation]):
-    shadow_variations = list(filter(lambda v: v.traffic.is_shadow, variations))
-    if len(shadow_variations) > 1:
-        shadow_variations_names = list(map(lambda v: v.name, shadow_variations))
-        raise QwakException(
-            f"The variations contain more than one shadow variation {shadow_variations_names}. "
-            f"Please go over the variations config file."
-        )
-
-
-def create_variation_from_variation_config(
-    variation_def: DeployConfig.Realtime.VariationConfig,
-) -> Variation:
-    try:
-        return Variation(
-            name=variation_def.name,
-            traffic=TrafficSpec(
-                percentage=variation_def.traffic.percentage,
-                is_shadow=variation_def.traffic.shadow,
-            ),
-        )
-    except Exception:
-        raise QwakException(
-            f"Could not parse variation {variation_def}. Please check you variation configuration file."
-        )

qwak_sdk/commands/models/deployments/deploy/_logic/variations.py

@@ -1,81 +0,0 @@
-from __future__ import annotations
-
-from typing import List
-
-from _qwak_proto.qwak.deployment.deployment_pb2 import TrafficSpec, Variation
-from qwak.exceptions import QwakException
-
-from qwak_sdk.commands.models._logic.variations import (
-    validate_percentages,
-    validate_shadow_variation,
-)
-
-
-def get_variations_for_deploy(
-    variation_name: str,
-    existing_variations: List[Variation],
-    requested_variations: List[Variation],
-    environment_name: str,
-    variation_protected_state: bool,
-):
-    if not existing_variations:
-        existing_variations: List[Variation] = [
-            Variation(
-                name=variation_name,
-                traffic=TrafficSpec(
-                    percentage=100,
-                    is_shadow=False,
-                ),
-                protected=variation_protected_state,
-            )
-        ]
-    requested_variations_names = set(
-        map(lambda variation: variation.name, requested_variations)
-    )
-    existing_variations_names = set(
-        map(lambda variation: variation.name, existing_variations)
-    )
-    if requested_variations_names:
-        if variation_name not in requested_variations_names:
-            raise QwakException(
-                f"Selected deployment to variation '{variation_name}' in {environment_name}, but provided the "
-                f"following variations {list(requested_variations_names)}. "
-                f"Please modify the variation name, or adjust the requested variations definitions"
-            )
-    elif variation_name not in existing_variations_names:
-        raise QwakException(
-            f"Selected deployment to existing variation '{variation_name}' in {environment_name}, but it is not one of "
-            f"the existing variations {list(existing_variations_names)}. "
-            f"Please modify the variation name, or supply the requested variations definitions with"
-            f" the necessary changes"
-        )
-    else:
-        # We are deploying to an existing variations without any change
-        return existing_variations
-
-    ignored_variations = existing_variations_names.difference(
-        requested_variations_names
-    )
-    if ignored_variations:
-        raise QwakException(
-            f"The given variation configuration does not contain the configuration for the following "
-            f"variations {list(ignored_variations)} in {environment_name}. You must include all the existing "
-            f"variations configuration when passing the configuration"
-        )
-
-    unexpected_variations = requested_variations_names.difference(
-        existing_variations_names
-    )
-    if variation_name in unexpected_variations:
-        unexpected_variations.remove(variation_name)
-    if unexpected_variations:
-        raise QwakException(
-            f"The given variation configuration contains the configuration for the following "
-            f"variations {list(unexpected_variations)} in {environment_name} which are not expected. When deploying, "
-            f"the variations configuration must contain all the existing variations, and an optional new variation."
-        )
-
-    validate_percentages(requested_variations)
-    validate_shadow_variation(requested_variations)
-
-    return requested_variations

qwak_sdk/commands/models/deployments/undeploy/_logic/variations.py

@@ -1,74 +0,0 @@
-from __future__ import annotations
-
-from typing import List, Set
-
-from _qwak_proto.qwak.deployment.deployment_pb2 import Variation
-from qwak.exceptions import QwakException
-
-from qwak_sdk.commands.models.deployments.deploy._logic.variations import (
-    validate_percentages,
-    validate_shadow_variation,
-)
-
-
-def validate_variations_for_undeploy(
-    variation_name: str,
-    existing_variations_names: Set[str],
-    requested_variations: List[Variation],
-    environment_name: str,
-):
-    if variation_name not in existing_variations_names:
-        if not variation_name:
-            raise QwakException(
-                f"No selected variation. You must specify a variation since there is more than one deployed in "
-                f"{environment_name}. Currently {list(existing_variations_names)} are deployed"
-            )
-        raise QwakException(
-            f"The selected variation {variation_name} is not deployed in {environment_name}. "
-            f"Only {list(existing_variations_names)} are deployed"
-        )
-
-    requested_variations_names = set(
-        map(lambda variation: variation.name, requested_variations)
-    )
-    if variation_name in requested_variations_names:
-        raise QwakException(
-            f"Selected undeploy to variation '{variation_name}', but it is part of "
-            f"the requested variations {list(requested_variations_names)}. "
-            f"Please modify the variation name, or supply the requested variations definitions with"
-            f" the necessary changes"
-        )
-    if variation_name not in existing_variations_names:
-        raise QwakException(
-            f"Selected undeploy to variation {variation_name} but it is not one of "
-            f"the existing variations {list(existing_variations_names)}. "
-            f"Please modify the variation name."
-        )
-
-    ignored_variations = existing_variations_names.difference(
-        requested_variations_names
-    )
-    if variation_name in ignored_variations:
-        ignored_variations.remove(variation_name)
-
-    if len(existing_variations_names) != 2 and ignored_variations:
-        raise QwakException(
-            f"The given variation configuration does not contain the configuration for the following "
-            f"variations {list(ignored_variations)}. You must include all the variations that will exist after "
-            f"undeploy in the configuration"
-        )
-
-    unexpected_variations = requested_variations_names.difference(
-        existing_variations_names
-    )
-
-    if unexpected_variations:
-        raise QwakException(
-            f"The given variation configuration contains the configuration for the following "
-            f"variations {list(unexpected_variations)} which are not expected. When undeploying, the variations "
-            f"configuration must contain all expected remaining variations (from the existing ones)."
-        )
-
-    if requested_variations:
-        validate_percentages(requested_variations)
-        validate_shadow_variation(requested_variations)

qwak_sdk/commands/models/runtime/traffic_update/_logic/execute_runtime_update_traffic.py

@@ -1,54 +0,0 @@
-from typing import Tuple
-
-from qwak.clients.administration.eco_system.client import EcosystemClient
-from qwak.clients.deployment.client import DeploymentManagementClient
-from qwak.tools.logger.logger import get_qwak_logger
-
-from qwak_sdk.commands.models._logic.variations import (
-    create_variation_from_variation_config,
-)
-from qwak_sdk.commands.models.deployments.deploy._logic.deploy_config import (
-    DeployConfig,
-)
-from qwak_sdk.commands.models.runtime.traffic_update._logic.variations import (
-    validate_requested_variations_against_existing_in_apply,
-)
-from qwak_sdk.inner.tools.config_handler import config_handler
-
-logger = get_qwak_logger()
-
-
-def execute_runtime_traffic_update(
-    model_id: str, from_file: str, environment_name: Tuple[str]
-):
-    deployment_client = DeploymentManagementClient()
-    ecosystem_client = EcosystemClient()
-    config: DeployConfig = config_handler(
-        config=DeployConfig,
-        from_file=from_file,
-        out_conf=False,
-        sections=("realtime",),
-        model_id=model_id,
-    )
-    requested_variations = list(
-        map(
-            create_variation_from_variation_config,
-            config.realtime.variations,
-        )
-    )
-    environment_names_to_details = ecosystem_client.get_environments_names_to_details(
-        environment_name if environment_name else config.realtime.environments
-    )
-    validate_requested_variations_against_existing_in_apply(
-        model_id,
-        requested_variations,
-        environment_names_to_details,
-        deployment_client,
-    )
-    environment_ids = [env.id for env in environment_names_to_details.values()]
-
-    deployment_client.apply_model_traffic_config(
-        model_id=model_id,
-        requested_variations=requested_variations,
-        environment_ids=environment_ids,
-    )

qwak_sdk/commands/models/runtime/traffic_update/_logic/variations.py

@@ -1,84 +0,0 @@
-from typing import Dict, List, Optional
-
-from _qwak_proto.qwak.deployment.deployment_pb2 import (
-    EnvironmentTrafficMessage,
-    Variation,
-)
-from _qwak_proto.qwak.ecosystem.v0.ecosystem_pb2 import EnvironmentDetails
-from qwak.clients.deployment.client import DeploymentManagementClient
-from qwak.exceptions import QwakException
-
-from qwak_sdk.commands.models._logic.variations import (
-    validate_percentages,
-    validate_shadow_variation,
-)
-
-
-def validate_requested_variations_against_existing_in_apply(
-    model_id: str,
-    requested_variations: List[Variation],
-    environment_names_to_details: Dict[str, EnvironmentDetails],
-    deployment_client: DeploymentManagementClient,
-):
-    model_traffic = deployment_client.get_model_traffic_config(model_id)
-    environments_traffic = dict(model_traffic.environment_to_model_traffic)
-    errors = []
-    for env_name, env_details in environment_names_to_details.items():
-        error = validate_requested_variation_in_envs_for_apply(
-            env_details, environments_traffic, requested_variations
-        )
-        if error:
-            errors.append(error)
-    if errors:
-        raise QwakException("\n".join(errors))
-
-
-def validate_requested_variation_in_envs_for_apply(
-    env_details: EnvironmentDetails,
-    environments_traffic: Dict[str, EnvironmentTrafficMessage],
-    requested_variations: List[Variation],
-) -> Optional[str]:
-    environment_traffic = environments_traffic.get(env_details.id)
-    existing_variations = environment_traffic.variations if environment_traffic else []
-    try:
-        validate_variation_for_apply(
-            existing_variations=existing_variations,
-            requested_variations=requested_variations,
-        )
-    except QwakException as e:
-        return e.message
-
-
-def validate_variation_for_apply(
-    existing_variations: List[Variation],
-    requested_variations: List[Variation],
-):
-    requested_variations_names = set(
-        map(lambda variation: variation.name, requested_variations)
-    )
-    existing_variations_names = set(
-        map(lambda variation: variation.name, existing_variations)
-    )
-    ignored_variations = existing_variations_names.difference(
-        requested_variations_names
-    )
-    if ignored_variations:
-        raise QwakException(
-            f"The given variation configuration does not contain the configuration for the following "
-            f"variations {list(ignored_variations)}. You must include all the existing variations when updating the "
-            f"model traffic {list(existing_variations_names)}"
-        )
-
-    unexpected_variations = requested_variations_names.difference(
-        existing_variations_names
-    )
-
-    if unexpected_variations:
-        raise QwakException(
-            f"The given variation configuration contains the configuration for the following "
-            f"variations {list(unexpected_variations)} which are not expected. You must include only the existing "
-            f"variations when updating the model traffic {list(existing_variations_names)}"
-        )
-
-    validate_percentages(requested_variations)
-    validate_shadow_variation(requested_variations)

qwak_sdk/commands/models/runtime/traffic_update/ui.py

@@ -1,37 +0,0 @@
-from typing import Tuple
-
-import click
-from qwak.exceptions import QwakException
-from qwak.tools.logger.logger import get_qwak_logger
-
-from qwak_sdk.commands.models.runtime.traffic_update._logic.execute_runtime_update_traffic import (
-    execute_runtime_traffic_update,
-)
-from qwak_sdk.inner.tools.cli_tools import QwakCommand
-
-logger = get_qwak_logger()
-
-
-@click.command("traffic_update", cls=QwakCommand)
-@click.option("-m", "--model-id", required=True, help="Model named ID")
-@click.option(
-    "--environment-name",
-    required=False,
-    type=str,
-    help="Environments to deploy on (if not specified uses your default environment)",
-    multiple=True,
-)
-@click.option(
-    "--from-file",
-    required=True,
-    help="The variations config file path",
-    type=click.Path(exists=True, resolve_path=True, dir_okay=False),
-)
-def runtime_traffic_update(
-    model_id: str, from_file: str, environment_name: Tuple[str], **kwargs
-):
-    try:
-        execute_runtime_traffic_update(model_id, from_file, environment_name)
-        logger.info(f"Successfully updated traffic for models {model_id}")
-    except Exception as e:
-        raise QwakException(f'Failed to apply traffic configurations. Error is "{e}"')