thestage 0.5.37__tar.gz → 0.5.39__tar.gz

{thestage-0.5.37 → thestage-0.5.39}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: thestage
-Version: 0.5.37
+Version: 0.5.39
 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)
@@ -19,7 +20,6 @@ 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: tabulate (>=0.9.0,<0.10.0)
-Requires-Dist: thestage-core (==0.0.16)
 Requires-Dist: typer[all] (>=0.9.0,<0.10.0)
 Description-Content-Type: text/markdown
 

{thestage-0.5.37 → thestage-0.5.39}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "thestage"
-version = "0.5.37"
+version = "0.5.39"
 description = ""
 authors = ["TheStage AI team <hello@thestage.ai>"]
 readme = "README.md"
@@ -28,11 +28,10 @@ gitpython = "^3.1.40"
 requests = "^2.31.0"
 python-gettext-translations = "^1.1.0"
 paramiko = "^3.4.0"
-#thestage-core = {git = "git@github.com:TheStageAI/thestage-core.git", branch = 'main'}
-#thestage_core = { path = "/Users/alexey/Documents/PYTHON_thestage_core", develop = true }
-thestage-core = "0.0.16"
+
 aioconsole = "^0.8.0"
 httpx = "^0.27.2"
+boto3 = "^1.35.80"
 
 [tool.poetry.dev-dependencies]
 pytest = "^7.4.3"

{thestage-0.5.37 → thestage-0.5.39}/thestage/__init__.py

@@ -1,3 +1,3 @@
 from . import *
 __app_name__ = "thestage"
-__version__ = "0.5.37"
+__version__ = "0.5.39"

{thestage-0.5.37 → thestage-0.5.39}/thestage/controllers/config_controller.py

@@ -2,8 +2,7 @@ import os
 from pathlib import Path
 
 import click
-from thestage_core.entities.config_entity import ConfigEntity
-from thestage_core.services.filesystem_service import FileSystemServiceCore
+from thestage.services.core_files.config_entity import ConfigEntity
 
 from thestage.entities.enums.yes_no_response import YesOrNoResponse
 from thestage.i18n.translation import __
@@ -33,7 +32,7 @@ def config_get():
         typer.echo(__('No configuration found'))
         raise typer.Exit(1)
 
-    config_provider.save_global_config(config=config)
+    config_provider.save_config(config=config)
 
     typer.echo(__('THESTAGE TOKEN: %token%', {'token': config.main.thestage_auth_token or ''}))
     typer.echo(__('THESTAGE API LINK: %link%', {'link': config.main.thestage_api_url or ''}))
@@ -80,7 +79,7 @@ def config_clear():
     local_path = get_current_directory()
     config_provider = ConfigProvider(local_path=local_path)
     config_dir = config_provider.get_full_config().runtime.config_global_path
-    config_provider.remove_all_config()
+    config_provider.clear_config()
     typer.echo(f'Removed {config_dir}')
 
     raise typer.Exit(0)

{thestage-0.5.37 → thestage-0.5.39}/thestage/controllers/container_controller.py

@@ -214,11 +214,6 @@ def put_file(
         typer.echo(__('Container unique ID is required'))
         raise typer.Exit(1)
 
-    # TODO delete when we change mappings
-    if destination_path.startswith("/app/") or destination_path == "/app":
-        typer.echo("/app/ is reserved for project code and must not contain custom files. Сonsider adding your files to the project repository")
-        raise typer.Exit(1)
-
     service_factory = validate_config_and_get_service_factory()
     config = service_factory.get_config_provider().get_full_config()
 
@@ -396,9 +391,16 @@ def stop_container(
     raise typer.Exit(0)
 
 
-@app.command(name="logs", no_args_is_help=True, help=__("Stream real-time container logs; log history is available in your TheStage AI account"))
+@app.command(name="logs", no_args_is_help=True, help=__("Stream real-time Docker container logs or view last logs for a container"))
 def container_logs(
         container_uid: Optional[str] = typer.Argument(help=__("Container unique id")),
+        logs_number: Optional[int] = typer.Option(
+            None,
+            '--number',
+            '-n',
+            help=__("Display a number of latest log entries. No real-time stream if provided."),
+            is_eager=False,
+        ),
 ):
     """
         Streams real-time container logs
@@ -412,21 +414,15 @@ def container_logs(
     service_factory = validate_config_and_get_service_factory()
     config = service_factory.get_config_provider().get_full_config()
 
-    container_service: ContainerService = service_factory.get_container_service()
     logging_service: LoggingService = service_factory.get_logging_service()
 
-    container: Optional[DockerContainerDto] = container_service.get_container(
-        config=config,
-        container_slug=container_uid,
-    )
-
-    if container:
-        logging_service.stream_container_logs(
+    if logs_number is None:
+        logging_service.stream_container_logs_with_controls(
             config=config,
-            container=container
+            container_uid=container_uid
         )
     else:
-        typer.echo(__("Container not found: %container_slug%", {'container_slug': container_uid}))
+        logging_service.print_last_container_logs(config=config, container_uid=container_uid, logs_number=logs_number)
 
     app_logger.info(f'Container logs - end')
     raise typer.Exit(0)

{thestage-0.5.37 → thestage-0.5.39}/thestage/controllers/project_controller.py

@@ -1,3 +1,4 @@
+import time
 from typing import Optional, List
 
 import re
@@ -102,11 +103,10 @@ def init(
     )
 
     typer.echo(__("Project successfully initialized at %path%", {"path": config.runtime.working_directory}))
-    typer.echo(__("Initialization complete"))
     raise typer.Exit(0)
 
 
-@app.command(name='run', no_args_is_help=True, help=__("Run task within project. By default, it uses the latest commit from the main branch and streams real-time task logs."))
+@app.command(name='run', no_args_is_help=True, help=__("Run a task within the project. By default, it uses the latest commit from the main branch and streams real-time task logs."))
 def run(
         command: Annotated[List[str], typer.Argument(
             help=__("Command to run (required)"),
@@ -140,6 +140,13 @@ def run(
             help=__("Disable real-time log streaming"),
             is_eager=False,
         ),
+        task_title: Optional[str] = typer.Option(
+            None,
+            "--title",
+            "-t",
+            help=__("Provide a custom task title. Git commit message is used by default."),
+            is_eager=False,
+        ),
 ):
     """
         Runs a task within a project
@@ -160,6 +167,7 @@ def run(
         run_command=" ".join(command),
         commit_hash=commit_hash,
         docker_container_slug=docker_container_slug,
+        task_title=task_title,
     )
 
     if enable_log_stream:
@@ -723,4 +731,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-0.5.37 → thestage-0.5.39}/thestage/controllers/utils_controller.py

@@ -2,8 +2,7 @@ import os
 import pathlib
 from typing import Optional, Dict, Tuple
 
-from thestage_core.entities.config_entity import ConfigEntity
-
+from thestage.services.core_files.config_entity import ConfigEntity
 from thestage.helpers.error_handler import error_handler
 from thestage.services.service_factory import ServiceFactory
 from thestage.services.config_provider.config_provider import ConfigProvider
@@ -24,6 +23,6 @@ def validate_config_and_get_service_factory(
 
     validation_service = service_factory.get_validation_service()
     validation_service.check_token(config=config)
-    config_provider.save_global_config(config=config)
+    config_provider.save_config(config=config)
 
     return service_factory

thestage-0.5.39/thestage/entities/file_item.py

@@ -0,0 +1,27 @@
+import os
+from pathlib import Path
+from typing import Optional, List
+
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class FileItemEntity(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    name: Optional[str] = Field(None)
+    path: Optional[str] = Field(None)
+    is_file: Optional[bool] = Field(False)
+    is_folder: Optional[bool] = Field(False)
+    file_size: Optional[int] = Field(None)
+    children: List['FileItemEntity'] = Field(default=[])
+
+    @staticmethod
+    def build_from_path(path: Path) -> 'FileItemEntity':
+        file_stat = os.stat(path)
+        return FileItemEntity(
+                name=path.name,
+                path=str(path.absolute()),
+                is_file=path.is_file(),
+                is_folder=path.is_dir(),
+                file_size=file_stat.st_size,
+            )

{thestage-0.5.37 → thestage-0.5.39}/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-0.5.39/thestage/exceptions/file_system_exception.py

@@ -0,0 +1,6 @@
+from thestage.exceptions.base_exception import BaseAbstractException
+
+
+class FileSystemException(BaseAbstractException):
+    def __init__(self, message: str):
+        super(FileSystemException, self).__init__(message=message)

{thestage-0.5.37 → thestage-0.5.39}/thestage/helpers/error_handler.py

@@ -6,10 +6,9 @@ import typer
 from click.exceptions import Exit, Abort
 from git import GitCommandError
 from paramiko.ssh_exception import PasswordRequiredException
-from thestage_core.exceptions.file_system_exception import FileSystemException
-from thestage_core.exceptions.http_error_exception import HttpClientException
 
 from thestage.config import THESTAGE_API_URL
+from thestage.exceptions.file_system_exception import FileSystemException
 from thestage.exceptions.remote_server_exception import RemoteServerException
 from thestage.i18n.translation import __
 from thestage.exceptions.git_access_exception import GitAccessException
@@ -17,6 +16,7 @@ from thestage.exceptions.auth_exception import AuthException
 from thestage.exceptions.business_logic_exception import BusinessLogicException
 from thestage.exceptions.config_exception import ConfigException
 from thestage.helpers.logger.app_logger import app_logger
+from thestage.services.clients.thestage_api.core.http_client_exception import HttpClientException
 
 
 def error_handler() -> Callable:

{thestage-0.5.37 → thestage-0.5.39}/thestage/helpers/logger/app_logger.py

@@ -2,13 +2,12 @@ import logging
 import platform
 from logging.handlers import RotatingFileHandler
 from pathlib import Path
-from typing import Any
 
-import typer
-from thestage_core.config import THESTAGE_CONFIG_DIR, THESTAGE_CONFIG_FILE
-from thestage_core.exceptions.file_system_exception import FileSystemException
+
+from thestage.config import THESTAGE_CONFIG_DIR
 
 from thestage.config import THESTAGE_LOGGING_FILE
+from thestage.exceptions.file_system_exception import FileSystemException
 
 
 def get_log_path_from_os() -> Path:

{thestage-0.5.37 → thestage-0.5.39}/thestage/services/abstract_service.py

@@ -3,8 +3,7 @@ from typing import List, Dict, Any, Optional
 
 import typer
 from tabulate import tabulate
-from thestage_core.entities.config_entity import ConfigEntity
-
+from thestage.services.core_files.config_entity import ConfigEntity
 from thestage.helpers.logger.app_logger import app_logger
 from thestage.i18n.translation import __
 from thestage.services.abstract_mapper import AbstractMapper

{thestage-0.5.37 → thestage-0.5.39}/thestage/services/app_config_service.py

@@ -2,8 +2,7 @@ import os
 
 import click
 import typer
-from thestage_core.entities.config_entity import ConfigEntity
-
+from thestage.services.core_files.config_entity import ConfigEntity
 from thestage.i18n.translation import __
 from thestage.entities.enums.yes_no_response import YesOrNoResponse
 from thestage.services.config_provider.config_provider import ConfigProvider
@@ -42,7 +41,7 @@ class AppConfigService:
         config.main.thestage_auth_token = token
 
         self.__validation_service.check_token(config=config)
-        self.__config_provider.save_global_config(config=config)
+        self.__config_provider.save_config(config=config)
 
     @staticmethod
     def app_remove_env():

{thestage-0.5.37 → thestage-0.5.39}/thestage/services/clients/git/git_client.py

@@ -7,11 +7,11 @@ import git
 import typer
 from git import Remote, Repo, GitCommandError, Commit
 from rich import print
-from thestage_core.services.filesystem_service import FileSystemServiceCore
 
 from thestage.color_scheme.color_scheme import ColorScheme
 from thestage.exceptions.git_access_exception import GitAccessException
 from thestage.git.ProgressPrinter import ProgressPrinter
+from thestage.services.filesystem_service import FileSystemServiceCore
 
 
 class GitLocalClient:
@@ -260,8 +260,8 @@ class GitLocalClient:
             self.__file_system_service.create_if_not_exists_file(gitignore_path)
             self.git_add_by_path(repo_path=path, file_path=str(gitignore_path))
 
-        is_present_tsr = self.__file_system_service.find_in_text_file(file=str(gitignore_path),
-                                                                      find=self.__git_ignore_thestage_line)
+        is_present_tsr = self.__file_system_service.find_line_in_text_file(file=str(gitignore_path),
+                                                                           find=self.__git_ignore_thestage_line)
         if not is_present_tsr:
             self.__file_system_service.add_line_to_text_file(file=str(gitignore_path),
                                                              new_line=self.__git_ignore_thestage_line)

{thestage-0.5.37 → thestage-0.5.39}/thestage/services/clients/thestage_api/api_client.py

@@ -2,14 +2,22 @@ from typing import Optional, List, Tuple, Dict, Iterator
 
 import httpx
 import requests
-from thestage_core.services.clients.thestage_api.api_client import TheStageApiClientCore
 
 from thestage.helpers.error_handler import error_handler
+from thestage.services.clients.thestage_api.core.api_client_core import TheStageApiClientCore
 from thestage.services.clients.thestage_api.dtos.docker_container_controller.docker_container_list_request import \
     DockerContainerListRequest
 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 \
@@ -514,65 +522,6 @@ class TheStageApiClient(TheStageApiClientCore):
         return ProjectRunTaskResponse.model_validate(response) if response else None
 
 
-    def get_container_log_stream(self, token: str, container_id: int) -> Iterator[str]:
-        request_headers = {'Content-Type': 'application/json'}
-        if token: request_headers['Authorization'] = f"Bearer {token}"
-
-        session = requests.Session()
-        request = DockerContainerLogStreamRequest(
-            dockerContainerId=container_id,
-        )
-
-        with session.post(
-            url=f"{self._get_host()}/user-api/v1/logging/stream/container",
-            headers=request_headers,
-            stream=True,
-            json=request.model_dump(),
-        ) as response:
-            for line in response.iter_lines():
-                if line:
-                    yield line.decode('utf-8')
-
-
-    async def get_container_log_stream_httpx(self, token: str, container_id: int) -> Iterator[str]:
-        request_headers = {'Content-Type': 'application/json'}
-        if token: request_headers['Authorization'] = f"Bearer {token}"
-
-        request = DockerContainerLogStreamRequest(
-            dockerContainerId=container_id,
-        )
-
-        async with httpx.stream(
-            method="POST",
-            url=f"{self._get_host()}/user-api/v1/logging/stream/container",
-            headers=request_headers,
-            json=request.model_dump(),
-        ) as response:
-            async for line in response.iter_lines():
-                if line:
-                    yield line.decode('utf-8')
-
-
-    async def get_task_log_stream_httpx(self, token: str, task_id: int):
-        request_headers = {'Content-Type': 'application/json'}
-        if token: request_headers['Authorization'] = f"Bearer {token}"
-
-        request = TaskLogStreamRequest(
-            taskId=task_id,
-        )
-
-        async with httpx.AsyncClient() as client:
-            async with client.stream(
-                    method="POST",
-                    url=f"{self._get_host()}/user-api/v1/logging/stream/task",
-                    headers=request_headers,
-                    json=request.model_dump(),
-            ) as response:
-                async for line in response.aiter_lines():
-                    if line:
-                        yield line
-
-
     async def poll_logs_httpx(self, token: str, docker_container_id: Optional[int], last_log_timestamp: str, last_log_id: str, task_id: Optional[int] = None, inference_simulator_id: Optional[int] = None) -> Optional[LogPollingResponse]:
         request_headers = {'Content-Type': 'application/json'}
         if token: request_headers['Authorization'] = f"Bearer {token}"
@@ -736,10 +685,58 @@ class TheStageApiClient(TheStageApiClientCore):
         return GetInferenceSimulatorResponse.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:
+    @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, container_id: Optional[int] = None) -> UserLogsQueryResponse:
         request = UserLogsQueryRequest(
             inferenceSimulatorId=inference_simulator_id,
             taskId=task_id,
+            containerId=container_id,
             limit=limit,
             ascendingOrder=False,
         )