thestage 0.6.8__tar.gz → 0.7.0__tar.gz

{thestage-0.6.8 → thestage-0.7.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: thestage
-Version: 0.6.8
+Version: 0.7.0
 Summary: 
 License-File: LICENSE.txt
 Author: TheStage AI team

{thestage-0.6.8 → thestage-0.7.0}/pyproject.toml

@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "thestage"
 
-version = "0.6.8"
+version = "0.7.0"
 
 description = ""
 authors = ["TheStage AI team <hello@thestage.ai>"]

{thestage-0.6.8 → thestage-0.7.0}/thestage/__init__.py

@@ -1,3 +1,3 @@
 from . import *
 __app_name__ = "thestage"
-__version__ = "0.6.8"
+__version__ = "0.7.0"

{thestage-0.6.8 → thestage-0.7.0}/thestage/cli_command.py

@@ -27,10 +27,13 @@ class CliCommand(str, Enum):
     CONTAINER_CONNECT = "CONTAINER_CONNECT"
     CONTAINER_UPLOAD = "CONTAINER_UPLOAD"
     CONTAINER_DOWNLOAD = "CONTAINER_DOWNLOAD"
+    CONTAINER_CREATE = "CONTAINER_CREATE"
     CONTAINER_START = "CONTAINER_START"
     CONTAINER_STOP = "CONTAINER_STOP"
+    CONTAINER_DELETE = "CONTAINER_DELETE"
     CONTAINER_RESTART = "CONTAINER_RESTART"
     CONTAINER_LOGS = "CONTAINER_LOGS"
+    CONTAINER_IMAGE_LS = "CONTAINER_IMAGE_LS"
     INSTANCE_RENTED_LS = "INSTANCE_RENTED_LS"
     INSTANCE_RENTED_CONNECT = "INSTANCE_RENTED_CONNECT"
     INSTANCE_SELF_HOSTED_LS = "INSTANCE_SELF_HOSTED_LS"

{thestage-0.6.8 → thestage-0.7.0}/thestage/docker_container/business/container_service.py

@@ -7,19 +7,24 @@ import typer
 
 from thestage.color_scheme.color_scheme import ColorScheme
 from thestage.connect.communication.connect_api_client import ConnectApiClient
+from thestage.docker_container.business.mapper.image_mapper import ImageMapper
 from thestage.docker_container.communication.docker_container_api_client import DockerContainerApiClient
 from thestage.docker_container.dto.container_entity import DockerContainerEntity
 from thestage.docker_container.dto.container_action_request import DockerContainerActionRequest
+from thestage.docker_container.dto.docker_container_create_request import DockerContainerCreateRequest, \
+    DockerContainerMappings
 from thestage.docker_container.dto.enum.container_pending_action import DockerContainerAction
 from thestage.docker_container.dto.enum.container_status import DockerContainerStatus
+from thestage.docker_container.dto.image_entity import DockerImageEntity
 from thestage.global_dto.enums.shell_type import ShellType
+from thestage.project.dto.project_config import ProjectConfig
 from thestage.services.clients.thestage_api.dtos.paginated_entity_list import PaginatedEntityList
 from thestage.docker_container.business.mapper.container_mapper import ContainerMapper
 from thestage.services.filesystem_service import FileSystemService
 from thestage.connect.business.remote_server_service import RemoteServerService
 from thestage.i18n.translation import __
 from thestage.services.abstract_service import AbstractService
-from thestage.docker_container.dto.container_response import DockerContainerDto
+from thestage.docker_container.dto.container_response import DockerContainerDto, DockerImageDto
 from thestage.helpers.error_handler import error_handler
 from thestage.config.business.config_provider import ConfigProvider
 
@@ -118,6 +123,7 @@ class ContainerService(AbstractService):
 
         return list
 
+
     # TODO delete this proxy method
     @error_handler()
     def get_container(
@@ -456,3 +462,117 @@ class ContainerService(AbstractService):
 
         if result.is_success:
             typer.echo(f'Docker container action scheduled: {action.value}')
+
+    @error_handler()
+    def create_docker_container(
+            self,
+            project_public_id: Optional[str],
+            project_slug: Optional[str],
+            container_slug: str,
+            rented_instance_public_id: Optional[str],
+            rented_instance_slug: Optional[str],
+            self_hosted_instance_public_id: Optional[str],
+            self_hosted_instance_slug: Optional[str],
+            image_name: str,
+            cpu_number: Optional[int],
+            gpu_number: Optional[int],
+            volume_mappings: Optional[Dict[str, str]],
+    ):
+        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 name, or run this command from within an initialized project directory")
+                raise typer.Exit(1)
+            project_public_id = project_config.public_id
+
+        docker_images_paginated = self.get_image_list()
+        if not docker_images_paginated or not docker_images_paginated.entities:
+            typer.echo("No Docker images available")
+            raise typer.Exit(1)
+
+        selected_image = None
+        selection_pool = docker_images_paginated.entities
+
+        if image_name is not None:
+            matches = [img for img in selection_pool if img.image == image_name]
+
+            if len(matches) == 1:
+                selected_image = matches[0]
+            elif len(matches) > 1:
+                typer.echo(f"Multiple tags found for repository '{image_name}':")
+                selection_pool = matches
+            else:
+                typer.echo(f"Image '{image_name}' not found. Select from available images:")
+
+        if selected_image is None:
+            for idx, img in enumerate(selection_pool, start=1):
+                typer.echo(f"{idx}) {img.image}:{img.tag}")
+
+            choice_str = typer.prompt("Choose which image to use")
+
+            try:
+                choice_index = int(choice_str)
+            except ValueError:
+                typer.echo("Error: Invalid input. Enter a number.")
+                raise typer.Exit(1)
+
+            if not (1 <= choice_index <= len(selection_pool)):
+                typer.echo("Error: Choice out of range.")
+                raise typer.Exit(1)
+
+            selected_image = selection_pool[choice_index - 1]
+
+        request_params = DockerContainerCreateRequest(
+            instance_rented_public_id=rented_instance_public_id,
+            instance_rented_slug=rented_instance_slug,
+            selfhosted_instance_public_id=self_hosted_instance_public_id,
+            selfhosted_instance_slug=self_hosted_instance_slug,
+            project_public_id=project_public_id,
+            project_slug=project_slug,
+            container_slug=container_slug,
+            docker_image_id=selected_image.id,
+            assigned_cpus=cpu_number,
+            is_unlimited_cpus=cpu_number is None,
+            assigned_gpus=gpu_number,
+            is_unlimited_gpus=gpu_number is None,
+            mappings=DockerContainerMappings(directory_mappings=volume_mappings)
+        )
+
+        result = self.__docker_container_api_client.create_container(
+            request_params
+        )
+        if result.is_success:
+            typer.echo(f'Docker container creation scheduled.')
+
+    @error_handler()
+    def print_docker_image_list(
+            self,
+            row: int,
+            page: int,
+    ):
+        typer.echo("Listing available Docker images:")
+
+        self.print(
+            func_get_data=self.get_image_list,
+            func_special_params={},
+            mapper=ImageMapper(),
+            headers=list(map(lambda x: x.alias, DockerImageEntity.model_fields.values())),
+            row=row,
+            page=page,
+            max_col_width=[50, 20],
+            show_index="never",
+        )
+
+    @error_handler()
+    def get_image_list(
+            self,
+            row: int = 5,
+            page: int = 1,
+    ) -> PaginatedEntityList[DockerImageDto]:
+
+        list = self.__docker_container_api_client.get_docker_image_list(
+            page=page,
+            limit=row,
+        )
+
+        return list

thestage-0.7.0/thestage/docker_container/business/mapper/image_mapper.py

@@ -0,0 +1,16 @@
+from typing import Optional
+
+from thestage.docker_container.dto.container_response import DockerImageDto
+from thestage.docker_container.dto.image_entity import DockerImageEntity
+from thestage.services.abstract_mapper import AbstractMapper
+
+
+class ImageMapper(AbstractMapper):
+    def build_entity(self, item: DockerImageDto) -> Optional[DockerImageEntity]:
+        if not item:
+            return None
+
+        return DockerImageEntity(
+            image=item.image or '',
+            tag=item.tag or '',
+        )

{thestage-0.6.8 → thestage-0.7.0}/thestage/docker_container/communication/docker_command.py

@@ -15,7 +15,6 @@ from thestage.logging.business.logging_service import LoggingService
 
 app = typer.Typer(no_args_is_help=True, help=__("Manage containers"))
 
-
 @app.command(name='ls', help=__("List containers"), **get_command_metadata(CliCommand.CONTAINER_LS))
 def list_containers(
         row: int = typer.Option(
@@ -289,7 +288,7 @@ def start_container(
     check_command_permission(command_name)
 
     if sum(v is not None for v in [container_public_id, container_slug]) != 1:
-        typer.echo("Please provide a single identifier for container - ID or name.")
+        typer.echo("Provide a single identifier for container - ID or name.")
         raise typer.Exit(1)
 
     service_factory = validate_config_and_get_service_factory()
@@ -301,7 +300,140 @@ def start_container(
         action=DockerContainerAction.START
     )
 
-    app_logger.info(f'Container start completed')
+    app_logger.info(f'Container is starting')
+    raise typer.Exit(0)
+
+
+@app.command(name="create", no_args_is_help=True, help=__("Create container"), **get_command_metadata(CliCommand.CONTAINER_CREATE))
+def create_container(
+        project_public_id: Optional[str] = typer.Option(
+            None,
+            '--project-id',
+            '-pid',
+            help=__("Project ID. By default, project info is taken from the current directory"),
+            is_eager=False,
+        ),
+        project_slug: Optional[str] = typer.Option(
+            None,
+            '--project-name',
+            '-pn',
+            help=__("Project name. By default, project info is taken from the current directory"),
+            is_eager=False,
+        ),
+        working_directory: Optional[str] = typer.Option(
+            None,
+            "--working-directory",
+            "-wd",
+            help=__("Full path to working directory"),
+            show_default=False,
+            is_eager=False,
+        ),
+        container_slug: Optional[str] = typer.Option(
+            None,
+            '--container-name',
+            '-cn',
+            help=__("Container name"),
+            is_eager=False,
+        ),
+        rented_instance_public_id: Optional[str] = typer.Option(
+            None,
+            '--rented-instance-id',
+            '-rid',
+            help=__("Rented instance ID on which container will run"),
+            is_eager=False,
+        ),
+        rented_instance_slug: Optional[str] = typer.Option(
+            None,
+            '--rented-instance-name',
+            '-rn',
+            help=__("Rented instance name on which container will run"),
+            is_eager=False,
+        ),
+        self_hosted_instance_public_id: Optional[str] = typer.Option(
+            None,
+            '--self-hosted-instance-id',
+            '-sid',
+            help=__("Self-hosted instance ID on which container will run"),
+            is_eager=False,
+        ),
+        self_hosted_instance_slug: Optional[str] = typer.Option(
+            None,
+            '--self-hosted-instance-name',
+            '-sn',
+            help=__("Self-hosted instance name on which container will run"),
+            is_eager=False,
+        ),
+        image_name: Optional[str] = typer.Option(
+            None,
+            '--image',
+            '-i',
+            help=__("Docker image name. If not provided, you will be prompted to select from available images"),
+            is_eager=False,
+        ),
+        gpu_number: Optional[int] = typer.Option(
+            None,
+            '--gpu-number',
+            '-gpun',
+            help=__("Number of assigned GPUs for container. No GPU limit by default"),
+            is_eager=False,
+        ),
+        cpu_number: Optional[int] = typer.Option(
+            None,
+            '--cpu-number',
+            '-cpun',
+            help=__("Number of assigned CPUs for container. No CPU limit by default"),
+            is_eager=False,
+        ),
+        volume_mounts: Optional[List[str]] = typer.Option(
+            None,
+            '--volume',
+            '-v',
+            help=__("Volume mounts. Can be used multiple times. Format: /host_dir:/container_dir"),
+            is_eager=False,
+        ),
+):
+    command_name = CliCommand.CONTAINER_CREATE
+    app_logger.info(f'Running {command_name} from {get_current_directory()}')
+    check_command_permission(command_name)
+
+    if sum(v is not None for v in [project_public_id, project_slug]) > 1:
+        typer.echo("Provide a single identifier for project - ID or name.")
+        raise typer.Exit(1)
+
+    if sum(v is not None for v in [rented_instance_public_id, rented_instance_slug, self_hosted_instance_public_id, self_hosted_instance_slug]) != 1:
+        typer.echo("Provide exactly one identifier (ID or Name) for either a rented or self-hosted instance.")
+        raise typer.Exit(1)
+
+    if not container_slug:
+        container_slug = typer.prompt("Enter container name")
+
+    volume_mappings_dict = {}
+    if volume_mounts:
+        for mount in volume_mounts:
+            if ":" not in mount:
+                typer.echo(f"Invalid volume format: {mount}. Expected host:container")
+                raise typer.Exit(1)
+            host_path, container_path = mount.split(":", 1)
+            volume_mappings_dict[host_path] = container_path
+
+    service_factory = validate_config_and_get_service_factory(working_directory=working_directory)
+    container_service: ContainerService = service_factory.get_container_service()
+
+    container_service.create_docker_container(
+        project_public_id=project_public_id,
+        project_slug=project_slug,
+        container_slug=container_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,
+        self_hosted_instance_slug=self_hosted_instance_slug,
+        image_name=image_name,
+        cpu_number=cpu_number,
+        gpu_number=gpu_number,
+        volume_mappings=volume_mappings_dict
+    )
+
+    app_logger.info(f'Container is creating')
     raise typer.Exit(0)
 
 
@@ -327,7 +459,7 @@ def stop_container(
     check_command_permission(command_name)
 
     if sum(v is not None for v in [container_public_id, container_slug]) != 1:
-        typer.echo("Please provide a single identifier for container - ID or name.")
+        typer.echo("Provide a single identifier for container - ID or name.")
         raise typer.Exit(1)
 
     service_factory = validate_config_and_get_service_factory()
@@ -339,10 +471,47 @@ def stop_container(
         action=DockerContainerAction.STOP
     )
 
-    app_logger.info(f'Container stop completed')
+    app_logger.info(f'Container is stopping')
     raise typer.Exit(0)
 
 
+@app.command(name="delete", no_args_is_help=True, help=__("Delete container"), **get_command_metadata(CliCommand.CONTAINER_DELETE))
+def delete_container(
+        container_public_id: str = typer.Option(
+            None,
+            '--container-id',
+            '-cid',
+            help=__("Container ID"),
+            is_eager=False,
+        ),
+        container_slug: str = typer.Option(
+            None,
+            '--container-name',
+            '-cn',
+            help=__("Container name"),
+            is_eager=False,
+        ),
+):
+    command_name = CliCommand.CONTAINER_DELETE
+    app_logger.info(f'Running {command_name} from {get_current_directory()}')
+    check_command_permission(command_name)
+
+    if sum(v is not None for v in [container_public_id, container_slug]) != 1:
+        typer.echo("Provide a single identifier for container - ID or name.")
+        raise typer.Exit(1)
+
+    service_factory = validate_config_and_get_service_factory()
+    container_service: ContainerService = service_factory.get_container_service()
+
+    container_service.request_docker_container_action(
+        container_public_id=container_public_id,
+        container_slug=container_slug,
+        action=DockerContainerAction.DELETE
+    )
+
+    app_logger.info(f'Container is deleting')
+    raise typer.Exit(0)
+
 @app.command(name="restart", no_args_is_help=True, help=__("Restart container"), **get_command_metadata(CliCommand.CONTAINER_RESTART))
 def restart_container(
         container_public_id: str = typer.Option(
@@ -365,7 +534,7 @@ def restart_container(
     check_command_permission(command_name)
 
     if sum(v is not None for v in [container_public_id, container_slug]) != 1:
-        typer.echo("Please provide a single identifier for container - ID or name.")
+        typer.echo("Provide a single identifier for container - ID or name.")
         raise typer.Exit(1)
 
     service_factory = validate_config_and_get_service_factory()
@@ -377,7 +546,7 @@ def restart_container(
         action=DockerContainerAction.RESTART
     )
 
-    app_logger.info(f'Container restart completed')
+    app_logger.info(f'Container is restarting')
     raise typer.Exit(0)
 
 
@@ -410,7 +579,7 @@ def container_logs(
     check_command_permission(command_name)
 
     if sum(v is not None for v in [container_public_id, container_slug]) != 1:
-        typer.echo("Please provide a single identifier for container - ID or name.")
+        typer.echo("Provide a single identifier for container - ID or name.")
         raise typer.Exit(1)
 
     service_factory = validate_config_and_get_service_factory()
@@ -425,4 +594,4 @@ def container_logs(
         logging_service.print_last_container_logs(container_public_id=container_public_id, container_slug=container_slug, logs_number=logs_number)
 
     app_logger.info(f'Container log streaming completed')
-    raise typer.Exit(0)
+    raise typer.Exit(0)

{thestage-0.6.8 → thestage-0.7.0}/thestage/docker_container/communication/docker_container_api_client.py

@@ -1,6 +1,10 @@
 from typing import Optional, List, Dict
 
 from thestage.config.business.config_provider import ConfigProvider
+from thestage.docker_container.dto.docker_container_create_request import DockerContainerCreateRequest
+from thestage.docker_container.dto.docker_container_create_response import DockerContainerCreateResult
+from thestage.docker_container.dto.docker_image_list_request import DockerImageListRequest
+from thestage.docker_container.dto.docker_image_list_response import DockerImageListResponse
 from thestage.global_dto.enums.order_direction_type import OrderDirectionType
 from thestage.services.clients.thestage_api.core.api_client_core import TheStageApiClientCore
 from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
@@ -11,7 +15,7 @@ from thestage.docker_container.dto.container_action_request import DockerContain
 from thestage.docker_container.dto.container_response import (
     DockerContainerDto,
     DockerContainerViewResponse,
-    ContainerBusinessStatusMapperResponse
+    ContainerBusinessStatusMapperResponse, DockerImageDto
 )
 from thestage.docker_container.dto.docker_container_list_request import DockerContainerListRequest
 from thestage.docker_container.dto.docker_container_list_response import DockerContainerListResponse
@@ -50,9 +54,32 @@ class DockerContainerApiClient(TheStageApiClientCore):
         )
 
         result = DockerContainerListResponse.model_validate(response) if response else None
-        # return result.paginatedList.entities, result.paginatedList.pagination_data.total_pages if result and result.is_success else None
         return result.paginatedList if result and result.is_success else None
 
+
+    def get_docker_image_list(
+            self,
+            page: int = 1,
+            limit: int = 10,
+    ) -> PaginatedEntityList[DockerImageDto]:
+        request = DockerImageListRequest(
+            entityFilterRequest=EntityFilterRequest(
+                orderByField="createdAt",
+                orderByDirection=OrderDirectionType.DESC,
+                page=page,
+                limit=limit,
+            ),
+        )
+        response = self._request(
+            method='POST',
+            url='/user-api/v1/docker-image/list',
+            data=request.model_dump(),
+            token=self.__config_provider.get_config().main.thestage_auth_token
+        )
+        result = DockerImageListResponse.model_validate(response) if response else None
+        return result.paginatedList if result and result.is_success else None
+
+
     def get_container(
             self,
             container_slug: Optional[str] = None,
@@ -72,6 +99,18 @@ class DockerContainerApiClient(TheStageApiClientCore):
 
         return DockerContainerViewResponse.model_validate(response).docker_container if response else None
 
+
+    def create_container(self, request_param: DockerContainerCreateRequest) -> Optional[DockerContainerCreateResult]:
+        response = self._request(
+            method='POST',
+            url='/user-api/v1/docker-container/create',
+            data=request_param.model_dump(by_alias=True),
+            token=self.__config_provider.get_config().main.thestage_auth_token
+        )
+
+        return DockerContainerCreateResult.model_validate(response) if response else None
+
+
     def container_action(
             self,
             request_param: DockerContainerActionRequest,
@@ -87,6 +126,7 @@ class DockerContainerApiClient(TheStageApiClientCore):
         result = TheStageBaseResponse.model_validate(response) if response else None
         return result
 
+
     def get_container_business_status_map(self) -> Optional[Dict[str, str]]:
         response = self._request(
             method='POST',

thestage-0.7.0/thestage/docker_container/communication/image_command.py

@@ -0,0 +1,40 @@
+import typer
+
+from thestage.cli_command import CliCommand
+from thestage.cli_command_helper import get_command_metadata, check_command_permission
+from thestage.controllers.utils_controller import validate_config_and_get_service_factory
+from thestage.docker_container.business.container_service import ContainerService
+from thestage.i18n.translation import __
+
+app = typer.Typer(no_args_is_help=True, help=__("Manage Docker images"))
+
+@app.command(name='ls', help=__("List available docker images"), **get_command_metadata(CliCommand.CONTAINER_LS))
+def list_available_images(
+        row: int = typer.Option(
+            5,
+            '--row',
+            '-r',
+            help=__("Set number of rows displayed per page"),
+            is_eager=False,
+        ),
+        page: int = typer.Option(
+            1,
+            '--page',
+            '-p',
+            help=__("Set starting page for displaying output"),
+            is_eager=False,
+        ),
+):
+    command_name = CliCommand.CONTAINER_IMAGE_LS
+    check_command_permission(command_name)
+
+    service_factory = validate_config_and_get_service_factory()
+    container_service: ContainerService = service_factory.get_container_service()
+
+    container_service.print_docker_image_list(
+        row=row,
+        page=page,
+    )
+
+    typer.echo(__("Images listing complete"))
+    raise typer.Exit(0)

{thestage-0.6.8 → thestage-0.7.0}/thestage/docker_container/dto/container_response.py

@@ -45,3 +45,10 @@ class ContainerBusinessStatusMapperResponse(TheStageBasePaginatedResponse):
     model_config = ConfigDict(use_enum_values=True)
 
     docker_container_status_map: Dict[str, str] = Field(default={}, alias='dockerContainerStatusMap')
+
+class DockerImageDto(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    id: Optional[int] = Field(None, alias='id')
+    image: Optional[str] = Field(None, alias='image')
+    tag: Optional[str] = Field(None, alias='tag')

thestage-0.7.0/thestage/docker_container/dto/docker_container_create_request.py

@@ -0,0 +1,28 @@
+from typing import Optional, Dict
+
+from pydantic import BaseModel, Field, ConfigDict
+
+
+class DockerContainerMappings(BaseModel):
+    model_config = ConfigDict(populate_by_name=True)
+
+    directory_mappings: Dict[str, str] = Field(default_factory=dict, alias='directoryMappings')
+    port_mappings: Dict[str, str] = Field(default_factory=dict, alias='portMappings')
+
+
+class DockerContainerCreateRequest(BaseModel):
+    model_config = ConfigDict(populate_by_name=True)
+
+    instance_rented_public_id: Optional[str] = Field(None, alias='instanceRentedPublicId')
+    instance_rented_slug: Optional[str] = Field(None, alias='instanceRentedSlug')
+    selfhosted_instance_public_id: Optional[str] = Field(None, alias='selfhostedInstancePublicId')
+    selfhosted_instance_slug: Optional[str] = Field(None, alias='selfhostedInstanceSlug')
+    project_public_id: Optional[str] = Field(..., alias='projectPublicId')
+    project_slug: Optional[str] = Field(..., alias='projectSlug')
+    container_slug: str = Field(..., alias='containerSlug')
+    docker_image_id: int = Field(..., alias='dockerImageId')
+    assigned_cpus: Optional[int] = Field(None, alias='assignedCpus')
+    is_unlimited_cpus: bool = Field(False, alias='isUnlimitedCpus')
+    assigned_gpus: Optional[int] = Field(None, alias='assignedGpus')
+    is_unlimited_gpus: bool = Field(False, alias='isUnlimitedGpus')
+    mappings: DockerContainerMappings = Field(default_factory=DockerContainerMappings, alias='mappings')

thestage-0.7.0/thestage/docker_container/dto/docker_container_create_response.py

@@ -0,0 +1,11 @@
+from typing import Optional
+
+from pydantic import Field, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+
+
+class DockerContainerCreateResult(TheStageBaseResponse):
+    model_config = ConfigDict(populate_by_name=True)
+
+    docker_container_public_id: Optional[str] = Field(None, alias='dockerContainerPublicId')

thestage-0.7.0/thestage/docker_container/dto/docker_image_list_request.py

@@ -0,0 +1,11 @@
+from typing import Optional, List
+
+from pydantic import Field, ConfigDict, BaseModel
+
+from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
+
+
+class DockerImageListRequest(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    entityFilterRequest: EntityFilterRequest = Field(None, alias='entityFilterRequest')

thestage-0.7.0/thestage/docker_container/dto/docker_image_list_response.py

@@ -0,0 +1,13 @@
+from typing import List, Optional
+
+from pydantic import Field, ConfigDict
+
+from thestage.docker_container.dto.container_response import DockerImageDto
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+from thestage.services.clients.thestage_api.dtos.paginated_entity_list import PaginatedEntityList
+
+
+class DockerImageListResponse(TheStageBaseResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
+    paginatedList: PaginatedEntityList[DockerImageDto] = Field(None, alias='paginatedList')

thestage-0.7.0/thestage/docker_container/dto/image_entity.py

@@ -0,0 +1,14 @@
+from typing import Optional
+
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class DockerImageEntity(BaseModel):
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        use_enum_values=True,
+    )
+
+    image: Optional[str] = Field(None, alias='IMAGE')
+    tag: Optional[str] = Field(None, alias='TAG')