thestage 0.5.36__py3-none-any.whl → 0.5.38__py3-none-any.whl

thestage/.env

@@ -1,5 +1,6 @@
 THESTAGE_CONFIG_DIR=.thestage
 THESTAGE_CONFIG_FILE=config.json
-THESTAGE_API_URL=https://backend-staging3.thestage.ai
-THESTAGE_API_URL=http://localhost:8100
-LOG_FILE=thestage.log
+THESTAGE_CLI_ENV=DEV
+THESTAGE_API_URL=https://backend-staging.thestage.ai
+THESTAGE_API_URL=https://backend.thestage.ai
+LOG_FILE=thestage.log

thestage/__init__.py

@@ -1,3 +1,3 @@
 from . import *
 __app_name__ = "thestage"
-__version__ = "0.5.36"
+__version__ = "0.5.38"

thestage/controllers/project_controller.py

@@ -1,3 +1,4 @@
+import time
 from typing import Optional, List
 
 import re
@@ -723,4 +724,138 @@ def inference_simulator_logs(
             logging_service.print_last_inference_simulator_logs(config=config, inference_simulator_id=inference_simulator_id, logs_number=logs_number)
 
     app_logger.info(f'Inference simulator logs - end')
-    raise typer.Exit(0)
+    raise typer.Exit(0)
+
+
+@inference_simulator_model_app.command("deploy-instance", no_args_is_help=True, help=__("Deploy an inference simulator model to an instance"))
+def deploy_inference_simulator_model_to_instance(
+        unique_id: Optional[str] = typer.Argument(help=__("Inference simulator model unique ID"), ),
+        rented_instance_unique_id: Optional[str] = typer.Option(
+            None,
+            '--rented-instance-unique-id',
+            '-ruid',
+            help=__("The rented instance unique ID on which the inference simulator model will be deployed"),
+            is_eager=False,
+        ),
+        self_hosted_instance_unique_id: Optional[str] = typer.Option(
+            None,
+            '--self-hosted-instance-unique-id',
+            '-suid',
+            help=__("The self-hosted instance unique ID on which the inference simulator model will be deployed"),
+            is_eager=False,
+        ),
+        working_directory: Optional[str] = typer.Option(
+            None,
+            "--working-directory",
+            "-wd",
+            help=__("Full path to working directory. By default, the current directory is used"),
+            show_default=False,
+            is_eager=False,
+        ),
+        enable_log_stream: Optional[bool] = typer.Option(
+            True,
+            " /--no-logs",
+            " /-nl",
+            help=__("Disable real-time log streaming"),
+            is_eager=False,
+        ),
+):
+    """
+        Deploy an inference simulator model to an instance
+    """
+
+    if unique_id and not re.match(r"^[a-zA-Z0-9-]+$", unique_id):
+        raise typer.BadParameter(__("Invalid UID format. The UID can only contain letters, numbers, and hyphens."))
+
+    unique_id_with_timestamp = f"{unique_id}-{int(time.time())}"
+
+    app_logger.info(f'Deploying an inference simulator model')
+
+    service_factory = validate_config_and_get_service_factory(working_directory=working_directory)
+    config = service_factory.get_config_provider().get_full_config()
+
+    project_service = service_factory.get_project_service()
+
+    project_service.project_deploy_inference_simulator_model_to_instance(
+        config=config,
+        unique_id=unique_id,
+        unique_id_with_timestamp=unique_id_with_timestamp,
+        rented_instance_unique_id=rented_instance_unique_id,
+        self_hosted_instance_unique_id=self_hosted_instance_unique_id,
+    )
+
+    if enable_log_stream:
+        logging_service: LoggingService = service_factory.get_logging_service()
+
+        logging_service.stream_inference_simulator_logs_with_controls(
+            config=config,
+            slug=unique_id_with_timestamp
+        )
+    raise typer.Exit(0)
+
+
+@inference_simulator_model_app.command("deploy-sagemaker", no_args_is_help=True, help=__("Deploy an inference simulator model to SageMaker"))
+def deploy_inference_simulator_model_to_sagemaker(
+        unique_id: Optional[str] = typer.Argument(help=__("Inference simulator model unique ID"), ),
+        arn: Optional[str] = typer.Option(
+            None,
+            '--amazon-resource-name',
+            '-arn',
+            help=__("The Amazon Resource Name of the IAM Role to use, e.g., arn:aws:iam::{aws_account_id}:role/{role}"),
+            is_eager=False,
+        ),
+        working_directory: Optional[str] = typer.Option(
+            None,
+            "--working-directory",
+            "-wd",
+            help=__("Full path to working directory. By default, the current directory is used"),
+            show_default=False,
+            is_eager=False,
+        ),
+        instance_type: Optional[str] = typer.Option(
+            None,
+            '--instance-type',
+            '-it',
+            help=__("Instance type on which the inference simulator model will be deployed"),
+            is_eager=False,
+        ),
+        initial_variant_weight: Optional[float] = typer.Option(
+            None,
+            "--initial-variant-weight",
+            "-ivw",
+            help=__("Initial Variant Weight. By default 1.0"),
+            show_default=False,
+            is_eager=False,
+        ),
+        initial_instance_count: Optional[int] = typer.Option(
+            None,
+            "--initial-instance-count",
+            "-iic",
+            help=__("Initial Instance Count"),
+            show_default=False,
+            is_eager=False,
+        ),
+
+):
+    """
+        Deploy an inference simulator model to SageMaker
+    """
+
+    if unique_id and not re.match(r"^[a-zA-Z0-9-]+$", unique_id):
+        raise typer.BadParameter(__("Invalid UID format. The UID can only contain letters, numbers, and hyphens."))
+
+    app_logger.info(f'Deploying an inference simulator model')
+
+    service_factory = validate_config_and_get_service_factory(working_directory=working_directory)
+    config = service_factory.get_config_provider().get_full_config()
+
+    project_service = service_factory.get_project_service()
+
+    project_service.project_deploy_inference_simulator_model_to_sagemaker(
+        config=config,
+        unique_id=unique_id,
+        arn=arn,
+        instance_type=instance_type,
+        initial_variant_weight=initial_variant_weight,
+        initial_instance_count=initial_instance_count,
+    )

thestage/entities/project_inference_simulator_model.py

@@ -11,6 +11,7 @@ class ProjectInferenceSimulatorModelEntity(BaseModel):
     )
 
     id: Optional[int] = Field(None, alias='ID')
+    slug: Optional[str] = Field(None, alias='UNIQUE_ID')
     status: Optional[str] = Field(None, alias='STATUS')
     commit_hash: Optional[str] = Field(None, alias='COMMIT_HASH')
     environment_metadata: Optional[str] = Field(None, alias='ENVIRONMENT_METADATA')

thestage/services/clients/thestage_api/api_client.py

@@ -10,6 +10,14 @@ from thestage.services.clients.thestage_api.dtos.docker_container_controller.doc
 from thestage.services.clients.thestage_api.dtos.docker_container_controller.docker_container_list_response import \
     DockerContainerListResponse
 from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
+from thestage.services.clients.thestage_api.dtos.inference_controller.deploy_inference_model_to_instance_request import \
+    DeployInferenceModelToInstanceRequest
+from thestage.services.clients.thestage_api.dtos.inference_controller.deploy_inference_model_to_instance_response import \
+    DeployInferenceModelToInstanceResponse
+from thestage.services.clients.thestage_api.dtos.inference_controller.deploy_inference_model_to_sagemaker_request import \
+    DeployInferenceModelToSagemakerRequest
+from thestage.services.clients.thestage_api.dtos.inference_controller.deploy_inference_model_to_sagemaker_response import \
+    DeployInferenceModelToSagemakerResponse
 from thestage.services.clients.thestage_api.dtos.inference_controller.get_inference_simulator_request import \
     GetInferenceSimulatorRequest
 from thestage.services.clients.thestage_api.dtos.inference_controller.get_inference_simulator_response import \
@@ -736,6 +744,53 @@ class TheStageApiClient(TheStageApiClientCore):
         return GetInferenceSimulatorResponse.model_validate(response) if response else None
 
 
+    @error_handler()
+    def deploy_inference_model_to_instance(
+            self,
+            token: str,
+            unique_id: str,
+            unique_id_with_timestamp: str,
+            rented_instance_unique_id: Optional[str] = None,
+            self_hosted_instance_unique_id: Optional[str] = None,
+
+    ) -> Optional[DeployInferenceModelToInstanceResponse]:
+        request = DeployInferenceModelToInstanceRequest(
+            inferenceSimulatorSlug=unique_id_with_timestamp,
+            modelSlug=unique_id,
+            instanceRentedSlug = rented_instance_unique_id,
+            selfhostedInstanceSlug = self_hosted_instance_unique_id
+        )
+
+        response = self._request(
+            method='POST',
+            url='/user-api/v1/inference-simulator-model/deploy/instance',
+            data=request.model_dump(),
+            token=token,
+        )
+        return DeployInferenceModelToInstanceResponse.model_validate(response) if response else None
+
+
+    @error_handler()
+    def deploy_inference_model_to_sagemaker(
+            self,
+            token: str,
+            unique_id: str,
+            arn: Optional[str] = None,
+    ) -> Optional[DeployInferenceModelToSagemakerResponse]:
+        request = DeployInferenceModelToSagemakerRequest(
+            modelSlug=unique_id,
+            arn=arn,
+        )
+
+        response = self._request(
+            method='POST',
+            url='/user-api/v1/inference-simulator-model/grant-user-arn-access',
+            data=request.model_dump(),
+            token=token,
+        )
+        return DeployInferenceModelToSagemakerResponse.model_validate(response) if response else None
+
+
     def query_user_logs(self, token: str, limit: int, task_id: Optional[int] = None, inference_simulator_id: Optional[int] = None) -> UserLogsQueryResponse:
         request = UserLogsQueryRequest(
             inferenceSimulatorId=inference_simulator_id,

thestage/services/clients/thestage_api/dtos/inference_controller/deploy_inference_model_to_instance_request.py

@@ -0,0 +1,14 @@
+from typing import Optional, List
+
+from pydantic import Field, ConfigDict, BaseModel
+
+from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
+
+
+class DeployInferenceModelToInstanceRequest(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    modelSlug: Optional[str] = Field(None, alias='modelSlug')
+    instanceRentedSlug: Optional[str] = Field(None, alias='instanceRentedSlug')
+    selfhostedInstanceSlug: Optional[str] = Field(None, alias='selfhostedInstanceSlug')
+    inferenceSimulatorSlug: Optional[str] = Field(None, alias='inferenceSimulatorSlug')

thestage/services/clients/thestage_api/dtos/inference_controller/deploy_inference_model_to_instance_response.py

@@ -0,0 +1,13 @@
+from typing import Optional
+
+from pydantic import Field, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+from thestage.services.project.dto.inference_simulator_model_dto import InferenceSimulatorModelDto
+
+
+class DeployInferenceModelToInstanceResponse(TheStageBaseResponse):
+    model_config = ConfigDict(use_enum_values=True)
+    inferenceSimulatorModel: Optional[InferenceSimulatorModelDto] = Field(None, alias='inferenceSimulatorModel')
+
+

thestage/services/clients/thestage_api/dtos/inference_controller/deploy_inference_model_to_sagemaker_request.py

@@ -0,0 +1,12 @@
+from typing import Optional, List
+
+from pydantic import Field, ConfigDict, BaseModel
+
+from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
+
+
+class DeployInferenceModelToSagemakerRequest(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    modelSlug: Optional[str] = Field(None, alias='modelSlug')
+    arn: Optional[str] = Field(None, alias='arn')

thestage/services/clients/thestage_api/dtos/inference_controller/deploy_inference_model_to_sagemaker_response.py

@@ -0,0 +1,12 @@
+from pydantic import Field, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+
+
+class DeployInferenceModelToSagemakerResponse(TheStageBaseResponse):
+    model_config = ConfigDict(use_enum_values=True)
+    modelId: str = Field(None, alias='modelId')
+    ecrImageUrl: str = Field(None, alias='ecrImageUrl')
+    s3ArtifactsUrl: str = Field(None, alias='s3ArtifactsUrl')
+
+

thestage/services/container/container_service.py

@@ -196,7 +196,7 @@ class ContainerService(AbstractService):
             typer.echo(__("Unable to connect to container: container system_name is missing"))
             raise typer.Exit(1)
 
-        starting_directory: str = '/app'
+        starting_directory: str = '/'
         workspace_mappings = {v for v in container.mappings.directory_mappings.values() if v.startswith('/workspace/') or v == '/workspace'}
         if len(workspace_mappings) > 0:
             starting_directory = '/workspace'

thestage/services/project/dto/inference_simulator_model_dto.py

@@ -5,6 +5,7 @@ class InferenceSimulatorModelDto(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
 
     id: Optional[int] = Field(None, alias='id')
+    slug: Optional[str] = Field(None, alias='slug')
     client_id: Optional[int] = Field(None, alias='clientId')
     instance_rented_id: Optional[int] = Field(None, alias='instanceRentedId')
     selfhosted_instance_id: Optional[int] = Field(None, alias='selfhostedInstanceId')

thestage/services/project/mapper/project_inference_simulator_model_mapper.py

@@ -12,6 +12,7 @@ class ProjectInferenceSimulatorModelMapper(AbstractMapper):
 
         return ProjectInferenceSimulatorModelEntity(
             id=item.id,
+            slug=item.slug,
             status=item.status or '',
             commit_hash=item.commit_hash or '',
             environment_metadata=item.environment_metadata or '',

thestage/services/project/project_service.py

@@ -1,8 +1,12 @@
 import os
+import time
+from datetime import datetime
 from pathlib import Path
 from typing import Optional, List
 
 import json
+
+import boto3
 import click
 import typer
 from git import Commit
@@ -18,6 +22,10 @@ from thestage.i18n.translation import __
 from thestage.services.clients.git.git_client import GitLocalClient
 from thestage.services.clients.thestage_api.dtos.container_response import DockerContainerDto
 from thestage.services.clients.thestage_api.dtos.enums.container_status import DockerContainerStatus
+from thestage.services.clients.thestage_api.dtos.inference_controller.deploy_inference_model_to_instance_response import \
+    DeployInferenceModelToInstanceResponse
+from thestage.services.clients.thestage_api.dtos.inference_controller.deploy_inference_model_to_sagemaker_response import \
+    DeployInferenceModelToSagemakerResponse
 from thestage.services.clients.thestage_api.dtos.inference_controller.get_inference_simulator_response import \
     GetInferenceSimulatorResponse
 from thestage.services.clients.thestage_api.dtos.paginated_entity_list import PaginatedEntityList
@@ -902,3 +910,170 @@ class ProjectService(AbstractService):
 
         return project_config
 
+    @error_handler()
+    def project_deploy_inference_simulator_model_to_instance(
+            self,
+            config: ConfigEntity,
+            unique_id: Optional[str] = None,
+            unique_id_with_timestamp: Optional[str] = None,
+            rented_instance_unique_id: Optional[str] = None,
+            self_hosted_instance_unique_id: Optional[str] = None,
+    ) -> None:
+        project_config: ProjectConfig = self.__get_fixed_project_config(config=config)
+        if not project_config:
+            typer.echo(
+                __("No project found at the path: %path%. Please initialize or clone a project first. Or provide path to project using --working-directory option.",
+                   {"path": config.runtime.working_directory}))
+            raise typer.Exit(1)
+
+        if rented_instance_unique_id and self_hosted_instance_unique_id:
+            typer.echo(__("Error: Cannot provide both rented and self-hosted instance unique IDs."))
+            raise typer.Exit(1)
+
+        if not rented_instance_unique_id and not self_hosted_instance_unique_id:
+            typer.echo(__("Error: Either a rented instance ID or a self-hosted instance unique ID must be provided."))
+            raise typer.Exit(1)
+
+        project_config: ProjectConfig = self.__config_provider.read_project_config()
+        if not project_config:
+            typer.echo(__("No project found at the path: %path%. Please initialize or clone a project first.",
+                          {"path": config.runtime.working_directory}))
+            raise typer.Exit(1)
+
+        typer.echo(__("Creating inference simulator with unique ID: %unique_id_with_timestamp%", {"unique_id_with_timestamp": unique_id_with_timestamp}))
+        deploy_model_to_instance_response: DeployInferenceModelToInstanceResponse = self.__thestage_api_client.deploy_inference_model_to_instance(
+            token=config.main.thestage_auth_token,
+            unique_id=unique_id,
+            unique_id_with_timestamp=unique_id_with_timestamp,
+            rented_instance_unique_id=rented_instance_unique_id,
+            self_hosted_instance_unique_id=self_hosted_instance_unique_id
+        )
+        if deploy_model_to_instance_response:
+            if deploy_model_to_instance_response.message:
+                typer.echo(deploy_model_to_instance_response.message)
+            if deploy_model_to_instance_response.is_success:
+                typer.echo("Inference simulator has been scheduled to run successfully.")
+            else:
+                typer.echo(__(
+                    'Inference simulator failed to run with an error: %server_massage%',
+                    {'server_massage': deploy_model_to_instance_response.message or ""}
+                ))
+                raise typer.Exit(1)
+        else:
+            typer.echo(__("Inference simulator failed to run with an error"))
+            raise typer.Exit(1)
+
+
+    @error_handler()
+    def project_deploy_inference_simulator_model_to_sagemaker(
+            self,
+            config: ConfigEntity,
+            unique_id: Optional[str] = None,
+            arn: Optional[str] = None,
+            instance_type: Optional[str] = None,
+            initial_variant_weight: Optional[float] = 1.0,
+            initial_instance_count: Optional[int] = None,
+    ) -> None:
+        project_config: ProjectConfig = self.__get_fixed_project_config(config=config)
+        if not project_config:
+            typer.echo(
+                __("No project found at the path: %path%. Please initialize or clone a project first. Or provide path to project using --working-directory option.",
+                   {"path": config.runtime.working_directory}))
+            raise typer.Exit(1)
+
+        if not instance_type:
+            typer.echo(__("Error: Instance type is required."))
+            raise typer.Exit(1)
+
+        if not initial_instance_count:
+            typer.echo(__("Error: Initial instance count is required."))
+            raise typer.Exit(1)
+
+        if not arn:
+            typer.echo(__("Error: ARN is required."))
+            raise typer.Exit(1)
+
+        project_config: ProjectConfig = self.__config_provider.read_project_config()
+        if not project_config:
+            typer.echo(__("No project found at the path: %path%. Please initialize or clone a project first.",
+                          {"path": config.runtime.working_directory}))
+            raise typer.Exit(1)
+
+        deploy_model_to_sagemaker_response: DeployInferenceModelToSagemakerResponse = self.__thestage_api_client.deploy_inference_model_to_sagemaker(
+            token=config.main.thestage_auth_token,
+            unique_id=unique_id,
+            arn=arn,
+        )
+
+        if not deploy_model_to_sagemaker_response.is_success:
+            typer.echo(__(
+                'Failed to prepare model for deployment with an error: %server_massage%',
+                {'server_massage': deploy_model_to_sagemaker_response.message or ""}
+            ))
+            raise typer.Exit(1)
+
+        model_id = deploy_model_to_sagemaker_response.modelId
+        image_uri = deploy_model_to_sagemaker_response.ecrImageUrl
+        model_uri = deploy_model_to_sagemaker_response.s3ArtifactsUrl
+        region = "us-east-1"
+        sm_client = boto3.client('sagemaker', region_name=region)
+
+        try:
+            container = {
+                "Image": image_uri,
+                "ModelDataUrl": model_uri,
+                "Environment": {
+                    "SAGEMAKER_TRITON_DEFAULT_MODEL_NAME": model_id,
+                    "THESTAGE_API_URL": config.main.thestage_api_url,
+                    "THESTAGE_AUTH_TOKEN": config.main.thestage_auth_token
+                },
+            }
+
+            sm_model_name = f"{unique_id}-{datetime.now().strftime('%Y%m%d-%H%M%S')}"
+            create_model_response = sm_client.create_model(
+                ModelName=sm_model_name,
+                ExecutionRoleArn=arn,
+                PrimaryContainer=container,
+            )
+            typer.echo(f"Model created successfully. Model ARN: {create_model_response['ModelArn']}")
+
+            endpoint_config_name = f"{unique_id}-{datetime.now().strftime('%Y%m%d-%H%M%S')}"
+            create_endpoint_config_response = sm_client.create_endpoint_config(
+                EndpointConfigName=endpoint_config_name,
+                ProductionVariants=[
+                    {
+                        "InstanceType": instance_type,
+                        "InitialVariantWeight": initial_variant_weight,
+                        "InitialInstanceCount": initial_instance_count,
+                        "ModelName": sm_model_name,
+                        "VariantName": "AllTraffic",
+                    }
+                ],
+            )
+            typer.echo(
+                f"Endpoint configuration created successfully. Endpoint Config ARN: {create_endpoint_config_response['EndpointConfigArn']}")
+
+            endpoint_name = f"{unique_id}-{datetime.now().strftime('%Y%m%d-%H%M%S')}"
+            create_endpoint_response = sm_client.create_endpoint(
+                EndpointName=endpoint_name,
+                EndpointConfigName=endpoint_config_name,
+            )
+            typer.echo(f"Endpoint created successfully. Endpoint ARN: {create_endpoint_response['EndpointArn']}")
+
+            typer.echo("Waiting for the endpoint to become active...")
+            while True:
+                resp = sm_client.describe_endpoint(EndpointName=endpoint_name)
+                status = resp["EndpointStatus"]
+                typer.echo(f"Status: {status}")
+                if status == "InService":
+                    break
+                elif status == "Failed":
+                    typer.echo(f"Endpoint creation failed. Reason: {resp.get('FailureReason', 'Unknown')}")
+                    raise typer.Exit(1)
+                time.sleep(60)
+
+            typer.echo(f"Endpoint is ready. ARN: {resp['EndpointArn']} Status: {status}")
+
+        except Exception as e:
+            typer.echo(__("Failed to deploy the inference simulator model to SageMaker: %error%", {"error": str(e)}))
+            raise typer.Exit(1)

{thestage-0.5.36.dist-info → thestage-0.5.38.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: thestage
-Version: 0.5.36
+Version: 0.5.38
 Summary: 
 Author: TheStage AI team
 Author-email: hello@thestage.ai
@@ -11,6 +11,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: aioconsole (>=0.8.0,<0.9.0)
+Requires-Dist: boto3 (>=1.35.80,<2.0.0)
 Requires-Dist: gitpython (>=3.1.40,<4.0.0)
 Requires-Dist: httpx (>=0.27.2,<0.28.0)
 Requires-Dist: paramiko (>=3.4.0,<4.0.0)

{thestage-0.5.36.dist-info → thestage-0.5.38.dist-info}/RECORD

@@ -1,5 +1,5 @@
-thestage/.env,sha256=xWuR3yPUo1cjCJ3gDt16DvCfaeHHwzN2flrH-TUpmiQ,177
-thestage/__init__.py,sha256=9DOcnuLrbhUXkNgjYX3B54sWbTZupOLthYt7Fk7zsxc,65
+thestage/.env,sha256=e0VVlgIiJ9t4WkAN89-LEhNoNvuUpV8_dtOrK2itceo,204
+thestage/__init__.py,sha256=4lvDPyXdqcfjRSHtVEx3UV0ZcUnvMXHqjYpF8GZl50g,65
 thestage/__main__.py,sha256=4ObdWrDRaIASaR06IxtFSsoMu58eyL0MnD64habvPj8,101
 thestage/color_scheme/color_scheme.py,sha256=jzdRCX0hi_XStXi4kvPHVItKlTm7dsD3fHIdeRQLeKw,87
 thestage/config/__init__.py,sha256=RNobilYVK1WAM1utcQ8ZuATKc9Zh9M9BAjCLZTnR_TA,428
@@ -9,7 +9,7 @@ thestage/controllers/base_controller.py,sha256=lX0XsBc7ZEPD_I56cN8IBAVuWGIkOkr7J
 thestage/controllers/config_controller.py,sha256=J08JI56Th1_vuJ71MjgjKnkgJTLoH3mlGtFrDY1oePU,5333
 thestage/controllers/container_controller.py,sha256=QNJbrZMWoOnhufnQNg5gg2fDfo1oVtAWFJTjPo3t-Y0,15445
 thestage/controllers/instance_controller.py,sha256=pFhkO7U2Ta0_1dzskEj8hbE7Izw_7I4SDbq5O5-bfIY,9757
-thestage/controllers/project_controller.py,sha256=BVfX8ZAOHy2fKaxoadqxnxHiMoI06ieg0n3PeIAJhpQ,27123
+thestage/controllers/project_controller.py,sha256=ELEQtWcesiQsJd5h3LjxEvLMKJ0VaX-0tRScSFy4-94,32295
 thestage/controllers/utils_controller.py,sha256=y_4QnjUtLkWG_M_WCpQSzX15kDM92JYFku-4FCmYBXU,1032
 thestage/debug_main.dist.py,sha256=UPIJ58yf-6FtXZj-FLAwxi7HononseuCYm9xb5KlxTs,783
 thestage/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -20,7 +20,7 @@ thestage/entities/enums/shell_type.py,sha256=-XcPiI3psjeIzd0VXWHdca5Ttgbu9M5oQPe
 thestage/entities/enums/tail_output_type.py,sha256=6S55VRRosrI3yZW8XeAGm4u12H4haeiBRvoaExUCWcU,84
 thestage/entities/enums/yes_no_response.py,sha256=PXXkB7KMB-XCHvKOBRuX6NRKyq8UW0G228OICRr2ZTk,86
 thestage/entities/project_inference_simulator.py,sha256=J7Kq3Stb9EElaa8LYLo5kzLLAzgG0ibg4u1iNo0ZUns,597
-thestage/entities/project_inference_simulator_model.py,sha256=3MnWG6Gv3jT8kHa6S2Dzd9X5ZTidbfuiIvmv3LAZZJ4,489
+thestage/entities/project_inference_simulator_model.py,sha256=_xfjaaCSUfVKFQLlsyMEvD4oboKAzIOl6YoMyS59XyQ,546
 thestage/entities/project_task.py,sha256=7uewPIC_HF32OviQHU7uyJ8KokCuK5OjB47r-ltdpyI,672
 thestage/entities/rented_instance.py,sha256=NiOZzOUi6Wt39S5uQBlDodEnMEB1_EA72nT0ZW16eCk,944
 thestage/entities/self_hosted_instance.py,sha256=YyQsuG4PJ8aMMaZyF0wo5Q1tXoeP0BwXA2MvrqEm_rg,761
@@ -46,12 +46,11 @@ thestage/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 thestage/services/abstract_mapper.py,sha256=_q7YLkPNRsNW5wOCqvZIu1KfpLkc7uVaAQKrMZtsGuY,218
 thestage/services/abstract_service.py,sha256=KRsn3xiVhsgX8kESBr4fzDJWbiBWk3LlScWe1uFQ8D8,3958
 thestage/services/app_config_service.py,sha256=1bJM8XzQo9L-NXSoBqSSdgu6yg9v3S0loOc-69055o4,1657
-thestage/services/clients/.DS_Store,sha256=nzq5skCC7S6pAZDF-RZCYmSwpDp-fIu7x4M5OFfbBUA,6148
 thestage/services/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 thestage/services/clients/git/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 thestage/services/clients/git/git_client.py,sha256=tkB7AeesiwOkA2SL2yrYj7Mu0Fgmpx9q56MaXx_we-s,11619
 thestage/services/clients/thestage_api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-thestage/services/clients/thestage_api/api_client.py,sha256=jHASaicVoqIkcBUPD21JMdlKpKTmazqc5fj8ra09emI,29158
+thestage/services/clients/thestage_api/api_client.py,sha256=6KBCd1hEDblqkHAVMSXUPgvSFuJA2jdttTdCg9_1cPk,31466
 thestage/services/clients/thestage_api/dtos/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 thestage/services/clients/thestage_api/dtos/base_response.py,sha256=R6jjhvv0U0XD8j5N5T36q-6liiEAUy7xINFA1H7CMB8,410
 thestage/services/clients/thestage_api/dtos/cloud_provider_region.py,sha256=EJNhQywINW0dtjXKHEDb-etkWQ7keXNUSDhH-h9cabs,1011
@@ -82,6 +81,10 @@ thestage/services/clients/thestage_api/dtos/enums/selfhosted_status.py,sha256=EA
 thestage/services/clients/thestage_api/dtos/enums/task_execution_status.py,sha256=MU14K1PvkLEFnUizeVt8x_u1IP2OVkyf5pLFdrUUtYg,303
 thestage/services/clients/thestage_api/dtos/enums/task_status.py,sha256=Zg4QkuHilWTC6D_lKsby7u4y1rSRs69_kzfeFWCWoFg,282
 thestage/services/clients/thestage_api/dtos/frontend_status.py,sha256=MKrqCmFWVe-8InJqavkqz1w1N35OcT3ndBrXOu7mKdA,312
+thestage/services/clients/thestage_api/dtos/inference_controller/deploy_inference_model_to_instance_request.py,sha256=5R8snv8kMqpyFtxiLKLNotGc-2a4cZYrf6_a8vY5wgg,612
+thestage/services/clients/thestage_api/dtos/inference_controller/deploy_inference_model_to_instance_response.py,sha256=KMq_ySkrQC-sGUB4Znbj2qImjSPHOLKosnsOqJFsMBg,496
+thestage/services/clients/thestage_api/dtos/inference_controller/deploy_inference_model_to_sagemaker_request.py,sha256=5NcSMRSlTon_zg9hI2SxRGd17Y5lRTGP3TXgOercjlw,408
+thestage/services/clients/thestage_api/dtos/inference_controller/deploy_inference_model_to_sagemaker_response.py,sha256=lis4effVtF39eW8gCkWzJKxMVhIMAg3_8JHub83JR6s,422
 thestage/services/clients/thestage_api/dtos/inference_controller/get_inference_simulator_request.py,sha256=syCIJgOYxzm4E53qRVLh4u1ZbSP23_O09WhELMH6QIg,337
 thestage/services/clients/thestage_api/dtos/inference_controller/get_inference_simulator_response.py,sha256=zTt5ewR3oh4qyL0pzt0QP2jANrUpA-5V8As1YqNy3hA,474
 thestage/services/clients/thestage_api/dtos/inference_controller/inference_simulator_list_for_project_request.py,sha256=zsDl-23C2hGZIYwk4aR8wCXP6KDdOObanLsPYkQOplw,508
@@ -129,7 +132,7 @@ thestage/services/config_provider/config_provider.py,sha256=OR61zGI_4bchMolCo0_g
 thestage/services/connect/connect_service.py,sha256=EWX7mBgkb42WAtWcbjHXG4She5iglNCPIY_yx3xwf_k,9488
 thestage/services/connect/dto/remote_server_config.py,sha256=yuO0tTAgUxCiQ-h1nVvWUMlCUtR-WB_eOH6KYspV7zQ,272
 thestage/services/container/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-thestage/services/container/container_service.py,sha256=AwmboxsEB0c6sszXbm9IxRMFodWkKQAmCycSypd5zrA,15076
+thestage/services/container/container_service.py,sha256=OPkhFTm2NiBixCv6JSGVvBb3-6HwOuu3B3hX_fZcxZk,15073
 thestage/services/container/mapper/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 thestage/services/container/mapper/container_mapper.py,sha256=ymIjBLGZnpodh0W2KwNUqRxT0VgM1hsCViNza1nhWHk,1088
 thestage/services/instance/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -144,19 +147,19 @@ thestage/services/logging/logging_constants.py,sha256=4Gk2tglHW_-jnjB8uVIh-ds4fA
 thestage/services/logging/logging_service.py,sha256=B9xxhimmkRGnf4gtViJxTv0XWOWlwQfqj2ct6fviQZI,20457
 thestage/services/project/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 thestage/services/project/dto/inference_simulator_dto.py,sha256=5U4uGp7VC1Yr-T0fqZiSNqZUIybs4J9sV25vjBbAUxI,1312
-thestage/services/project/dto/inference_simulator_model_dto.py,sha256=rvt2UDSLYNHWAb0_vvA3w4C2EeveM58l9fYLE1Jix9Q,1120
+thestage/services/project/dto/inference_simulator_model_dto.py,sha256=j4dT-7cduzLd59QnmnfQt_aFsiUucpyJFGb-9rNx5K8,1172
 thestage/services/project/dto/project_config.py,sha256=xLy1P5fwToRNZ7ktEBzfEmmvSqXlEGeLANqWlC8VV_4,557
 thestage/services/project/mapper/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 thestage/services/project/mapper/project_inference_simulator_mapper.py,sha256=UdOu9IIF5rlNPoWSaaeSKU3ODe8E5uSMgm2V5ywMWKE,812
-thestage/services/project/mapper/project_inference_simulator_model_mapper.py,sha256=wJsiIm6r0IvfiZci5QSytb_NH6dkhda7byTJKblSW1k,841
+thestage/services/project/mapper/project_inference_simulator_model_mapper.py,sha256=PWY0iWbXhvD-G0X0_aQZAFY2bqc0lvRJcQAyC8Y-88Q,869
 thestage/services/project/mapper/project_task_mapper.py,sha256=SHIEXjYwt4vm2B1X2QiI4sCPbBarum0bTOnmTWPOlto,813
-thestage/services/project/project_service.py,sha256=Wg-MIGKIF3eL6O8XGFXJoqk29n7S4akPCGHIXVkiB1E,42092
+thestage/services/project/project_service.py,sha256=ifrP1RrYcHE8U66eumMDvAFZhH4yz4DIbGaWO2_MCxY,50567
 thestage/services/remote_server_service.py,sha256=gOkEEgCcu4lFo__d9-zFQFBYA9IFbrHIZFhfHfIm7ww,23268
 thestage/services/service_factory.py,sha256=dnYc_Ih4TOIYjQInjzf6bpeALXRmWq1TYMIVHvEeONs,5104
 thestage/services/task/dto/task_dto.py,sha256=PJwrUsLLAoO2uA9xvzb27b9iYAoNiBcsHSxKERh2VFo,2335
 thestage/services/validation_service.py,sha256=7rjJlMCduCzHUnOkLL22t3rr4mr-XudCtyWzZKnCo7A,1621
-thestage-0.5.36.dist-info/LICENSE.txt,sha256=U9QrxfdD7Ie7r8z1FleuvOGQvgCF1m0Mjd78cFvWaHE,572
-thestage-0.5.36.dist-info/METADATA,sha256=hP5I0XeWi72viBzqKOX4Wkk13QCujsK_hAlTLwKcdLY,5506
-thestage-0.5.36.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-thestage-0.5.36.dist-info/entry_points.txt,sha256=57pMhs8zaCM-jgeTffC0WVqCsh35Uq_dUDmzXR80CI4,47
-thestage-0.5.36.dist-info/RECORD,,
+thestage-0.5.38.dist-info/LICENSE.txt,sha256=U9QrxfdD7Ie7r8z1FleuvOGQvgCF1m0Mjd78cFvWaHE,572
+thestage-0.5.38.dist-info/METADATA,sha256=RUISssQQGMvnUff6aA0-FTx5K5nYMguGKE9Us2_lsBc,5546
+thestage-0.5.38.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+thestage-0.5.38.dist-info/entry_points.txt,sha256=57pMhs8zaCM-jgeTffC0WVqCsh35Uq_dUDmzXR80CI4,47
+thestage-0.5.38.dist-info/RECORD,,

{thestage-0.5.36.dist-info → thestage-0.5.38.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.0
+Generator: poetry-core 1.8.1
 Root-Is-Purelib: true
 Tag: py3-none-any