thestage 0.6.1__py3-none-any.whl → 0.6.3__py3-none-any.whl

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

@@ -1,13 +1,13 @@
-from typing import Optional, List
-
-from pydantic import Field, ConfigDict, BaseModel
-
-from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
-
-
-class InferenceSimulatorModelListForProjectRequest(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    entityFilterRequest: EntityFilterRequest = Field(None, alias='entityFilterRequest')
-    projectSlug: str = Field(None, alias='projectSlug')
+from typing import Optional, List
+
+from pydantic import Field, ConfigDict, BaseModel
+
+from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
+
+
+class InferenceSimulatorModelListForProjectRequest(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    entityFilterRequest: EntityFilterRequest = Field(None, alias='entityFilterRequest')
+    projectSlug: str = Field(None, alias='projectSlug')
     statuses: Optional[List[str]] = None

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

@@ -1,13 +1,13 @@
-
-from pydantic import Field, ConfigDict
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
-from thestage.services.clients.thestage_api.dtos.paginated_entity_list import PaginatedEntityList
-from thestage.services.project.dto.inference_simulator_dto import InferenceSimulatorDto
-from thestage.services.project.dto.inference_simulator_model_dto import InferenceSimulatorModelDto
-
-
-class InferenceSimulatorModelListForProjectResponse(TheStageBaseResponse):
-    model_config = ConfigDict(use_enum_values=True)
-
-    inferenceSimulatorModels: PaginatedEntityList[InferenceSimulatorModelDto] = Field(None, alias='inferenceSimulatorModels')
+
+from pydantic import Field, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+from thestage.services.clients.thestage_api.dtos.paginated_entity_list import PaginatedEntityList
+from thestage.services.project.dto.inference_simulator_dto import InferenceSimulatorDto
+from thestage.services.project.dto.inference_simulator_model_dto import InferenceSimulatorModelDto
+
+
+class InferenceSimulatorModelListForProjectResponse(TheStageBaseResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
+    inferenceSimulatorModels: PaginatedEntityList[InferenceSimulatorModelDto] = Field(None, alias='inferenceSimulatorModels')

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

@@ -1,11 +1,11 @@
-from typing import Dict
-
-from pydantic import Field, ConfigDict
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBasePaginatedResponse
-
-
-class InferenceSimulatorModelStatusMapperResponse(TheStageBasePaginatedResponse):
-    model_config = ConfigDict(use_enum_values=True)
-
-    inference_simulator_model_status_map: Dict[str, str] = Field(default={}, alias='inferenceSimulatorModelStatusMap')
+from typing import Dict
+
+from pydantic import Field, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBasePaginatedResponse
+
+
+class InferenceSimulatorModelStatusMapperResponse(TheStageBasePaginatedResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
+    inference_simulator_model_status_map: Dict[str, str] = Field(default={}, alias='inferenceSimulatorModelStatusMap')

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

@@ -1,11 +1,11 @@
-from typing import Dict
-
-from pydantic import Field, ConfigDict
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBasePaginatedResponse
-
-
-class InferenceSimulatorStatusMapperResponse(TheStageBasePaginatedResponse):
-    model_config = ConfigDict(use_enum_values=True)
-
-    inference_simulator_status_map: Dict[str, str] = Field(default={}, alias='inferenceSimulatorStatusMap')
+from typing import Dict
+
+from pydantic import Field, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBasePaginatedResponse
+
+
+class InferenceSimulatorStatusMapperResponse(TheStageBasePaginatedResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
+    inference_simulator_status_map: Dict[str, str] = Field(default={}, alias='inferenceSimulatorStatusMap')

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

@@ -1,18 +1,18 @@
-from typing import Optional, Dict, List
-
-from pydantic import BaseModel, ConfigDict, Field
-
-
-class DockerContainerInstalledServiceDto(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    name: Optional[str] = Field(None, alias='name')
-    port: Optional[int] = Field(None, alias='port')
-    url: Optional[str] = Field(None, alias='url')
-    variables: Dict[str, str] = Field(default_factory=dict, alias='variables')
-
-
-class DockerContainerInstalledServicesDto(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
+from typing import Optional, Dict, List
+
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class DockerContainerInstalledServiceDto(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    name: Optional[str] = Field(None, alias='name')
+    port: Optional[int] = Field(None, alias='port')
+    url: Optional[str] = Field(None, alias='url')
+    variables: Dict[str, str] = Field(default_factory=dict, alias='variables')
+
+
+class DockerContainerInstalledServicesDto(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
     installed_services: List[DockerContainerInstalledServiceDto] = Field(default_factory=list, alias='installedServices')

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

@@ -1,20 +1,20 @@
-
-from typing import Optional, List
-
-from pydantic import Field, BaseModel, ConfigDict
-
-from thestage.services.clients.thestage_api.dtos.enums.gpu_name import InstanceGpuType
-
-
-class InstanceDaemonGpuDto(BaseModel):
-
-    gpu_id: Optional[int] = Field(None, alias='gpuId')
-    name: Optional[str] = Field(None, alias='name')
-    uuid: Optional[str] = Field(None, alias='uuid')
-    type: Optional[InstanceGpuType] = Field(InstanceGpuType.UNKNOWN, alias='type')
-
-
-class InstanceDetectedGpusDto(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    gpus: List[InstanceDaemonGpuDto] = Field(default_factory=list, alias='gpus')
+
+from typing import Optional, List
+
+from pydantic import Field, BaseModel, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.enums.gpu_name import InstanceGpuType
+
+
+class InstanceDaemonGpuDto(BaseModel):
+
+    gpu_id: Optional[int] = Field(None, alias='gpuId')
+    name: Optional[str] = Field(None, alias='name')
+    uuid: Optional[str] = Field(None, alias='uuid')
+    type: Optional[InstanceGpuType] = Field(InstanceGpuType.UNKNOWN, alias='type')
+
+
+class InstanceDetectedGpusDto(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    gpus: List[InstanceDaemonGpuDto] = Field(default_factory=list, alias='gpus')

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

@@ -1,71 +1,71 @@
-from datetime import datetime
-from typing import Optional, List, 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')
-    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')
-    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):
-    paginated_list: Optional[PaginatedEntityList[InstanceRentedDto]] = Field(None, alias='paginatedEntityList')
-
-
-class InstanceRentedItemResponse(TheStageBasePaginatedResponse):
-    model_config = ConfigDict(use_enum_values=True)
-
-    instance_rented: Optional[InstanceRentedDto] = Field(None, alias='instanceRented')
-
-
-class InstanceRentedBusinessStatusMapperResponse(TheStageBasePaginatedResponse):
-    model_config = ConfigDict(use_enum_values=True)
-
-    instance_rented_business_status_map: Dict[str, str] = Field(default={}, alias='instanceRentedBusinessStatusMap')
+from datetime import datetime
+from typing import Optional, List, 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')
+    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')
+    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):
+    paginated_list: Optional[PaginatedEntityList[InstanceRentedDto]] = Field(None, alias='paginatedEntityList')
+
+
+class InstanceRentedItemResponse(TheStageBasePaginatedResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
+    instance_rented: Optional[InstanceRentedDto] = Field(None, alias='instanceRented')
+
+
+class InstanceRentedBusinessStatusMapperResponse(TheStageBasePaginatedResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
+    instance_rented_business_status_map: Dict[str, str] = Field(default={}, alias='instanceRentedBusinessStatusMap')

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

@@ -1,7 +1,7 @@
-from pydantic import Field, ConfigDict, BaseModel
-
-
-class DockerContainerLogStreamRequest(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    dockerContainerId: int = Field(None, alias='dockerContainerId')
+from pydantic import Field, ConfigDict, BaseModel
+
+
+class DockerContainerLogStreamRequest(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    dockerContainerId: int = Field(None, alias='dockerContainerId')

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

@@ -1,13 +1,13 @@
-from typing import Optional
-
-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')
-    lastLogId: Optional[str] = Field(None, alias='lastLogId')
-    dockerContainerId: Optional[int] = Field(None, alias='dockerContainerId')
-    lastLogTimestamp: Optional[str] = Field(None, alias='lastLogTimestamp')
+from typing import Optional
+
+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')
+    lastLogId: Optional[str] = Field(None, alias='lastLogId')
+    dockerContainerId: Optional[int] = Field(None, alias='dockerContainerId')
+    lastLogTimestamp: Optional[str] = Field(None, alias='lastLogTimestamp')

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

@@ -1,14 +1,14 @@
-from typing import Optional, List
-
-from pydantic import Field, ConfigDict, BaseModel
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
-from thestage.services.logging.dto.log_message import LogMessage
-
-
-class LogPollingResponse(TheStageBaseResponse):
-    model_config = ConfigDict(use_enum_values=True)
-
-    logs: Optional[List[LogMessage]] = Field(None, alias='logs')
-    lastLogId: Optional[str] = Field(None, alias='lastLogId')
-    lastLogTimestamp: Optional[str] = Field(None, alias='lastLogTimestamp')
+from typing import Optional, List
+
+from pydantic import Field, ConfigDict, BaseModel
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+from thestage.services.logging.dto.log_message import LogMessage
+
+
+class LogPollingResponse(TheStageBaseResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
+    logs: Optional[List[LogMessage]] = Field(None, alias='logs')
+    lastLogId: Optional[str] = Field(None, alias='lastLogId')
+    lastLogTimestamp: Optional[str] = Field(None, alias='lastLogTimestamp')

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

@@ -1,7 +1,7 @@
-from pydantic import Field, ConfigDict, BaseModel
-
-
-class TaskLogStreamRequest(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    taskId: int = Field(None, alias='taskId')
+from pydantic import Field, ConfigDict, BaseModel
+
+
+class TaskLogStreamRequest(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    taskId: int = Field(None, alias='taskId')

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

@@ -1,21 +1,21 @@
-from typing import Optional
-
-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')
-    limit: Optional[int] = Field(None, alias='limit')
-    ascendingOrder: Optional[bool] = Field(None, alias='ascendingOrder')
+from typing import Optional
+
+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')
+    limit: Optional[int] = Field(None, alias='limit')
+    ascendingOrder: Optional[bool] = Field(None, alias='ascendingOrder')

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

@@ -1,14 +1,14 @@
-from typing import Optional, List
-
-from pydantic import Field, ConfigDict, BaseModel
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
-from thestage.services.logging.dto.log_message import LogMessage
-
-
-class UserLogsQueryResponse(TheStageBaseResponse):
-    model_config = ConfigDict(use_enum_values=True)
-
-    queryResult: Optional[List[LogMessage]] = Field(None, alias='queryResult')
-    entriesFound: Optional[int] = Field(None, alias='entriesFound')
-    totalCount: Optional[int] = Field(None, alias='totalCount')
+from typing import Optional, List
+
+from pydantic import Field, ConfigDict, BaseModel
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+from thestage.services.logging.dto.log_message import LogMessage
+
+
+class UserLogsQueryResponse(TheStageBaseResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
+    queryResult: Optional[List[LogMessage]] = Field(None, alias='queryResult')
+    entriesFound: Optional[int] = Field(None, alias='entriesFound')
+    totalCount: Optional[int] = Field(None, alias='totalCount')

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

@@ -1,11 +1,11 @@
-from typing import List, Optional, Any, Generic, TypeVar
-
-from pydantic import Field, ConfigDict, BaseModel
-
-from thestage.services.clients.thestage_api.dtos.pagination_data import PaginationData
-
-T = TypeVar('T')
-
-class PaginatedEntityList(BaseModel, Generic[T]):
-    entities: List[T] = Field(default_factory=list, alias='entities')
-    pagination_data: Optional[PaginationData] = Field(None, alias='paginationData')
+from typing import List, Optional, Any, Generic, TypeVar
+
+from pydantic import Field, ConfigDict, BaseModel
+
+from thestage.services.clients.thestage_api.dtos.pagination_data import PaginationData
+
+T = TypeVar('T')
+
+class PaginatedEntityList(BaseModel, Generic[T]):
+    entities: List[T] = Field(default_factory=list, alias='entities')
+    pagination_data: Optional[PaginationData] = Field(None, alias='paginationData')

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

@@ -1,10 +1,10 @@
-from typing import Optional
-
-from pydantic import BaseModel, Field
-
-
-class PaginationData(BaseModel):
-    current_page: Optional[int] = Field(None, alias='currentPage')
-    limit: Optional[int] = Field(None, alias='limit')
-    total_pages: Optional[int] = Field(None, alias='totalPages')
-    objects_count: Optional[int] = Field(None, alias='objectsCount')
+from typing import Optional
+
+from pydantic import BaseModel, Field
+
+
+class PaginationData(BaseModel):
+    current_page: Optional[int] = Field(None, alias='currentPage')
+    limit: Optional[int] = Field(None, alias='limit')
+    total_pages: Optional[int] = Field(None, alias='totalPages')
+    objects_count: Optional[int] = Field(None, alias='objectsCount')

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

@@ -1,14 +1,14 @@
-from typing import Optional
-
-from pydantic import BaseModel, Field, ConfigDict
-
-from thestage.services.clients.thestage_api.dtos.enums.currency_type import CurrencyTypeEnumDto
-
-
-class PriceDefinitionDto(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    currency: CurrencyTypeEnumDto = Field(CurrencyTypeEnumDto.UNKNOWN, alias='currency')
-    cents_amount_in_unit: Optional[int] = Field(None, alias='centsAmountInUnit')
-    cents_value: Optional[str] = Field(None, alias='centsValue')
-    decimal_value: Optional[str] = Field(None, alias='decimalValue')
+from typing import Optional
+
+from pydantic import BaseModel, Field, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.enums.currency_type import CurrencyTypeEnumDto
+
+
+class PriceDefinitionDto(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    currency: CurrencyTypeEnumDto = Field(CurrencyTypeEnumDto.UNKNOWN, alias='currency')
+    cents_amount_in_unit: Optional[int] = Field(None, alias='centsAmountInUnit')
+    cents_value: Optional[str] = Field(None, alias='centsValue')
+    decimal_value: Optional[str] = Field(None, alias='decimalValue')

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

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

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

@@ -1,10 +1,10 @@
-from pydantic import Field, ConfigDict, BaseModel
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
-
-
-class ProjectGetDeploySshKeyResponse(TheStageBaseResponse):
-    model_config = ConfigDict(use_enum_values=True)
-
-    publicKey: str = Field(None, alias='publicKey')
-    privateKey: str = Field(None, alias='privateKey')
+from pydantic import Field, ConfigDict, BaseModel
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+
+
+class ProjectGetDeploySshKeyResponse(TheStageBaseResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
+    publicKey: str = Field(None, alias='publicKey')
+    privateKey: str = Field(None, alias='privateKey')

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

@@ -1,9 +1,9 @@
-from typing import Optional
-
-from pydantic import Field, ConfigDict, BaseModel
-
-
-class ProjectPushInferenceSimulatorModelRequest(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
+from typing import Optional
+
+from pydantic import Field, ConfigDict, BaseModel
+
+
+class ProjectPushInferenceSimulatorModelRequest(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
     slug: Optional[str] = Field(None, alias='slug')

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

@@ -1,7 +1,7 @@
-from pydantic import ConfigDict
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
-
-
-class ProjectPushInferenceSimulatorModelResponse(TheStageBaseResponse):
+from pydantic import ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+
+
+class ProjectPushInferenceSimulatorModelResponse(TheStageBaseResponse):
     model_config = ConfigDict(use_enum_values=True)