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

thestage/services/clients/thestage_api/dtos/container_response.py

@@ -2,12 +2,8 @@ from typing import Optional, List, Dict
 
 from pydantic import Field, BaseModel, ConfigDict
 
-from thestage.services.clients.thestage_api.dtos.docker_container_assigned_device import DockerContainerAssignedDeviceDto
 from thestage.services.clients.thestage_api.dtos.docker_container_mapping import DockerContainerMappingDto
-from thestage.services.clients.thestage_api.dtos.enums.container_pending_action import DockerContainerAction
-from thestage.services.clients.thestage_api.dtos.enums.container_status import DockerContainerStatus
 from thestage.services.clients.thestage_api.dtos.frontend_status import FrontendStatusDto
-from thestage.services.clients.thestage_api.dtos.installed_service import DockerContainerInstalledServicesDto
 from thestage.services.clients.thestage_api.dtos.instance_rented_response import InstanceRentedDto
 from thestage.services.clients.thestage_api.dtos.project_response import ProjectDto
 from thestage.services.clients.thestage_api.dtos.selfhosted_instance_response import SelfHostedInstanceDto
@@ -18,30 +14,18 @@ from thestage.services.clients.thestage_api.dtos.pagination_data import Paginati
 class DockerContainerDto(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
 
-    id: Optional[int] = Field(None, alias='id')
-    instance_rented_id: Optional[int] = Field(None, alias='instanceRentedId')
-    selfhosted_instance_id: Optional[int] = Field(None, alias='selfhostedInstanceId')
+    public_id: Optional[str] = Field(None, alias='publicId')
     instance_rented: Optional[InstanceRentedDto] = Field(None, alias='instanceRented')
     selfhosted_instance: Optional[SelfHostedInstanceDto] = Field(None, alias='selfhostedInstance')
-
-    project_id: Optional[int] = Field(None, alias='projectId')
     project: Optional[ProjectDto] = Field(None, alias='project')
-
-    is_default: Optional[bool] = Field(None, alias='isDefault')
     system_name: Optional[str] = Field(None, alias='systemName')
     title: Optional[str] = Field(None, alias='title')
     slug: Optional[str] = Field(None, alias='slug')
     docker_image: Optional[str] = Field(None, alias='dockerImage')
-    assigned_devices: Optional[DockerContainerAssignedDeviceDto] = Field(None, alias='assignedDevices')
     mappings: Optional[DockerContainerMappingDto] = Field(None, alias='mappings')
 
     frontend_status: Optional[FrontendStatusDto] = Field(None, alias='frontendStatus')
 
-    pending_action: Optional[DockerContainerAction] = Field(None, alias='pendingAction')
-    installed_services: Optional[DockerContainerInstalledServicesDto] = Field(None, alias='installedServices')
-    created_at: Optional[str] = Field(None, alias='createdAt')
-    updated_at: Optional[str] = Field(None, alias='updatedAt')
-
 
 class DockerContainerPaginated(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
@@ -52,10 +36,6 @@ class DockerContainerPaginated(BaseModel):
     total_pages: Optional[int] = Field(None, alias='totalPages')
     pagination_data: Optional[PaginationData] = Field(None, alias='paginationData')
 
-#
-# class DockerContainerListResponse(TheStageBasePaginatedResponse):
-#     paginated_list: Optional[DockerContainerPaginated] = Field(None, alias='paginatedList')
-
 
 class DockerContainerViewResponse(TheStageBaseResponse):
     docker_container: Optional[DockerContainerDto] = Field(None, alias='dockerContainer')

thestage/services/clients/thestage_api/dtos/docker_container_controller/docker_container_list_request.py

@@ -10,4 +10,5 @@ class DockerContainerListRequest(BaseModel):
 
     entityFilterRequest: EntityFilterRequest = Field(None, alias='entityFilterRequest')
     statuses: Optional[List[str]] = Field(None, alias='statuses')
-    projectId: Optional[int] = Field(None, alias='projectId')
+    projectPublicId: Optional[str] = Field(None, alias='projectPublicId')
+    projectSlug: Optional[str] = Field(None, alias='projectSlug')

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

@@ -8,7 +8,11 @@ from thestage.services.clients.thestage_api.dtos.entity_filter_request import En
 class DeployInferenceModelToInstanceRequest(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
 
+    modelPublicId: Optional[str] = Field(None, alias='modelPublicId')
     modelSlug: Optional[str] = Field(None, alias='modelSlug')
+
+    instanceRentedPublicId: Optional[str] = Field(None, alias='instanceRentedPublicId')
     instanceRentedSlug: Optional[str] = Field(None, alias='instanceRentedSlug')
+
+    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

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

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

@@ -8,5 +8,6 @@ from thestage.services.clients.thestage_api.dtos.entity_filter_request import En
 class DeployInferenceModelToSagemakerRequest(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
 
+    modelPublicId: Optional[str] = Field(None, alias='modelPublicId')
     modelSlug: Optional[str] = Field(None, alias='modelSlug')
     arn: Optional[str] = Field(None, alias='arn')

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

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

thestage/services/clients/thestage_api/dtos/inference_controller/{inference_simulator_list_for_project_request.py → inference_simulator_list_request.py}

@@ -5,10 +5,11 @@ from pydantic import Field, ConfigDict, BaseModel
 from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
 
 
-class InferenceSimulatorListForProjectRequest(BaseModel):
+class InferenceSimulatorListRequest(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
 
     entityFilterRequest: EntityFilterRequest = Field(None, alias='entityFilterRequest')
-    projectSlug: str = Field(None, alias='projectSlug')
+    projectPublicId: Optional[str] = Field(None, alias='projectPublicId')
+    projectSlug: Optional[str] = Field(None, alias='projectSlug')
     statuses: Optional[List[str]] = Field(None, alias='statuses')
 

thestage/services/clients/thestage_api/dtos/inference_controller/{inference_simulator_list_for_project_response.py → inference_simulator_list_response.py}

@@ -6,7 +6,7 @@ from thestage.services.clients.thestage_api.dtos.paginated_entity_list import Pa
 from thestage.services.project.dto.inference_simulator_dto import InferenceSimulatorDto
 
 
-class InferenceSimulatorListForProjectResponse(TheStageBaseResponse):
+class InferenceSimulatorListResponse(TheStageBaseResponse):
     model_config = ConfigDict(use_enum_values=True)
 
     inferenceSimulators: PaginatedEntityList[InferenceSimulatorDto] = Field(None, alias='inferenceSimulators')

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

@@ -9,5 +9,6 @@ class InferenceSimulatorModelListForProjectRequest(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
 
     entityFilterRequest: EntityFilterRequest = Field(None, alias='entityFilterRequest')
-    projectSlug: str = Field(None, alias='projectSlug')
+    projectPublicId: Optional[str] = Field(None, alias='projectPublicId')
+    projectSlug: Optional[str] = Field(None, alias='projectSlug')
     statuses: Optional[List[str]] = None

thestage/services/clients/thestage_api/dtos/instance_rented_response.py

@@ -1,58 +1,25 @@
-from datetime import datetime
-from typing import Optional, List, Dict
+from typing import Optional, Dict
 
 from pydantic import Field, BaseModel, ConfigDict
 
-from thestage.services.clients.thestage_api.dtos.enums.daemon_status import DaemonStatus
 from thestage.services.clients.thestage_api.dtos.frontend_status import FrontendStatusDto
-from thestage.services.clients.thestage_api.dtos.instance_detected_gpus import InstanceDetectedGpusDto
-from thestage.services.clients.thestage_api.dtos.enums.instance_type import InstanceTypeEnumDto
 from thestage.services.clients.thestage_api.dtos.base_response import TheStageBasePaginatedResponse
 from thestage.services.clients.thestage_api.dtos.paginated_entity_list import PaginatedEntityList
-from thestage.services.clients.thestage_api.dtos.cloud_provider_region import CloudProviderRegionDto
-from thestage.services.clients.thestage_api.dtos.enums.disk_type import DiskTypeEnumDto
 from thestage.services.clients.thestage_api.dtos.enums.gpu_name import InstanceGpuType
-from thestage.services.clients.thestage_api.dtos.enums.power_status import PowerStatusEnumDto
-from thestage.services.clients.thestage_api.dtos.enums.instance_rented_status import InstanceRentedBusinessStatus
-from thestage.services.clients.thestage_api.dtos.price_definition import PriceDefinitionDto
 from thestage.services.clients.thestage_api.dtos.enums.cpu_type import InstanceCpuType
 
 
 class InstanceRentedDto(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
 
-    id: Optional[int] = Field(None, alias='id')
-    display_price_per_hour: Optional[PriceDefinitionDto] = Field(None, alias='displayPricePerHour')
-    total_money_spent: Optional[List[PriceDefinitionDto]] = Field(default_factory=list, alias='totalMoneySpent')
-    client_id: Optional[int] = Field(None, alias='clientId')
-    instance_type: Optional[InstanceTypeEnumDto] = Field(InstanceTypeEnumDto.UNKNOWN, alias='instanceType', validate_default=True)
-    provider_id: Optional[str] = Field(None, alias='providerId')
-    cloud_provider_region_id: Optional[int] = Field(None, alias='cloudProviderRegionId')
-    cloud_provider_region: Optional[CloudProviderRegionDto] = Field(None, alias='cloudProviderRegion')
+    public_id: Optional[str] = Field(None, alias='publicId')
+    slug: Optional[str] = Field(None, alias='slug')
     cpu_type: Optional[InstanceCpuType] = Field(InstanceCpuType.UNKNOWN, alias='cpuType')
-    cpu_name: Optional[str] = Field(None, alias='cpuName')
     cpu_cores: Optional[int] = Field(None, alias='cpuCores')
-    gpu_type: Optional[InstanceGpuType] = Field(Optional[InstanceGpuType.UNKNOWN], alias='gpuType')
-    gpu_name: Optional[str] = Field(None, alias='gpuName')
-    gpu_memory_gb: Optional[int] = Field(None, alias='gpuMemoryGb')
-    gpu_count: Optional[int] = Field(None, alias='gpuCount')
-    disk_type: Optional[DiskTypeEnumDto] = Field(DiskTypeEnumDto.UNKNOWN, alias='diskType')
-    disk_size_gb: Optional[int] = Field(None, alias='diskSizeGb')
-    ram_size_gb: Optional[int] = Field(None, alias='ramSizeGb')
-    average_renting_time_minutes: Optional[int] = Field(None, alias='averageRentingTimeMinutes')
-
-    business_status: InstanceRentedBusinessStatus = Field(InstanceRentedBusinessStatus.UNKNOWN, alias='businessStatus')
+    gpu_type: Optional[InstanceGpuType] = Field(InstanceGpuType.UNKNOWN, alias='gpuType')
     frontend_status: Optional[FrontendStatusDto] = Field(None, alias='frontendStatus')
-
-    slug: Optional[str] = Field(None, alias='slug')
-    title: Optional[str] = Field(None, alias='title')
-    power_status: PowerStatusEnumDto = Field(PowerStatusEnumDto.UNKNOWN, alias='powerStatus')
-    created_at: Optional[datetime] = Field(None, alias='createdAt')
-    updated_at: Optional[datetime] = Field(None, alias='updatedAt')
     ip_address: Optional[str] = Field(None, alias='ipAddress')
     host_username: Optional[str] = Field(None, alias='hostUsername')
-    detected_gpus: Optional[InstanceDetectedGpusDto] = Field(None, alias='detectedGpus')
-    daemon_status: Optional[DaemonStatus] = Field(DaemonStatus.UNKNOWN, alias='daemonStatus')
 
 
 class InstanceRentedListResponse(TheStageBasePaginatedResponse):

thestage/services/clients/thestage_api/dtos/logging_controller/log_polling_request.py

@@ -6,8 +6,8 @@ from pydantic import Field, ConfigDict, BaseModel
 class LogPollingRequest(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
 
-    taskId: Optional[int] = Field(None, alias='taskId')
-    inferenceSimulatorId: Optional[int] = Field(None, alias='inferenceSimulatorId')
+    taskPublicId: Optional[str] = Field(None, alias='taskPublicId')
+    inferenceSimulatorPublicId: Optional[str] = Field(None, alias='inferenceSimulatorPublicId')
     lastLogId: Optional[str] = Field(None, alias='lastLogId')
-    dockerContainerId: Optional[int] = Field(None, alias='dockerContainerId')
+    dockerContainerPublicId: Optional[str] = Field(None, alias='dockerContainerPublicId')
     lastLogTimestamp: Optional[str] = Field(None, alias='lastLogTimestamp')

thestage/services/clients/thestage_api/dtos/logging_controller/user_logs_query_request.py

@@ -6,16 +6,8 @@ from pydantic import Field, ConfigDict, BaseModel
 class UserLogsQueryRequest(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
 
-    containerRunId: Optional[int] = Field(None, alias='containerRunId')
-    containerId: Optional[int] = Field(None, alias='containerId')
-    message: Optional[str] = Field(None, alias='message')
-    instanceId: Optional[int] = Field(None, alias='instanceId')
-    instanceType: Optional[str] = Field(None, alias='instanceType') # ENUM
-    taskId: Optional[int] = Field(None, alias='taskId')
-    inferenceSimulatorId: Optional[int] = Field(None, alias='inferenceSimulatorId')
-    fromTimestamp: Optional[str] = Field(None, alias='fromTimestamp')
-    toTimestamp: Optional[str] = Field(None, alias='toTimestamp')
-    logType: Optional[str] = Field(None, alias='logType') # ENUM
-    offset: Optional[int] = Field(None, alias='offset')
+    containerPublicId: Optional[str] = Field(None, alias='containerPublicId')
+    taskPublicId: Optional[str] = Field(None, alias='taskPublicId')
+    inferenceSimulatorPublicId: Optional[str] = Field(None, alias='inferenceSimulatorPublicId')
     limit: Optional[int] = Field(None, alias='limit')
     ascendingOrder: Optional[bool] = Field(None, alias='ascendingOrder')

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

@@ -4,4 +4,5 @@ from pydantic import Field, ConfigDict, BaseModel
 class ProjectGetDeploySshKeyRequest(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
 
+    projectPublicId: str = Field(None, alias='projectPublicId')
     projectSlug: str = Field(None, alias='projectSlug')

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

@@ -6,4 +6,5 @@ from pydantic import Field, ConfigDict, BaseModel
 class ProjectPushInferenceSimulatorModelRequest(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
 
-    slug: Optional[str] = Field(None, alias='slug')
+    inferenceSimulatorPublicId: Optional[str] = Field(None, alias='inferenceSimulatorPublicId')
+    inferenceSimulatorSlug: Optional[str] = Field(None, alias='inferenceSimulatorSlug')

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

@@ -6,10 +6,8 @@ from pydantic import Field, ConfigDict, BaseModel
 class ProjectRunTaskRequest(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
 
-    projectSlug: str = Field(None, alias='projectSlug')
-    instanceRentedSlug: Optional[str] = Field(None, alias='instanceRentedSlug')
-    selfhostedInstanceSlug: Optional[str] = Field(None, alias='selfhostedInstanceSlug')
-    dockerContainerSlug: Optional[str] = Field(None, alias='dockerContainerSlug')
+    projectPublicId: str = Field(None, alias='projectPublicId')
+    dockerContainerPublicId: str = Field(None, alias='dockerContainerPublicId')
     commitHash: Optional[str] = Field(None, alias='commitHash')
     runCommand: str = Field(None, alias='runCommand')
     taskTitle: Optional[str] = Field(None, alias='taskTitle')

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/clients/thestage_api/dtos/project_controller/project_start_inference_simulator_request.py

@@ -6,9 +6,11 @@ from pydantic import Field, ConfigDict, BaseModel
 class ProjectStartInferenceSimulatorRequest(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
 
-    projectSlug: str = Field(None, alias='projectSlug')
-    instanceRentedUId: Optional[str] = Field(None, alias='instanceRentedUId')
-    selfhostedInstanceUId: Optional[str] = Field(None, alias='selfhostedInstanceUId')
+    projectPublicId: str = Field(None, alias='projectPublicId')
+    instanceRentedPublicId: Optional[str] = Field(None, alias='instanceRentedPublicId')
+    instanceRentedSlug: Optional[str] = Field(None, alias='instanceRentedSlug')
+    selfhostedInstancePublicId: Optional[str] = Field(None, alias='selfhostedInstancePublicId')
+    selfhostedInstanceSlug: Optional[str] = Field(None, alias='selfhostedInstanceSlug')
     commitHash: Optional[str] = Field(None, alias='commitHash')
     inferenceDir: Optional[str] = Field(None, alias='inferenceDir')
     isSkipInstallation: Optional[bool] = Field(False, alias='isSkipInstallation')

thestage/services/clients/thestage_api/dtos/project_response.py

@@ -3,29 +3,17 @@ from typing import Optional, List
 from pydantic import Field, BaseModel, ConfigDict
 
 from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
-from thestage.services.clients.thestage_api.dtos.selfhosted_instance_response import SelfHostedInstanceDto
-from thestage.services.clients.thestage_api.dtos.instance_rented_response import InstanceRentedDto
 
 
 class ProjectDto(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
 
-    id: Optional[int] = Field(None, alias='id')
-    client_id: Optional[int] = Field(None, alias='сlientId')
-    ssh_key_deploy_id: Optional[int] = Field(None, alias='sshKeyDeployId')
-    name: Optional[str] = Field(None, alias='name')
+    public_id: Optional[str] = Field(None, alias='publicId')
     slug: Optional[str] = Field(None, alias='slug')
-    description: Optional[str] = Field(None, alias='description')
-    github_username: Optional[str] = Field(None, alias='githubCollaboratorUsername')
-    last_commit_hash: Optional[str] = Field(None, alias='lastCommitHash')
-    last_commit_description: Optional[str] = Field(None, alias='lastCommitDescription')
     git_repository_url: Optional[str] = Field(None, alias='gitRepositoryUrl')
     git_repository_name: Optional[str] = Field(None, alias='gitRepositoryName')
-    task_count: Optional[int] = Field(None, alias='taskCount')
-    favourite_task_count: Optional[int] = Field(None, alias='favouriteTaskCount')
-    last_task_run_date: Optional[str] = Field(None, alias='lastTaskRunDate')
-    created_at: Optional[str] = Field(None, alias='createdAt')
-    updated_at: Optional[str] = Field(None, alias='updatedAt')
+    last_commit_hash: Optional[str] = Field(None, alias='lastCommitHash')
+    last_commit_description: Optional[str] = Field(None, alias='lastCommitDescription')
 
 
 class ProjectViewResponse(TheStageBaseResponse):

thestage/services/clients/thestage_api/dtos/selfhosted_instance_response.py

@@ -1,43 +1,25 @@
 from datetime import datetime
-from typing import Optional, List, Dict
+from typing import Optional, Dict
 
 from pydantic import Field, BaseModel, ConfigDict
 
 from thestage.services.clients.thestage_api.dtos.base_response import TheStageBasePaginatedResponse
 from thestage.services.clients.thestage_api.dtos.frontend_status import FrontendStatusDto
 from thestage.services.clients.thestage_api.dtos.paginated_entity_list import PaginatedEntityList
-from thestage.services.clients.thestage_api.dtos.pagination_data import PaginationData
 from thestage.services.clients.thestage_api.dtos.instance_detected_gpus import InstanceDetectedGpusDto
-from thestage.services.clients.thestage_api.dtos.enums.instance_type import InstanceTypeEnumDto
-from thestage.services.clients.thestage_api.dtos.enums.selfhosted_status import SelfhostedBusinessStatus
 from thestage.services.clients.thestage_api.dtos.enums.cpu_type import InstanceCpuType
 
 
 class SelfHostedInstanceDto(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
 
-    instance_type: Optional[InstanceTypeEnumDto] = Field(InstanceTypeEnumDto.UNKNOWN, alias='instanceType')
-    id: Optional[int] = Field(None, alias='id')
-    ram_size_gb: Optional[int] = Field(None, alias='ramSizeGb')
-    client_id: Optional[int] = Field(None, alias='clientId')
+    public_id: Optional[str] = Field(None, alias='publicId')
     slug: Optional[str] = Field(None, alias='slug')
-    title: Optional[str] = Field(None, alias='title')
     cpu_type: Optional[InstanceCpuType] = Field(InstanceCpuType.UNKNOWN, alias='cpuType')
-    cpu_name: Optional[str] = Field(None, alias='cpuName')
     cpu_cores: Optional[int] = Field(None, alias='cpuCores')
-    gpu_count: Optional[int] = Field(None, alias='gpuCount')
     detected_gpus: Optional[InstanceDetectedGpusDto] = Field(None, alias='detectedGpus')
-    hardware_profile_id: Optional[str] = Field(None, alias='hardwareProfileId')
-    disk_size_gb: Optional[int] = Field(None, alias='diskSizeGb')
-
     frontend_status: Optional[FrontendStatusDto] = Field(None, alias='frontendStatus')
-    business_status: SelfhostedBusinessStatus = Field(SelfhostedBusinessStatus.UNKNOWN, alias='businessStatus')
-
-    updated_at: Optional[datetime] = Field(None, alias='updatedAt')
-    created_at: Optional[datetime] = Field(None, alias='createdAt')
     ip_address: Optional[str] = Field(None, alias='ipAddress')
-    # not present now
-    host_username: Optional[str] = Field(None, alias='hostUsername')
 
 
 class SelfHostedInstanceListResponse(TheStageBasePaginatedResponse):

thestage/services/clients/thestage_api/dtos/ssh_key_controller/add_ssh_key_to_user_response.py

@@ -9,4 +9,4 @@ from thestage.services.task.dto.task_dto import TaskDto
 class AddSshKeyToUserResponse(TheStageBaseResponse):
     model_config = ConfigDict(use_enum_values=True)
 
-    sshKeyPairId: Optional[int] = Field(None, alias='sshKeyPairId')
+    sshKeyPairPublicId: Optional[str] = Field(None, alias='sshKeyPairPublicId')

thestage/services/clients/thestage_api/dtos/ssh_key_controller/add_ssh_public_key_to_instance_request.py

@@ -1,8 +1,10 @@
+from typing import Optional
+
 from pydantic import Field, ConfigDict, BaseModel
 
 
 class AddSshPublicKeyToInstanceRequest(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
 
-    instanceRentedId: int = Field(None, alias='instanceRentedId')
-    sshPublicKeyId: int = Field(None, alias='sshPublicKeyId')
+    instanceRentedPublicId: str = Field(None, alias='instanceRentedPublicId')
+    sshPublicKeyPublicId: str = Field(None, alias='sshPublicKeyPublicId')

thestage/services/clients/thestage_api/dtos/ssh_key_controller/is_user_has_public_ssh_key_response.py

@@ -9,4 +9,4 @@ class IsUserHasSshPublicKeyResponse(TheStageBaseResponse):
     model_config = ConfigDict(use_enum_values=True)
 
     isUserHasPublicKey: bool = Field(None, alias='isUserHasPublicKey')
-    sshKeyPairId: Optional[int] = Field(None, alias='sshKeyPairId')
+    sshKeyPairPublicId: Optional[str] = Field(None, alias='sshKeyPairPublicId')

thestage/services/clients/thestage_api/dtos/task_controller/task_list_for_project_request.py

@@ -1,3 +1,5 @@
+from typing import Optional
+
 from pydantic import Field, ConfigDict, BaseModel
 
 from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
@@ -7,4 +9,5 @@ class TaskListForProjectRequest(BaseModel):
     model_config = ConfigDict(use_enum_values=True)
 
     entityFilterRequest: EntityFilterRequest = Field(None, alias='entityFilterRequest')
-    projectSlug: str = Field(None, alias='projectSlug')
+    projectPublicId: Optional[str] = Field(None, alias='projectPublicId')
+    projectSlug: Optional[str] = Field(None, alias='projectSlug')

thestage/services/clients/thestage_api/dtos/task_controller/task_view_response.py

@@ -8,5 +8,3 @@ from thestage.services.task.dto.task_dto import TaskDto
 
 class TaskViewResponse(TheStageBaseResponse):
     task: Optional[TaskDto] = Field(None, alias='task')
-    task_view_url: Optional[str] = Field(None, alias='websiteTaskViewUrl')
-    task_output_url: Optional[str] = Field(None, alias='websiteTaskOutputUrl')

thestage/services/config_provider/config_provider.py

@@ -76,11 +76,11 @@ class ConfigProvider:
         self.__save_config_file(data=project_config.model_dump(), file_path=project_data_filepath)
 
 
-    def save_project_deploy_ssh_key(self, deploy_ssh_key: str, project_slug: str, project_id: int) -> str:
+    def save_project_deploy_ssh_key(self, deploy_ssh_key: str, project_public_id: str) -> str:
         deploy_key_dirpath = self.get_global_config_path().joinpath('project_deploy_keys')
         self._file_system_service.create_if_not_exists_dir(deploy_key_dirpath)
 
-        deploy_key_filepath = deploy_key_dirpath.joinpath(f'project_deploy_key_{project_id}_{project_slug}')
+        deploy_key_filepath = deploy_key_dirpath.joinpath(f'project_deploy_key__{project_public_id}')
         self._file_system_service.create_if_not_exists_file(deploy_key_filepath)
 
         text_file = open(deploy_key_filepath, "w")