thestage 0.5.47__py3-none-any.whl → 0.5.49__py3-none-any.whl

thestage/services/project/dto/inference_simulator_dto.py

@@ -1,23 +1,23 @@
-from typing import Optional
-from pydantic import BaseModel, ConfigDict, Field
-
-class InferenceSimulatorDto(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    id: Optional[int] = Field(None, alias="id")
-    client_id: Optional[int] = Field(None, alias="clientId")
-    title: Optional[str] = Field(None, alias="title")
-    slug: Optional[str] = Field(None, alias="slug")
-    docker_container_id: Optional[int] = Field(None, alias="dockerContainerId")
-    instance_rented_id: Optional[int] = Field(None, alias="instanceRentedId")
-    selfhosted_instance_id: Optional[int] = Field(None, alias="selfhostedInstanceId")
-    project_id: Optional[int] = Field(None, alias="projectId")
-    status: Optional[str] = Field(None, alias="status")
-    commit_hash: Optional[str] = Field(None, alias="commitHash")
-    http_endpoint: Optional[str] = Field(None, alias="httpEndpoint")
-    grpc_endpoint: Optional[str] = Field(None, alias="grpcEndpoint")
-    metrics_endpoint: Optional[str] = Field(None, alias="metricsEndpoint")
-    inference_metadata: Optional[str] = Field(None, alias="inferenceMetadata")
-    qlip_serve_metadata: Optional[str] = Field(None, alias="qlipServeMetadata")
-    created_at: Optional[str] = Field(None, alias="createdAt")
+from typing import Optional
+from pydantic import BaseModel, ConfigDict, Field
+
+class InferenceSimulatorDto(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    id: Optional[int] = Field(None, alias="id")
+    client_id: Optional[int] = Field(None, alias="clientId")
+    title: Optional[str] = Field(None, alias="title")
+    slug: Optional[str] = Field(None, alias="slug")
+    docker_container_id: Optional[int] = Field(None, alias="dockerContainerId")
+    instance_rented_id: Optional[int] = Field(None, alias="instanceRentedId")
+    selfhosted_instance_id: Optional[int] = Field(None, alias="selfhostedInstanceId")
+    project_id: Optional[int] = Field(None, alias="projectId")
+    status: Optional[str] = Field(None, alias="status")
+    commit_hash: Optional[str] = Field(None, alias="commitHash")
+    http_endpoint: Optional[str] = Field(None, alias="httpEndpoint")
+    grpc_endpoint: Optional[str] = Field(None, alias="grpcEndpoint")
+    metrics_endpoint: Optional[str] = Field(None, alias="metricsEndpoint")
+    inference_metadata: Optional[str] = Field(None, alias="inferenceMetadata")
+    qlip_serve_metadata: Optional[str] = Field(None, alias="qlipServeMetadata")
+    created_at: Optional[str] = Field(None, alias="createdAt")
     updated_at: Optional[str] = Field(None, alias="updatedAt")

thestage/services/project/dto/inference_simulator_model_dto.py

@@ -1,21 +1,21 @@
-from typing import Optional
-from pydantic import BaseModel, ConfigDict, Field
-
-class InferenceSimulatorModelDto(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    id: Optional[int] = Field(None, alias='id')
-    slug: Optional[str] = Field(None, alias='slug')
-    client_id: Optional[int] = Field(None, alias='clientId')
-    instance_rented_id: Optional[int] = Field(None, alias='instanceRentedId')
-    selfhosted_instance_id: Optional[int] = Field(None, alias='selfhostedInstanceId')
-    project_id: Optional[int] = Field(None, alias='projectId')
-    status: Optional[str] = Field(None, alias='status')
-    environment_metadata: Optional[str] = Field(None, alias='environmentMetadata')
-    commit_hash: Optional[str] = Field(None, alias='commitHash')
-    cpu_architecture: Optional[str] = Field(None, alias='cpuArchitecture')
-    gpu_model: Optional[str] = Field(None, alias='gpuModel')
-    ecr_image_url: Optional[str] = Field(None, alias='ecrImageUrl')
-    s3_artifacts_url: Optional[str] = Field(None, alias='s3ArtifactsUrl')
-    created_at: Optional[str] = Field(None, alias='createdAt')
+from typing import Optional
+from pydantic import BaseModel, ConfigDict, Field
+
+class InferenceSimulatorModelDto(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    id: Optional[int] = Field(None, alias='id')
+    slug: Optional[str] = Field(None, alias='slug')
+    client_id: Optional[int] = Field(None, alias='clientId')
+    instance_rented_id: Optional[int] = Field(None, alias='instanceRentedId')
+    selfhosted_instance_id: Optional[int] = Field(None, alias='selfhostedInstanceId')
+    project_id: Optional[int] = Field(None, alias='projectId')
+    status: Optional[str] = Field(None, alias='status')
+    environment_metadata: Optional[str] = Field(None, alias='environmentMetadata')
+    commit_hash: Optional[str] = Field(None, alias='commitHash')
+    cpu_architecture: Optional[str] = Field(None, alias='cpuArchitecture')
+    gpu_model: Optional[str] = Field(None, alias='gpuModel')
+    ecr_image_url: Optional[str] = Field(None, alias='ecrImageUrl')
+    s3_artifacts_url: Optional[str] = Field(None, alias='s3ArtifactsUrl')
+    created_at: Optional[str] = Field(None, alias='createdAt')
     updated_at: Optional[str] = Field(None, alias='updatedAt')

thestage/services/project/dto/project_config.py

@@ -1,14 +1,14 @@
-from typing import Optional
-
-from pydantic import Field, ConfigDict, BaseModel
-
-
-class ProjectConfig(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    id: int = Field(None, alias='id')
-    slug: str = Field(None, alias='slug')
-    git_repository_url: str = Field(None, alias='git_repository_url')
-    deploy_key_path: str = Field(None, alias='deploy_key_path')
-    default_container_uid: Optional[str] = Field(None, alias='default_container_uid')
-    prompt_for_default_container: bool = Field(True, alias='prompt_for_default_container')
+from typing import Optional
+
+from pydantic import Field, ConfigDict, BaseModel
+
+
+class ProjectConfig(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    id: int = Field(None, alias='id')
+    slug: str = Field(None, alias='slug')
+    git_repository_url: str = Field(None, alias='git_repository_url')
+    deploy_key_path: str = Field(None, alias='deploy_key_path')
+    default_container_uid: Optional[str] = Field(None, alias='default_container_uid')
+    prompt_for_default_container: bool = Field(True, alias='prompt_for_default_container')

thestage/services/project/mapper/project_inference_simulator_mapper.py

@@ -1,21 +1,21 @@
-from typing import Optional
-
-from thestage.entities.project_inference_simulator import ProjectInferenceSimulatorEntity
-from thestage.services.abstract_mapper import AbstractMapper
-from thestage.services.project.dto.inference_simulator_dto import InferenceSimulatorDto
-
-
-class ProjectInferenceSimulatorMapper(AbstractMapper):
-
-    def build_entity(self, item: InferenceSimulatorDto) -> Optional[ProjectInferenceSimulatorEntity]:
-        if not item:
-            return None
-
-        return ProjectInferenceSimulatorEntity(
-            slug=item.slug or '',
-            status=item.status or '',
-            http_endpoint=item.http_endpoint or '',
-            grpc_endpoint=item.grpc_endpoint or '',
-            metrics_endpoint=item.metrics_endpoint or '',
-            started_at=item.created_at or '',
-        )
+from typing import Optional
+
+from thestage.entities.project_inference_simulator import ProjectInferenceSimulatorEntity
+from thestage.services.abstract_mapper import AbstractMapper
+from thestage.services.project.dto.inference_simulator_dto import InferenceSimulatorDto
+
+
+class ProjectInferenceSimulatorMapper(AbstractMapper):
+
+    def build_entity(self, item: InferenceSimulatorDto) -> Optional[ProjectInferenceSimulatorEntity]:
+        if not item:
+            return None
+
+        return ProjectInferenceSimulatorEntity(
+            slug=item.slug or '',
+            status=item.status or '',
+            http_endpoint=item.http_endpoint or '',
+            grpc_endpoint=item.grpc_endpoint or '',
+            metrics_endpoint=item.metrics_endpoint or '',
+            started_at=item.created_at or '',
+        )

thestage/services/project/mapper/project_inference_simulator_model_mapper.py

@@ -1,21 +1,21 @@
-from typing import Optional
-
-from thestage.entities.project_inference_simulator_model import ProjectInferenceSimulatorModelEntity
-from thestage.services.abstract_mapper import AbstractMapper
-from thestage.services.project.dto.inference_simulator_model_dto import InferenceSimulatorModelDto
-
-
-class ProjectInferenceSimulatorModelMapper(AbstractMapper):
-    def build_entity(self, item: InferenceSimulatorModelDto) -> Optional[ProjectInferenceSimulatorModelEntity]:
-        if not item:
-            return None
-
-        return ProjectInferenceSimulatorModelEntity(
-            id=item.id,
-            slug=item.slug,
-            status=item.status or '',
-            commit_hash=item.commit_hash or '',
-            environment_metadata=item.environment_metadata or '',
-            started_at=item.created_at or '',
-            finished_at=item.updated_at or ''
-        )
+from typing import Optional
+
+from thestage.entities.project_inference_simulator_model import ProjectInferenceSimulatorModelEntity
+from thestage.services.abstract_mapper import AbstractMapper
+from thestage.services.project.dto.inference_simulator_model_dto import InferenceSimulatorModelDto
+
+
+class ProjectInferenceSimulatorModelMapper(AbstractMapper):
+    def build_entity(self, item: InferenceSimulatorModelDto) -> Optional[ProjectInferenceSimulatorModelEntity]:
+        if not item:
+            return None
+
+        return ProjectInferenceSimulatorModelEntity(
+            id=item.id,
+            slug=item.slug,
+            status=item.status or '',
+            commit_hash=item.commit_hash or '',
+            environment_metadata=item.environment_metadata or '',
+            started_at=item.created_at or '',
+            finished_at=item.updated_at or ''
+        )

thestage/services/project/mapper/project_task_mapper.py

@@ -1,22 +1,22 @@
-from typing import Optional
-
-from thestage.entities.project_task import ProjectTaskEntity
-from thestage.services.abstract_mapper import AbstractMapper
-from thestage.services.task.dto.task_dto import TaskDto
-
-
-class ProjectTaskMapper(AbstractMapper):
-
-    def build_entity(self, item: TaskDto) -> Optional[ProjectTaskEntity]:
-        if not item:
-            return None
-
-        return ProjectTaskEntity(
-            id=item.id or '',
-            title=item.title or '',
-            status=item.frontend_status.status_translation or '',
-            docker_container_id=item.docker_container_id or '',
-            docker_container_slug=(str(item.docker_container.slug) or '') if item.docker_container else '',
-            started_at=item.started_at or '',
-            finished_at=item.finished_at or '',
-        )
+from typing import Optional
+
+from thestage.entities.project_task import ProjectTaskEntity
+from thestage.services.abstract_mapper import AbstractMapper
+from thestage.services.task.dto.task_dto import TaskDto
+
+
+class ProjectTaskMapper(AbstractMapper):
+
+    def build_entity(self, item: TaskDto) -> Optional[ProjectTaskEntity]:
+        if not item:
+            return None
+
+        return ProjectTaskEntity(
+            id=item.id or '',
+            title=item.title or '',
+            status=item.frontend_status.status_translation or '',
+            docker_container_id=item.docker_container_id or '',
+            docker_container_slug=(str(item.docker_container.slug) or '') if item.docker_container else '',
+            started_at=item.started_at or '',
+            finished_at=item.finished_at or '',
+        )