thestage 0.6.6__py3-none-any.whl → 0.6.7__py3-none-any.whl

thestage/__init__.py

@@ -1,3 +1,3 @@
 from . import *
 __app_name__ = "thestage"
-__version__ = "0.6.6"
+__version__ = "0.6.7"

thestage/controllers/container_controller.py

@@ -2,9 +2,12 @@
 import re
 from pathlib import Path
 from typing import Optional, List
+from rich import print
 
 from thestage.cli_command import CliCommand
 from thestage.cli_command_helper import get_command_metadata, check_command_permission
+from thestage.color_scheme.color_scheme import ColorScheme
+from thestage.services.clients.thestage_api.api_client import TheStageApiClient
 from thestage.services.clients.thestage_api.dtos.enums.container_pending_action import DockerContainerAction
 from thestage.services.clients.thestage_api.dtos.container_response import DockerContainerDto
 from thestage.i18n.translation import __
@@ -215,7 +218,7 @@ def container_connect(
 @app.command(name="upload", no_args_is_help=True, help=__("Upload file to container"), **get_command_metadata(CliCommand.CONTAINER_UPLOAD))
 def upload_file(
         source_path: str = typer.Argument(help=__("Source file path"),),
-        destination: Optional[str] = typer.Argument(help=__("Destination directory path in container. Format: container_name:/path/to/file"),),
+        destination: Optional[str] = typer.Argument(help=__("Destination directory path in container. Format: container_id_or_name:/path/to/file"),),
         username: Optional[str] = typer.Option(
             None,
             '--username',
@@ -228,50 +231,22 @@ def upload_file(
     app_logger.info(f'Running {command_name} from {get_current_directory()}')
     check_command_permission(command_name)
 
-    container_args = re.match(r"^([\w\W]+?):([\w\W]+)$", destination)
-
-    if container_args is None:
-        typer.echo(__('Container name and source file path are required as the second argument'))
-        typer.echo(__('Example: container_uid:/path/to/file'))
-        raise typer.Exit(1)
-    container_slug = container_args.groups()[0]
-    destination_path = container_args.groups()[1].rstrip("/")
-
-    if not container_slug:
-        typer.echo(__('Container name is required'))
-        raise typer.Exit(1)
-
     service_factory = validate_config_and_get_service_factory()
     container_service: ContainerService = service_factory.get_container_service()
 
-    container: Optional[DockerContainerDto] = container_service.get_container(
-        container_slug=container_slug,
+    container_service.put_file_to_container(
+        source_path=source_path,
+        destination=destination,
+        username_param=username,
     )
 
-    if container:
-        container_service.check_if_container_running(
-            container=container
-        )
-
-        typer.echo(f"Uploading file(s) to container '{container_slug}'")
-
-        container_service.put_file_to_container(
-            container=container,
-            src_path=source_path,
-            destination_path=destination_path,
-            username_param=username,
-            copy_only_folder_contents=source_path.endswith("/")
-        )
-    else:
-        typer.echo(__("Container not found: %container_item%", {'container_item': container_slug}))
-
     app_logger.info(f'End send files to container')
     raise typer.Exit(0)
 
 
 @app.command(name="download", no_args_is_help=True, help=__("Download file from container"), **get_command_metadata(CliCommand.CONTAINER_DOWNLOAD))
 def download_file(
-        source_path: str = typer.Argument(help=__("Source file path in container. Format: container_name:/path/to/file"),),
+        source: str = typer.Argument(help=__("Source file path in container. Format: container_name:/path/to/file"),),
         destination_path: str = typer.Argument(help=__("Destination directory path on local machine"),),
         username: Optional[str] = typer.Option(
             None,
@@ -285,43 +260,15 @@ def download_file(
     app_logger.info(f'Running {command_name} from {get_current_directory()}')
     check_command_permission(command_name)
 
-    container_args = re.match(r"^([\w\W]+?):([\w\W]+)$", source_path)
-
-    if container_args is None:
-        typer.echo(__('Container name and source directory path are required as the first argument'))
-        typer.echo(__('Example: container-uid:/path/to/file'))
-        raise typer.Exit(1)
-    container_slug = container_args.groups()[0]
-    source_path = container_args.groups()[1]
-
-    if not container_slug:
-        typer.echo(__('Container name is required'))
-        raise typer.Exit(1)
-
     service_factory = validate_config_and_get_service_factory()
     container_service: ContainerService = service_factory.get_container_service()
 
-    container: Optional[DockerContainerDto] = container_service.get_container(
-        container_slug=container_slug,
+    container_service.get_file_from_container(
+        source=source,
+        destination_path=destination_path.rstrip("/"),
+        username_param=username,
     )
 
-    if container:
-        container_service.check_if_container_running(
-            container=container
-        )
-
-        typer.echo(f"Downloading files from container: '{container_slug}'")
-
-        container_service.get_file_from_container(
-            container=container,
-            src_path=source_path,
-            destination_path=destination_path.rstrip("/"),
-            username_param=username,
-            copy_only_folder_contents=source_path.endswith("/"),
-        )
-    else:
-        typer.echo(f"Container not found: {container_slug}")
-
     app_logger.info(f'End download files from container')
     raise typer.Exit(0)
 

thestage/controllers/instance_controller.py

@@ -66,14 +66,14 @@ def rented_list(
 def instance_connect(
         public_id: Optional[str] = typer.Option(
             None,
-            '--id',
+            '--rented-instance-id',
             '-rid',
             help="Rented instance ID",
             is_eager=False,
         ),
         slug: Optional[str] = typer.Option(
             None,
-            '--name',
+            '--rented-instance-name',
             '-rn',
             help="Rented instance name",
             is_eager=False,
@@ -156,14 +156,14 @@ def self_hosted_list(
 def self_hosted_connect(
         public_id: Optional[str] = typer.Option(
             None,
-            '--id',
+            '--self-hosted-instance-id',
             '-sid',
             help="Self-hosted instance ID",
             is_eager=False,
         ),
         slug: Optional[str] = typer.Option(
             None,
-            '--name',
+            '--self-hosted-instance-name',
             '-sn',
             help="Self-hosted instance name",
             is_eager=False,

thestage/controllers/project_controller.py

@@ -63,7 +63,8 @@ def clone(
         raise typer.Exit(1)
 
     if not working_directory:
-        working_directory = get_current_directory().joinpath(project_slug)
+        project_dir_name = project_public_id if project_slug is None else project_slug
+        working_directory = get_current_directory().joinpath(project_dir_name)
 
     service_factory = validate_config_and_get_service_factory(working_directory=working_directory)
     project_service = service_factory.get_project_service()
@@ -842,7 +843,7 @@ def inference_simulator_logs(
         ),
         slug: Optional[str] = typer.Option(
             None,
-            '--name',
+            '--inference-simulator-name',
             '-isn',
             help="Inference simulator name",
             is_eager=False,
@@ -952,17 +953,14 @@ def deploy_inference_simulator_model_to_instance(
     check_command_permission(command_name)
 
     if model_slug and not re.match(r"^[a-zA-Z0-9-]+$", model_slug):
-        raise typer.BadParameter(__("Invalid UID format. The UID can only contain letters, numbers, and hyphens."))
-
-    new_inference_simulator_slug = f"{model_slug}-{int(time.time())}"
+        raise typer.BadParameter("Invalid name format. Name can only contain letters, numbers, and hyphens.")
 
     service_factory = validate_config_and_get_service_factory(working_directory=working_directory)
     project_service = service_factory.get_project_service()
 
-    project_service.project_deploy_inference_simulator_model_to_instance(
+    inference_simulator_public_id = project_service.project_deploy_inference_simulator_model_to_instance(
         model_public_id=model_public_id,
         model_slug=model_slug,
-        new_inference_simulator_slug=new_inference_simulator_slug,
         rented_instance_public_id=rented_instance_public_id,
         rented_instance_slug=rented_instance_slug,
         self_hosted_instance_public_id=self_hosted_instance_public_id,
@@ -973,7 +971,7 @@ def deploy_inference_simulator_model_to_instance(
         logging_service: LoggingService = service_factory.get_logging_service()
 
         logging_service.stream_inference_simulator_logs_with_controls(
-            slug=new_inference_simulator_slug
+            public_id=inference_simulator_public_id
         )
     raise typer.Exit(0)
 
@@ -982,14 +980,14 @@ def deploy_inference_simulator_model_to_instance(
 def deploy_inference_simulator_model_to_sagemaker(
         model_public_id: Optional[str] = typer.Option(
             None,
-            '--id',
+            '--model-id',
             '-mid',
             help="Inference simulator model ID",
             is_eager=False,
         ),
         model_slug: Optional[str] = typer.Option(
             None,
-            '--name',
+            '--model-name',
             '-mn',
             help="Inference simulator model name",
             is_eager=False,

thestage/entities/project_inference_simulator_model.py

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

thestage/helpers/error_handler.py

@@ -106,8 +106,8 @@ def error_handler() -> Callable:
                     raise e100
                 else:
                     typer.echo(__('Undefined error occurred'))
-                    typer.echo(e100.__class__.__name__)
-                    print(traceback.format_exc())
+                    # typer.echo(e100.__class__.__name__)
+                    # print(traceback.format_exc())
                     # TODO send all exceptions to backend?
                     app_logger.error(f'{traceback.format_exc()}')
                     raise typer.Exit(1)

thestage/main.py

@@ -19,7 +19,7 @@ def main():
             config_provider.update_allowed_commands_and_is_token_valid(validate_token_response=token_info)
         except Exception as e:
             app_logger.error(f'{traceback.format_exc()}')
-            print('Unexpected error occurred')  # TODO inquire what we want here if backend is offline
+            print('Error connecting to TheStage servers')  # TODO inquire what we want here if backend is offline
             print(f'Application logs path: {str(get_log_path_from_os())}')
             return
 

thestage/services/clients/git/git_client.py

@@ -196,7 +196,7 @@ class GitLocalClient:
     ) -> Optional[str]:
         repo = self.__get_repo(path=path)
         if repo.head.is_detached:
-            line_color = ColorScheme.GIT_HEADLESS
+            line_color = ColorScheme.GIT_HEADLESS.value
             print(f'[{line_color}]Committing in detached head state at {repo.head.commit.hexsha}[/{line_color}]')
         commit_name = name if name else f"Auto commit {str(datetime.datetime.now().date())}"
         commit = repo.git.commit('--allow-empty', '-m', commit_name, )

thestage/services/clients/thestage_api/api_client.py

@@ -667,7 +667,6 @@ class TheStageApiClient(TheStageApiClientCore):
             self,
             model_public_id: str,
             model_slug: str,
-            new_inference_simulator_slug: str,
             rented_instance_public_id: Optional[str] = None,
             rented_instance_slug: Optional[str] = None,
             self_hosted_instance_public_id: Optional[str] = None,
@@ -675,10 +674,9 @@ class TheStageApiClient(TheStageApiClientCore):
 
     ) -> Optional[DeployInferenceModelToInstanceResponse]:
         request = DeployInferenceModelToInstanceRequest(
-            inferenceSimulatorSlug=new_inference_simulator_slug,
             modelPublicId=model_public_id,
             modelSlug=model_slug,
-            instancePublicId=rented_instance_public_id,
+            instanceRentedPublicId=rented_instance_public_id,
             instanceRentedSlug=rented_instance_slug,
             selfhostedInstancePublicId=self_hosted_instance_public_id,
             selfhostedInstanceSlug=self_hosted_instance_slug
@@ -735,7 +733,7 @@ class TheStageApiClient(TheStageApiClientCore):
         return result
 
 
-    def resolve_connect_options(
+    def resolve_user_input(
             self,
             entity_identifier: str
     ) -> Optional[ConnectResolveOptionsResponse]:
@@ -745,7 +743,7 @@ class TheStageApiClient(TheStageApiClientCore):
 
         response = self._request(
             method='POST',
-            url='/user-api/v1/connect/resolve-options',
+            url='/user-api/v1/resolve-user-input',
             data=data,
             token=self.__config_provider.get_config().main.thestage_auth_token,
         )

thestage/services/clients/thestage_api/dtos/base_controller/connect_resolve_response.py

@@ -11,6 +11,7 @@ class EntityMatchData(BaseModel):
     publicId: str = Field(alias='publicId')
     frontendStatus: FrontendStatusDto = Field(alias='frontendStatus')
     canConnect: bool = Field(alias='canConnect')
+    canDownloadUploadOnContainer: bool = Field(alias='canDownloadUploadOnContainer')
     matchedField: Optional[str] = Field(None, alias='matchedField')
 
 

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

@@ -16,5 +16,3 @@ class DeployInferenceModelToInstanceRequest(BaseModel):
 
     selfhostedInstancePublicId: Optional[str] = Field(None, alias='selfhostedInstancePublicId')
     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

@@ -9,4 +9,6 @@ from thestage.services.project.dto.inference_simulator_model_dto import Inferenc
 class DeployInferenceModelToInstanceResponse(TheStageBaseResponse):
     model_config = ConfigDict(use_enum_values=True)
 
+    inferenceSimulatorPublicId: Optional[str] = Field(None, alias='inferenceSimulatorPublicId')
+
 

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

@@ -8,5 +8,5 @@ from thestage.services.clients.thestage_api.dtos.entity_filter_request import En
 class GetInferenceSimulatorRequest(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
 
-    public_id: Optional[str] = Field(None, alias='publicId')
+    publicId: Optional[str] = Field(None, alias='publicId')
     slug: Optional[str] = Field(None, alias='slug')

thestage/services/clients/thestage_api/dtos/project_controller/project_run_task_response.py

@@ -1,3 +1,5 @@
+from typing import Optional, List
+
 from pydantic import Field, ConfigDict
 
 from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
@@ -8,3 +10,4 @@ class ProjectRunTaskResponse(TheStageBaseResponse):
     model_config = ConfigDict(use_enum_values=True)
 
     task: TaskDto = Field(None, alias='task')
+    tasksInQueue: Optional[List[TaskDto]] = Field(None, alias='tasksInQueue')

thestage/services/connect/connect_service.py

@@ -45,7 +45,7 @@ class ConnectService(AbstractService):
             username: Optional[str],
             private_key_path: Optional[str],
     ):
-        resolved_options = self.__thestage_api_client.resolve_connect_options(entity_identifier=input_entity_identifier)
+        resolved_options = self.__thestage_api_client.resolve_user_input(entity_identifier=input_entity_identifier)
         entities_available_for_connect_count = 0
         task_presence = False
         container_presence = False
@@ -56,7 +56,7 @@ class ConnectService(AbstractService):
         if resolved_options.taskMatchData:
             for task_item in resolved_options.taskMatchData:
                 message = f"Found a task with with matching {task_item.matchedField} in status: '{task_item.frontendStatus.status_translation}' (ID: {task_item.publicId})"
-                line_color = ColorScheme.SUCCESS if task_item.canConnect else 'default'
+                line_color = ColorScheme.SUCCESS.value if task_item.canConnect else 'default'
                 print(f"[{line_color}]{message}[{line_color}]")
                 if task_item.canConnect:
                     task_presence = True
@@ -66,7 +66,7 @@ class ConnectService(AbstractService):
         if resolved_options.dockerContainerMatchData:
             for container_item in resolved_options.dockerContainerMatchData:
                 message = f"Found a container with matching {container_item.matchedField} in status: '{container_item.frontendStatus.status_translation}' (ID: {container_item.publicId})"
-                line_color = ColorScheme.SUCCESS if container_item.canConnect else 'default'
+                line_color = ColorScheme.SUCCESS.value if container_item.canConnect else 'default'
                 print(f"[{line_color}]{message}[{line_color}]")
                 if container_item.canConnect:
                     container_presence = True
@@ -76,7 +76,7 @@ class ConnectService(AbstractService):
         if resolved_options.instanceRentedMatchData:
             for instance_rented_item in resolved_options.instanceRentedMatchData:
                 message = f"Found a rented instance with matching {instance_rented_item.matchedField} in status: '{instance_rented_item.frontendStatus.status_translation}'  (ID: {instance_rented_item.publicId})"
-                line_color = ColorScheme.SUCCESS if instance_rented_item.canConnect else 'default'
+                line_color = ColorScheme.SUCCESS.value if instance_rented_item.canConnect else 'default'
                 print(f"[{line_color}]{message}[{line_color}]")
 
                 if instance_rented_item.canConnect:
@@ -87,7 +87,7 @@ class ConnectService(AbstractService):
         if resolved_options.selfhostedInstanceMatchData:
             for selfhosted_item in resolved_options.selfhostedInstanceMatchData:
                 message = f"Found a self-hosted instance with matching {selfhosted_item.matchedField} in status: '{selfhosted_item.frontendStatus.status_translation}' (ID: {selfhosted_item.publicId})"
-                line_color = ColorScheme.SUCCESS if selfhosted_item.canConnect else 'default'
+                line_color = ColorScheme.SUCCESS.value if selfhosted_item.canConnect else 'default'
                 print(f"[{line_color}]{message}[{line_color}]")
 
                 if selfhosted_item.canConnect:
@@ -153,7 +153,7 @@ class ConnectService(AbstractService):
 
             # if no instances found - exit 1
             if instance_rented is None:
-                typer.echo(f"No rented instance found with matching unique ID '{instance_slug}'")
+                typer.echo(f"No rented instance found with matching identifier")
                 raise typer.Exit(1)
 
         note_to_send: Optional[str] = None

thestage/services/container/container_service.py

@@ -1,7 +1,11 @@
+import re
 from pathlib import Path
 from typing import List, Tuple, Optional, Dict
+from rich import print
 
 import typer
+
+from thestage.color_scheme.color_scheme import ColorScheme
 from thestage.entities.container import DockerContainerEntity
 from thestage.services.clients.thestage_api.dtos.container_param_request import DockerContainerActionRequestDto
 from thestage.services.clients.thestage_api.dtos.enums.container_pending_action import DockerContainerAction
@@ -150,7 +154,7 @@ class ContainerService(AbstractService):
             if private_key_path:
                 typer.echo(f'Using configured private key for {ip_address}: {private_key_path}')
             else:
-                typer.echo(f'Using SSH agent to connect to {ip_address}')
+                typer.echo(f'Using SSH agent to connect to {ip_address} as {username}')
         else:
             self.__config_provider.update_remote_server_config_entry(ip_address, Path(private_key_path))
             typer.echo(f'Updated private key path for {ip_address}: {private_key_path}')
@@ -267,22 +271,49 @@ class ContainerService(AbstractService):
     @error_handler()
     def put_file_to_container(
             self,
-            container: DockerContainerDto,
-            src_path: str,
-            copy_only_folder_contents: bool,
-            destination_path: Optional[str] = None,
-            username_param: Optional[str] = None,
+            source_path: str,
+            destination: str,
+            username_param: Optional[str],
     ):
-        if not self.__file_system_service.check_if_path_exist(file=src_path):
-            typer.echo(__("File not found at specified path"))
+        container_args = re.match(r"^([\w\W]+?):([\w\W]+)$", destination)
+        if container_args is None:
+            typer.echo(__('Container name and source file path are required as the second argument'))
+            typer.echo(__('Example: container_name:/path/to/file'))
             raise typer.Exit(1)
+        container_identifier = container_args.groups()[0]
+        destination_path = container_args.groups()[1].rstrip("/")
 
-        username, ip_address, private_key_path = self.get_server_auth(
-            container=container,
-            username_param=username_param,
-            private_key_path_override=None
+        if not container_identifier:
+            typer.echo('Container identifier (container_id_or_name) is required')
+            raise typer.Exit(1)
+
+        resolved_options = self.__thestage_api_client.resolve_user_input(entity_identifier=container_identifier)
+        container_public_id = None
+        valid_container_count = 0
+        for container_item in resolved_options.dockerContainerMatchData:
+            message = f"Found a container with matching {container_item.matchedField} in status: '{container_item.frontendStatus.status_translation}' (ID: {container_item.publicId})"
+            line_color = ColorScheme.SUCCESS.value if container_item.canDownloadUploadOnContainer else 'default'
+            print(f"[{line_color}]{message}[{line_color}]")
+            if container_item.canDownloadUploadOnContainer:
+                valid_container_count += 1
+                container_public_id = container_item.publicId
+
+        if valid_container_count != 1:
+            typer.echo(f"Failed to resolve the container by provided identifier, as total of {valid_container_count} containers are valid options")
+            raise typer.Exit(1)
+
+        container: Optional[DockerContainerDto] = self.__thestage_api_client.get_container(
+            container_public_id=container_public_id,
         )
 
+        if not container:
+            typer.echo(f"Unexpected error: container '{container_public_id}' not found")
+            raise typer.Exit(1)
+
+        if not self.__file_system_service.check_if_path_exist(file=source_path):
+            typer.echo(__("File not found at specified path"))
+            raise typer.Exit(1)
+
         if not container.mappings or not container.mappings.directory_mappings:
             typer.echo(__("Mapping folders not found"))
             raise typer.Exit(1)
@@ -296,10 +327,18 @@ class ContainerService(AbstractService):
             typer.echo(__("Cannot find matching container volume mapping for specified file path"))
             raise typer.Exit(1)
 
+        username, ip_address, private_key_path = self.get_server_auth(
+            container=container,
+            username_param=username_param,
+            private_key_path_override=None
+        )
+
+        copy_only_folder_contents = source_path.endswith("/")
+
         self.__remote_server_service.upload_data_to_container(
             ip_address=ip_address,
             username=username,
-            src_path=src_path,
+            src_path=source_path,
             dest_path=destination_path,
             instance_path=instance_path,
             container_path=container_path,
@@ -310,31 +349,67 @@ class ContainerService(AbstractService):
     @error_handler()
     def get_file_from_container(
             self,
-            container: DockerContainerDto,
-            src_path: str,
-            copy_only_folder_contents: bool,
-            destination_path: Optional[str] = None,
+            source: str,
+            destination_path: str,
             username_param: Optional[str] = None,
     ):
-        username, ip_address, private_key_path = self.get_server_auth(
-            container=container,
-            username_param=username_param,
-            private_key_path_override=None,
+        container_args = re.match(r"^([\w\W]+?):([\w\W]+)$", source)
+
+        if container_args is None:
+            typer.echo(__('Container name and source directory path are required as the first argument'))
+            typer.echo(__('Example: container_name:/path/to/file'))
+            raise typer.Exit(1)
+        container_identifier = container_args.groups()[0]
+        source_path = container_args.groups()[1]
+
+        if not container_identifier:
+            typer.echo('Container identifier (container_id_or_name) is required')
+            raise typer.Exit(1)
+
+        resolved_options = self.__thestage_api_client.resolve_user_input(entity_identifier=container_identifier)
+        container_public_id = None
+        valid_container_count = 0
+        for container_item in resolved_options.dockerContainerMatchData:
+            message = f"Found a container with matching {container_item.matchedField} in status: '{container_item.frontendStatus.status_translation}' (ID: {container_item.publicId})"
+            line_color = ColorScheme.SUCCESS.value if container_item.canDownloadUploadOnContainer else 'default'
+            print(f"[{line_color}]{message}[{line_color}]")
+            if container_item.canDownloadUploadOnContainer:
+                valid_container_count += 1
+                container_public_id = container_item.publicId
+
+        if valid_container_count != 1:
+            typer.echo(f"Failed to resolve the container by provided identifier, as total of {valid_container_count} containers are valid options")
+            raise typer.Exit(1)
+
+        container: Optional[DockerContainerDto] = self.__thestage_api_client.get_container(
+            container_public_id=container_public_id,
         )
 
+        if not container:
+            typer.echo(f"Unexpected error: container '{container_public_id}' not found")
+            raise typer.Exit(1)
+
         if not container.mappings or not container.mappings.directory_mappings:
             typer.echo(__("Mapping folders not found"))
             raise typer.Exit(1)
 
         instance_path, container_path = self._get_new_path_from_mapping(
             directory_mapping=container.mappings.directory_mappings,
-            destination_path=src_path,
+            destination_path=source_path,
         )
 
         if not instance_path and not container_path:
             typer.echo(__("Cannot find matching container volume mapping for specified file path"))
             raise typer.Exit(1)
 
+        username, ip_address, private_key_path = self.get_server_auth(
+            container=container,
+            username_param=username_param,
+            private_key_path_override=None,
+        )
+
+        copy_only_folder_contents=source_path.endswith("/")
+
         self.__remote_server_service.download_data_from_container(
             ip_address=ip_address,
             username=username,

thestage/services/logging/logging_service.py

@@ -162,15 +162,10 @@ class LoggingService:
                        }))
                 raise typer.Exit(1)
         else:
-            typer.echo(__("Inference simulator with unique ID %slug% was not found", {'slug': inference_simulator.slug}))
+            typer.echo("Inference simulator was not found")
             raise typer.Exit(1)
 
-        typer.echo(__(
-            f"Log stream for inference simulator %slug% started",
-            {
-                'slug': str(inference_simulator.slug),
-            }
-        ))
+        typer.echo(f"Log stream for inference simulator '{inference_simulator.public_id}' started")
 
         typer.echo(__("CTRL+D to disconnect from log stream."))
 
@@ -184,7 +179,8 @@ class LoggingService:
 
         if input_task in done:
             print_task_or_inference_simulator_logs.cancel()
-            typer.echo(__(f"Disconnected from log stream. You can try to reconnect with 'thestage project inference-simulator logs {slug}'."))
+            inference_logs_cmd_id = f"-isn {slug}" if slug else f"-isid {public_id}"
+            typer.echo(f"Disconnected from log stream. You can try to reconnect with 'thestage project inference-simulator logs {inference_logs_cmd_id}'")
 
 
     @error_handler()
@@ -285,6 +281,7 @@ class LoggingService:
                     consecutive_error_count = 0
                     errors_started_at = None
                     log_wait_remaining_limit = 0   # no log delays after reconnect
+                    print_nonblocking("Reconnected to log stream", writer, BytePrintStyle.GREEN)
 
                 last_iteration_log_timestamp = logs_response.lastLogTimestamp
                 last_log_id = logs_response.lastLogId

thestage/services/project/dto/project_config.py

@@ -10,6 +10,5 @@ class ProjectConfig(BaseModel):
     public_id: str = Field(None, alias='public_id')
     git_repository_url: str = Field(None, alias='git_repository_url')
     deploy_key_path: str = Field(None, alias='deploy_key_path')
-    default_container_uid: Optional[str] = Field(None, alias='default_container_uid') # TODO delete
-    default_container_public_id: Optional[str] = Field(None, alias='default_container_public_id') # TODO use
+    default_container_public_id: Optional[str] = Field(None, alias='default_container_public_id')
     prompt_for_default_container: bool = Field(True, alias='prompt_for_default_container')

thestage/services/project/project_service.py

@@ -278,7 +278,7 @@ class ProjectService(AbstractService):
 
         if (project_config.prompt_for_default_container is None or project_config.prompt_for_default_container) and (docker_container_slug or docker_container_public_id) and (project_config.default_container_public_id != container.public_id):
             set_default_container_answer: str = typer.prompt(
-                text=f"Would you like to set '{docker_container_slug}' as a default container for this project installation?",
+                text=f"Would you like to set the container '{container.slug}' (ID: '{container.public_id}') as default for this project installation?",
                 show_choices=True,
                 default=YesOrNoResponse.YES.value,
                 type=click.Choice([r.value for r in YesOrNoResponse]),
@@ -415,6 +415,10 @@ class ProjectService(AbstractService):
                 print(f"[{ColorScheme.WARNING.value}]{run_task_response.message}[{ColorScheme.WARNING.value}]")
             if run_task_response.is_success and run_task_response.task:
                 typer.echo(f"Task '{run_task_response.task.title}' has been scheduled successfully. Task ID: {run_task_response.task.public_id}")
+                if run_task_response.tasksInQueue:
+                    typer.echo(f"There are tasks in queue ahead of this new task:")
+                    for queued_task_item in run_task_response.tasksInQueue:
+                        typer.echo(f"{queued_task_item.public_id} - {queued_task_item.frontend_status.status_translation}")
                 return run_task_response.task
             else:
                 typer.echo(f'The task failed with an error: {run_task_response.message}')
@@ -630,14 +634,14 @@ class ProjectService(AbstractService):
 
         if metadata:
             typer.echo("qlip_serve_metadata:")
-            typer.echo(json.dumps(json.loads(metadata), indent=4))
+            typer.echo(json.dumps(metadata, indent=4))
 
             if not file_path:
                 file_path = Path(os.getcwd()) / "metadata.json"
                 typer.echo(__("No file path provided. Saving metadata to %file_path%", {"file_path": str(file_path)}))
 
             try:
-                parsed_metadata = json.loads(metadata)
+                parsed_metadata = metadata
 
                 output_file = Path(file_path)
                 output_file.parent.mkdir(parents=True, exist_ok=True)
@@ -935,12 +939,11 @@ class ProjectService(AbstractService):
             self,
             model_public_id: Optional[str] = None,
             model_slug: Optional[str] = None,
-            new_inference_simulator_slug: Optional[str] = None,
             rented_instance_public_id: Optional[str] = None,
             rented_instance_slug: Optional[str] = None,
             self_hosted_instance_public_id: Optional[str] = None,
             self_hosted_instance_slug: Optional[str] = None,
-    ) -> None:
+    ) -> str:
         config = self.__config_provider.get_config()
         project_config: ProjectConfig = self.__get_fixed_project_config()
         if not project_config:
@@ -959,11 +962,10 @@ class ProjectService(AbstractService):
             typer.echo("Please provide a single model identifier - name or ID.")
             raise typer.Exit(1)
 
-        typer.echo(f"Creating inference simulator with unique ID: {new_inference_simulator_slug}")
+        typer.echo(f"Creating inference simulator")
         deploy_model_to_instance_response: DeployInferenceModelToInstanceResponse = self.__thestage_api_client.deploy_inference_model_to_instance(
             model_public_id=model_public_id,
             model_slug=model_slug,
-            new_inference_simulator_slug=new_inference_simulator_slug,
             rented_instance_public_id=rented_instance_public_id,
             rented_instance_slug=rented_instance_slug,
             self_hosted_instance_public_id=self_hosted_instance_public_id,
@@ -973,7 +975,7 @@ class ProjectService(AbstractService):
             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.")
+                typer.echo(f"Inference simulator '{deploy_model_to_instance_response.inferenceSimulatorPublicId}' has been scheduled to run successfully.")
             else:
                 typer.echo(__(
                     'Inference simulator failed to run with an error: %server_massage%',
@@ -984,6 +986,8 @@ class ProjectService(AbstractService):
             typer.echo(__("Inference simulator failed to run with an error"))
             raise typer.Exit(1)
 
+        return deploy_model_to_instance_response.inferenceSimulatorPublicId
+
 
     @error_handler()
     def project_deploy_inference_simulator_model_to_sagemaker(
@@ -1142,7 +1146,7 @@ class ProjectService(AbstractService):
         if not project_public_id and not project_slug:
             project_config: ProjectConfig = self.__config_provider.read_project_config()
             if not project_config:
-                typer.echo(__("Provide the project unique ID or run this command from within an initialized project directory"))
+                typer.echo(__("Provide the project identifier or run this command from within an initialized project directory"))
                 raise typer.Exit(1)
             project_public_id = project_config.public_id
 
@@ -1195,7 +1199,7 @@ class ProjectService(AbstractService):
         if not project_public_id and not project_slug:
             project_config: ProjectConfig = self.__config_provider.read_project_config()
             if not project_config:
-                typer.echo(__("Provide the project unique ID or run this command from within an initialized project directory"))
+                typer.echo(__("Provide the project identifier or run this command from within an initialized project directory"))
                 raise typer.Exit(1)
             project_public_id = project_config.public_id
 
@@ -1238,7 +1242,7 @@ class ProjectService(AbstractService):
             headers=list(map(lambda x: x.alias, ProjectInferenceSimulatorModelEntity.model_fields.values())),
             row=row,
             page=page,
-            max_col_width=[100, 100, 100, 100, 100, 100, 100, 100],
+            max_col_width=[100, 100, 100, 100, 25],
             show_index="never",
         )
 
@@ -1247,7 +1251,7 @@ class ProjectService(AbstractService):
         if not project_slug and not project_public_id:
             project_config: ProjectConfig = self.__config_provider.read_project_config()
             if not project_config:
-                typer.echo(__("Provide the project unique ID or run this command from within an initialized project directory"))
+                typer.echo(__("Provide the project identifier or run this command from within an initialized project directory"))
                 raise typer.Exit(1)
             project_public_id = project_config.public_id
 

{thestage-0.6.6.dist-info → thestage-0.6.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: thestage
-Version: 0.6.6
+Version: 0.6.7
 Summary: 
 License-File: LICENSE.txt
 Author: TheStage AI team
@@ -21,6 +21,7 @@ Requires-Dist: pydantic (>=2.4.2,<3.0.0)
 Requires-Dist: python-dotenv (>=1.0.0,<2.0.0)
 Requires-Dist: python-gettext-translations (>=1.1.0,<2.0.0)
 Requires-Dist: requests (>=2.31.0,<3.0.0)
+Requires-Dist: rich (>=14.2.0,<15.0.0)
 Requires-Dist: tabulate (>=0.9.0,<0.10.0)
 Requires-Dist: typer[all] (>=0.15.2,<0.16.0)
 Description-Content-Type: text/markdown

{thestage-0.6.6.dist-info → thestage-0.6.7.dist-info}/RECORD

@@ -1,5 +1,5 @@
 thestage/.env,sha256=lHeR-H4TiBp4z06bWt7oND9LMMVXP3WCKY7lQltDM3M,129
-thestage/__init__.py,sha256=IO0lPlZY3Ps8mQl7mW6sfOiw93IVvBGApb2QecBNa6U,64
+thestage/__init__.py,sha256=h6612fk--j1jt0iLyVGKNfXLdDLrKmuH0Vay8qaYWMY,64
 thestage/__main__.py,sha256=4ObdWrDRaIASaR06IxtFSsoMu58eyL0MnD64habvPj8,101
 thestage/cli_command.py,sha256=P93LXDruJWecmF5xn-D2uhMOSD9iSJl-gP4oG9aD56k,2173
 thestage/cli_command_helper.py,sha256=f5NZQ4MZJ7yyxPvkQW7ilIAxN_19Mx_zzSqedhZtd6g,1939
@@ -10,9 +10,9 @@ thestage/config/env_base.py,sha256=RNBQ17yk1ieu1kdUlM7Qe7mDCoxstgGUwwhe265o4dQ,3
 thestage/controllers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 thestage/controllers/base_controller.py,sha256=BjTWHPynPrOcD0Jb3h6bqeIMSZxy6Elja3Rdgl93mXU,2508
 thestage/controllers/config_controller.py,sha256=WEw4H5ZfJfUhfBimLa1Lava-h8zp2TZ4GyHVyzeTMdo,5964
-thestage/controllers/container_controller.py,sha256=oAFDWelChvRRIWTeeOMVufIOQXfqSS9NnF8kqEnhrNk,18244
-thestage/controllers/instance_controller.py,sha256=147zqteL_F8lhF4blgRLJHemFnhQ_EGsK2nbSxjV26c,7396
-thestage/controllers/project_controller.py,sha256=8rahTBA6Xt-nlSVBIyeMCOrsakDYgK6rM3KHjonQAr0,40509
+thestage/controllers/container_controller.py,sha256=nQCLBR4wqvNfXuxU-mdJYCvFSbAL4LweEzPTp0FKwbs,16337
+thestage/controllers/instance_controller.py,sha256=EmXs5BLgvocSN7PuNPe3Yo4ZgZoARSh0aKmFMTbvsEc,7470
+thestage/controllers/project_controller.py,sha256=00thKf4p_SDHYF9eRmipqzE7MHogp3DvgdxNAaMt_zg,40526
 thestage/controllers/utils_controller.py,sha256=ac-TLWC2wrmxJYNFOltoQKCTpFaKB6p937odona64OI,912
 thestage/debug_main.dist.py,sha256=UPIJ58yf-6FtXZj-FLAwxi7HononseuCYm9xb5KlxTs,783
 thestage/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -24,7 +24,7 @@ thestage/entities/enums/tail_output_type.py,sha256=6S55VRRosrI3yZW8XeAGm4u12H4ha
 thestage/entities/enums/yes_no_response.py,sha256=PXXkB7KMB-XCHvKOBRuX6NRKyq8UW0G228OICRr2ZTk,86
 thestage/entities/file_item.py,sha256=hxUbdne6AYFHJfuQ8l0XQO86IuC8Pyx_OrPw_0lhN_o,828
 thestage/entities/project_inference_simulator.py,sha256=Rgj5byZDq-uFI2ugOIMfW7CgJNYHfUyidhUbZ_kMoDk,647
-thestage/entities/project_inference_simulator_model.py,sha256=bg0ZVkz5lCHvUYIT3F4Z2xkwIy1rYIrbWHKSW-T_yJA,575
+thestage/entities/project_inference_simulator_model.py,sha256=8xzC6pZZQzMAh6DASS2ecYzv9EQZCc-fTpl5nOtJXzY,570
 thestage/entities/project_task.py,sha256=6UQaJMTuqDQJtbmbafD1iLjgE7CrlPL7AsGkIU91nNQ,594
 thestage/entities/rented_instance.py,sha256=DyekXeui3YiiKZmqtvb40n6j6w4_ly80TSjcAbjWlcU,628
 thestage/entities/self_hosted_instance.py,sha256=blsOdCMZPYM_xK4Zx4ZdGD43Ayy8j7ifX30O3fZ2XiM,630
@@ -38,14 +38,14 @@ thestage/exceptions/git_access_exception.py,sha256=ZyB5qrz6fok1jOEKtt3PXP9vq3vOm
 thestage/exceptions/remote_server_exception.py,sha256=Z1R5Wjw3w7VdqURhjidGkNhJL9lsA76txkAZPA5Uv70,595
 thestage/git/ProgressPrinter.py,sha256=z99BPq3KYycClL-7fmsERIIuZ03papdIYOhR8A7b9AA,573
 thestage/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-thestage/helpers/error_handler.py,sha256=-f4i11vca1kMjQNFfp3PU1qPaiLg-aq5j9VPgnPVSGA,5366
+thestage/helpers/error_handler.py,sha256=dVwdFvqYnlg-tOtNKVisyZipn0N2u7ZyHt5qVV2a6Rg,5370
 thestage/helpers/exception_hook.py,sha256=jekDs7smzI1UjXkuMut_uM725GL3c6Gr_zmyEYdDjV8,334
 thestage/helpers/logger/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 thestage/helpers/logger/app_logger.py,sha256=4dgZ_8k28R887rmIHmCTn2QVg0stqTk8hcXa5CF4VJc,1607
 thestage/helpers/ssh_util.py,sha256=JuDwddHxEGcA24Y8a-jLv339cG-jq4hEaBAl5TSVVFw,1262
 thestage/i18n/en_GB/messages.po,sha256=BuVIhd5TRQkgFkAbTGvbSRuO88lSJGpnk9TT2J9BC8E,32375
 thestage/i18n/translation.py,sha256=c62OicQ4phSMuqDe7hqGebIsk0W2-8ZJUfgfdtjjqEc,284
-thestage/main.py,sha256=P59_DqDeb6metD6RBCmpp1crEh9VhsKTYPt4SUdPfS4,1789
+thestage/main.py,sha256=O3NpGl4UE8GfsrdgDP9KZ5Oxe3-EDtBH5IlL-eFdgJs,1800
 thestage/services/.env,sha256=K2VpzFAVjD75XawAHZdR0HWmypryA_mXNY4WHNgR-wQ,184
 thestage/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 thestage/services/abstract_mapper.py,sha256=_q7YLkPNRsNW5wOCqvZIu1KfpLkc7uVaAQKrMZtsGuY,218
@@ -53,13 +53,13 @@ thestage/services/abstract_service.py,sha256=fsGXb0plZ9SvGBWAY4y599BH196YoFP3BJ7
 thestage/services/app_config_service.py,sha256=bbiXxcLoh_XC3mgsAo9ciksblPvUEcnG2oCIm1jMyo4,1679
 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=BzfiGizn0rByUL8x-XNbzJ_Zs169CwvLJI7NZ6rp4kI,16443
+thestage/services/clients/git/git_client.py,sha256=nUCszU5ymWUcf7YqeEhZqBhR17hwyDgMg00DTXOWIOY,16449
 thestage/services/clients/thestage_api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-thestage/services/clients/thestage_api/api_client.py,sha256=VLH_-WSbX4n3AFlLgQHJcLMEu28rTCVX59L2zGyPW7M,32257
+thestage/services/clients/thestage_api/api_client.py,sha256=pLxsHGqu3fEcPvmi9IRFeLCi7LalRUvM-gvo6GGLFF4,32141
 thestage/services/clients/thestage_api/core/api_client_core.py,sha256=lEffuh30f6oV2lp90UuET90OUxnrMh2PHjquF6_sT6E,3453
 thestage/services/clients/thestage_api/core/http_client_exception.py,sha256=JH-874Gu9T1b1_FpPBLqdyt9U0PyhpwRCe_oDc6c_jI,385
 thestage/services/clients/thestage_api/dtos/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-thestage/services/clients/thestage_api/dtos/base_controller/connect_resolve_response.py,sha256=EvbEMdYYXhkfqWeU9kajBUO8yJcAM8y4CvA8GbaiM3c,1021
+thestage/services/clients/thestage_api/dtos/base_controller/connect_resolve_response.py,sha256=mDSAq2gcmERAejnd8rXkAam39bH0WrTw5V4mQ_79hB8,1106
 thestage/services/clients/thestage_api/dtos/base_response.py,sha256=R6jjhvv0U0XD8j5N5T36q-6liiEAUy7xINFA1H7CMB8,410
 thestage/services/clients/thestage_api/dtos/container_param_request.py,sha256=Os3FTzbOIcEIFMpJFsFm0hhGV1ZfIyKrUw9OtwRw1Z0,449
 thestage/services/clients/thestage_api/dtos/container_response.py,sha256=hMrQe6RvurGONfFweSTRHi44AP1HVvIFPRmAZ1g8RHY,2445
@@ -80,11 +80,11 @@ thestage/services/clients/thestage_api/dtos/enums/selfhosted_status.py,sha256=_O
 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=xOO4ky_2CEGWCnPuBtEhhZ8BxJArFKZR_WnGYcqV7b8,870
-thestage/services/clients/thestage_api/dtos/inference_controller/deploy_inference_model_to_instance_response.py,sha256=mX_Parrm8ddTY9nqR4AYlI5WXAVpx1hI7yUEqJiiOFQ,383
+thestage/services/clients/thestage_api/dtos/inference_controller/deploy_inference_model_to_instance_request.py,sha256=wNDn4QhNrLONBZsyDC1p5o6jsgd-CQb5JQrxNBh9QK8,781
+thestage/services/clients/thestage_api/dtos/inference_controller/deploy_inference_model_to_instance_response.py,sha256=62iS_Jq6D0FScLwjwj64CxJ6JlKWwcN4LM2R-AmaUqs,480
 thestage/services/clients/thestage_api/dtos/inference_controller/deploy_inference_model_to_sagemaker_request.py,sha256=CAEzVOlud8dR3ifZLu09LIKAxt2iwZnjq4HmD0a6HXg,478
 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=63nSt4XucJHhHf99m-bdBTe8T6j-tLIQzs1XI71pd3U,399
+thestage/services/clients/thestage_api/dtos/inference_controller/get_inference_simulator_request.py,sha256=FctVvLAvUfBm-kTwozowyblHPAHZbBO0PAMz4_BA_O4,398
 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_request.py,sha256=JVpjCSfEHijhxojAFSu1d5jSeZX18xiTObUxnHbMs1Y,582
 thestage/services/clients/thestage_api/dtos/inference_controller/inference_simulator_list_response.py,sha256=KSPIabvvqDvpitMQElo3S37OQPRpTT-2tJWjelEZ1uM,544
@@ -108,7 +108,7 @@ thestage/services/clients/thestage_api/dtos/project_controller/project_get_deplo
 thestage/services/clients/thestage_api/dtos/project_controller/project_push_inference_simulator_model_request.py,sha256=pt4qI8EKFWOPoKHKMdWBzrW7vB_mb2Gstm1dVsob05U,378
 thestage/services/clients/thestage_api/dtos/project_controller/project_push_inference_simulator_model_response.py,sha256=aNaD4xVkhbwlD_rI3D0-2gp-F4L6CxxUkAOSLJZPl1I,249
 thestage/services/clients/thestage_api/dtos/project_controller/project_run_task_request.py,sha256=AkW-Gn2M8-NcRU0SWdmQ0vQ-LemPY4lh-UQBtB0Mka8,498
-thestage/services/clients/thestage_api/dtos/project_controller/project_run_task_response.py,sha256=iISmJTHbxSRJsXHT4qBTMP3-Co9fWCHI5s2UgB_-GV8,340
+thestage/services/clients/thestage_api/dtos/project_controller/project_run_task_response.py,sha256=cvwpZxb32614gFRzlTWWXQpUdDTrqHE5Gi4h3BV9q3g,453
 thestage/services/clients/thestage_api/dtos/project_controller/project_start_inference_simulator_request.py,sha256=4PGvswsyz8lx-A0xFrZEBEHbw8pEKv7WrnuDi3nZgfY,819
 thestage/services/clients/thestage_api/dtos/project_controller/project_start_inference_simulator_response.py,sha256=Q7XlbZ4vgk9QQL2z11BeUkKDn-5p8j06eNI0GmpZNjk,430
 thestage/services/clients/thestage_api/dtos/project_response.py,sha256=Q3gDfuiuVXls2WwOVgmKT9MGfOoM2GyE4n9YtmYFiUU,810
@@ -128,10 +128,10 @@ thestage/services/clients/thestage_api/dtos/user_controller/user_profile.py,sha2
 thestage/services/clients/thestage_api/dtos/validate_token_response.py,sha256=nmKbqRPkwtzhv37QheA_MO4CpEPmGM0VUC2gYckcrl8,418
 thestage/services/config_provider/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 thestage/services/config_provider/config_provider.py,sha256=CR2xSWRMIqsspek3iIs2qZ60ThHVNmTNTS6R4uheAuE,10238
-thestage/services/connect/connect_service.py,sha256=-La0JfVHEWu29zv9rP5TYUShJNErOVIpza0ysZIRFLk,9750
+thestage/services/connect/connect_service.py,sha256=w5Iq98HRZrfrUE5sfMMRAMPDZUh4jmeGDpXDXgAHBMs,9752
 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=q-fArCJlOaNrQVfc87Kb5RMyHh3x55acn1dA7hPtYMc,14409
+thestage/services/container/container_service.py,sha256=vNBNVGYtDRQCBTSKWq8vN-j8XuFE5NfwzzoqT96J08Y,18189
 thestage/services/container/mapper/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 thestage/services/container/mapper/container_mapper.py,sha256=GXSQX3H69M0wlZvXzO2jZs-Pqq70gDBvNmLKNvw6axU,1164
 thestage/services/core_files/config_entity.py,sha256=_55m8h2_KXIxl8Y71Yyk3wTj-q1mo918YU1yltzVqgQ,1075
@@ -146,22 +146,22 @@ thestage/services/logging/dto/log_message.py,sha256=k2clfz2fQnQ-ycFI8g8WYJ_XOjK0
 thestage/services/logging/dto/log_type.py,sha256=a6JWnq0ZjJ-2BQrG-fKYYy3UeJS2U2ZzE5P_EXglBfE,95
 thestage/services/logging/exception/log_polling_exception.py,sha256=rKQ7AtNCGKkk5OINIyyjvLT92PU5i_yJUH-Msr9hXQw,226
 thestage/services/logging/logging_constants.py,sha256=4Gk2tglHW_-jnjB8uVIh-ds4fAVBqNW8igfQt8k7Quc,137
-thestage/services/logging/logging_service.py,sha256=L4_vhPBz1ZG3X17U_9pnq9WV3EYKR4fiztC0s1au1PM,17770
+thestage/services/logging/logging_service.py,sha256=44YjcsYo-C3i8R8wFoz0JdhnqY6Tl90aI3M6nWH2yyQ,17819
 thestage/services/project/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 thestage/services/project/dto/inference_simulator_dto.py,sha256=skyxLgzwbR_EyVSCnojAwI3kZyJ5LJefP_0RpawlKa8,719
 thestage/services/project/dto/inference_simulator_model_dto.py,sha256=jNIPwqX_A7jgkcdugdEUpjpjo0SZwlHPK2D4K7i0Nj0,642
-thestage/services/project/dto/project_config.py,sha256=E7ZuvYxWVJuPiK9SbiRbRSNv_FE_kCXvkXE9AkcnSRE,694
+thestage/services/project/dto/project_config.py,sha256=EdaW2qQsw9LVyRh0ktxFNT48d7qqdiaKYCW03uvDqKc,583
 thestage/services/project/mapper/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 thestage/services/project/mapper/project_inference_simulator_mapper.py,sha256=rXb0fhZVjkx8E3tSnB7B6CMC275_reDG8mlZscXz0zk,856
 thestage/services/project/mapper/project_inference_simulator_model_mapper.py,sha256=JvCUVp7d4o2-CCxkA6aesSEz1iUJcIfBIkXdN_hHpT0,924
 thestage/services/project/mapper/project_task_mapper.py,sha256=8y4IqZOVXJUrI-946sZn295hyZGVtws61aETrhA9b1s,727
-thestage/services/project/project_service.py,sha256=3Mbs1ghnUsd7p565bf5s7vmRFStmYUfXYEoVIvmuv9s,61541
+thestage/services/project/project_service.py,sha256=RtFN9eH_5_heTRs-P5-c5dpDEF7yWDep__rB9sTLse4,61824
 thestage/services/remote_server_service.py,sha256=p_M0eL2Mtdz8_BCtNEhHXndaNDCFh_RgksQpWl0HDtE,23325
 thestage/services/service_factory.py,sha256=WosSbicSvY9dHOkS2OqGCtHkmKdJE1f_K7uEnB3Auaw,4291
 thestage/services/task/dto/task_dto.py,sha256=S1CzhCLW3B_ovSJA1pRBPerRvbnTk1c6yAHf_7RGZbc,1029
 thestage/services/validation_service.py,sha256=iopaDKuC_V_4Z_Zly5BFaLL74wG66z9URE86_I-9Hi4,2243
-thestage-0.6.6.dist-info/METADATA,sha256=KuzMtxjv2_kqNOkgwBkYyN-ybm3hHMzgLfBPbFd0BT8,5583
-thestage-0.6.6.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-thestage-0.6.6.dist-info/entry_points.txt,sha256=57pMhs8zaCM-jgeTffC0WVqCsh35Uq_dUDmzXR80CI4,47
-thestage-0.6.6.dist-info/licenses/LICENSE.txt,sha256=U9QrxfdD7Ie7r8z1FleuvOGQvgCF1m0Mjd78cFvWaHE,572
-thestage-0.6.6.dist-info/RECORD,,
+thestage-0.6.7.dist-info/METADATA,sha256=dyt0Kh8J-VQvPTX_ZKILOUkcTsWeowX0RRV5Xl9kJEQ,5622
+thestage-0.6.7.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+thestage-0.6.7.dist-info/entry_points.txt,sha256=57pMhs8zaCM-jgeTffC0WVqCsh35Uq_dUDmzXR80CI4,47
+thestage-0.6.7.dist-info/licenses/LICENSE.txt,sha256=U9QrxfdD7Ie7r8z1FleuvOGQvgCF1m0Mjd78cFvWaHE,572
+thestage-0.6.7.dist-info/RECORD,,