thestage 0.6.3__tar.gz → 0.6.4__tar.gz

{thestage-0.6.3 → thestage-0.6.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: thestage
-Version: 0.6.3
+Version: 0.6.4
 Summary: 
 Author: TheStage AI team
 Author-email: hello@thestage.ai

{thestage-0.6.3 → thestage-0.6.4}/pyproject.toml

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

{thestage-0.6.3 → thestage-0.6.4}/thestage/entities/project_inference_simulator_model.py

@@ -1,4 +1,4 @@
-from typing import Optional
+from typing import Optional, Any, Dict
 
 from pydantic import BaseModel, ConfigDict, Field
 
@@ -14,4 +14,4 @@ class ProjectInferenceSimulatorModelEntity(BaseModel):
     slug: Optional[str] = Field(None, alias='UNIQUE_ID')
     status: Optional[str] = Field(None, alias='STATUS')
     commit_hash: Optional[str] = Field(None, alias='COMMIT_HASH')
-    environment_metadata: Optional[str] = Field(None, alias='ENVIRONMENT_METADATA')
+    environment_metadata: Optional[Dict[str, Any]] = Field(None, alias='ENVIRONMENT_METADATA')

{thestage-0.6.3 → thestage-0.6.4}/thestage/services/project/dto/inference_simulator_dto.py

@@ -1,4 +1,4 @@
-from typing import Optional
+from typing import Optional, Any, Dict
 from pydantic import BaseModel, ConfigDict, Field
 
 class InferenceSimulatorDto(BaseModel):
@@ -17,7 +17,7 @@ class InferenceSimulatorDto(BaseModel):
     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")
+    inference_metadata: Optional[Dict[str, Any]] = Field(None, alias="inferenceMetadata")
+    qlip_serve_metadata: Optional[Dict[str, Any]] = Field(None, alias="qlipServeMetadata")
     created_at: Optional[str] = Field(None, alias="createdAt")
     updated_at: Optional[str] = Field(None, alias="updatedAt")

{thestage-0.6.3 → thestage-0.6.4}/thestage/services/project/dto/inference_simulator_model_dto.py

@@ -1,4 +1,4 @@
-from typing import Optional
+from typing import Optional, Any, Dict
 from pydantic import BaseModel, ConfigDict, Field
 
 class InferenceSimulatorModelDto(BaseModel):
@@ -11,7 +11,7 @@ class InferenceSimulatorModelDto(BaseModel):
     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')
+    environment_metadata: Optional[Dict[str, Any]] = 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')