pipekit-sdk 2.0.0__tar.gz → 2.0.1__tar.gz

{pipekit_sdk-2.0.0 → pipekit_sdk-2.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pipekit-sdk
-Version: 2.0.0
+Version: 2.0.1
 Summary: Pipekit Python SDK
 Home-page: https://pipekit.io
 License: BSD-3-Clause

{pipekit_sdk-2.0.0 → pipekit_sdk-2.0.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pipekit-sdk"
-version = "2.0.0"
+version = "2.0.1"
 description = "Pipekit Python SDK"
 authors = ["Pipekit <ci@pipekit.io>"]
 license = "BSD-3-Clause"

{pipekit_sdk-2.0.0 → pipekit_sdk-2.0.1}/src/pipekit_sdk/models/__init__.py

@@ -4,8 +4,8 @@ from __future__ import annotations
 
 from typing import Any
 
-from pipekit_sdk.models._config import BaseModel
+from pydantic import RootModel
 
 
-class Model(BaseModel):
-    __root__: Any
+class Model(RootModel[Any]):
+    root: Any

pipekit_sdk-2.0.1/src/pipekit_sdk/models/_config.py

@@ -0,0 +1,9 @@
+from pydantic import BaseModel as PydanticBaseModel
+from pydantic import ConfigDict
+
+
+class BaseModel(PydanticBaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,  # support populating object by Field alias
+        use_enum_values=True,  # supports using enums, which are then unpacked to obtain the actual `.value`
+    )

{pipekit_sdk-2.0.0 → pipekit_sdk-2.0.1}/src/pipekit_sdk/models/model.py

@@ -2,10 +2,9 @@
 
 from __future__ import annotations
 
-from datetime import datetime
 from typing import Dict, List, Optional
 
-from pydantic import Field
+from pydantic import AwareDatetime, Field
 from typing_extensions import Annotated
 
 from pipekit_sdk.models._config import BaseModel
@@ -13,7 +12,7 @@ from pipekit_sdk.models._config import BaseModel
 
 class Cluster(BaseModel):
     api_key: Annotated[Optional[str], Field(alias="apiKey")] = None
-    created_at: Annotated[datetime, Field(alias="createdAt")]
+    created_at: Annotated[AwareDatetime, Field(alias="createdAt")]
     description: str
     identity_uuid: Annotated[str, Field(alias="identityUUID")]
     is_active: Annotated[bool, Field(alias="isActive")]
@@ -23,7 +22,7 @@ class Cluster(BaseModel):
     pipekit_agent_version: Annotated[str, Field(alias="pipekitAgentVersion")]
     port: Optional[int] = None
     type: List[str]
-    updated_at: Annotated[datetime, Field(alias="updatedAt")]
+    updated_at: Annotated[AwareDatetime, Field(alias="updatedAt")]
     uuid: str
     workflow_group_priorities: Annotated[Optional[Dict[str, int]], Field(alias="workflowGroupPriorities")] = None
 
@@ -43,7 +42,7 @@ class GitInfo(BaseModel):
 
 class Logs(BaseModel):
     container_name: Annotated[str, Field(alias="containerName")]
-    created_at: Annotated[datetime, Field(alias="createdAt")]
+    created_at: Annotated[AwareDatetime, Field(alias="createdAt")]
     identity_uuid: Annotated[str, Field(alias="identityUUID")]
     log_type: Annotated[str, Field(alias="logType")]
     node_id: Annotated[Optional[str], Field(alias="nodeId")] = None
@@ -52,7 +51,7 @@ class Logs(BaseModel):
     pod_name: Annotated[str, Field(alias="podName")]
     run_uuid: Annotated[str, Field(alias="runUUID")]
     stream_cursor: Annotated[str, Field(alias="streamCursor")]
-    updated_at: Annotated[datetime, Field(alias="updatedAt")]
+    updated_at: Annotated[AwareDatetime, Field(alias="updatedAt")]
 
 
 class Pipe(BaseModel):
@@ -77,7 +76,7 @@ class PipeRun(BaseModel):
     check_id: Annotated[Optional[int], Field(alias="checkId")] = None
     cluster_name: Annotated[Optional[str], Field(alias="clusterName")] = None
     cluster_uuid: Annotated[Optional[str], Field(alias="clusterUUID")] = None
-    created_at: Annotated[Optional[datetime], Field(alias="createdAt")] = None
+    created_at: Annotated[Optional[AwareDatetime], Field(alias="createdAt")] = None
     git_info: Annotated[Optional[GitInfo], Field(alias="gitInfo")] = None
     identity_uuid: Annotated[Optional[str], Field(alias="identityUUID")] = None
     message: Optional[str] = None
@@ -92,6 +91,6 @@ class PipeRun(BaseModel):
     source_name: Annotated[Optional[str], Field(alias="sourceName")] = None
     source_type: Annotated[Optional[str], Field(alias="sourceType")] = None
     status: Optional[str] = None
-    updated_at: Annotated[Optional[datetime], Field(alias="updatedAt")] = None
+    updated_at: Annotated[Optional[AwareDatetime], Field(alias="updatedAt")] = None
     user_name: Annotated[Optional[str], Field(alias="userName")] = None
     uuid: Optional[str] = None

{pipekit_sdk-2.0.0 → pipekit_sdk-2.0.1}/src/pipekit_sdk/models/v1.py

@@ -2,10 +2,9 @@
 
 from __future__ import annotations
 
-from datetime import datetime
 from typing import Any, Dict, List, Optional
 
-from pydantic import Field
+from pydantic import AwareDatetime, Field, RootModel
 from typing_extensions import Annotated
 
 from pipekit_sdk.models._config import BaseModel
@@ -276,8 +275,8 @@ class FCVolumeSource(BaseModel):
     ] = None
 
 
-class FieldsV1(BaseModel):
-    __root__: Annotated[
+class FieldsV1(RootModel[Any]):
+    root: Annotated[
         Any,
         Field(
             description=(
@@ -1317,7 +1316,7 @@ class TCPSocketAction(BaseModel):
 
 
 class Time(BaseModel):
-    time: Annotated[datetime, Field(alias="Time")]
+    time: Annotated[AwareDatetime, Field(alias="Time")]
 
 
 class Toleration(BaseModel):

{pipekit_sdk-2.0.0 → pipekit_sdk-2.0.1}/src/pipekit_sdk/models/v1alpha1.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional
 
-from pydantic import Field
+from pydantic import Field, RootModel
 from typing_extensions import Annotated
 
 from pipekit_sdk.models._config import BaseModel
@@ -12,8 +12,8 @@ from pipekit_sdk.models._config import BaseModel
 from . import intstr, v1
 
 
-class Amount(BaseModel):
-    __root__: Any
+class Amount(RootModel[Any]):
+    root: Any
 
 
 class ArtGCStatus(BaseModel):
@@ -190,8 +190,8 @@ class Histogram(BaseModel):
     value: str
 
 
-class Item(BaseModel):
-    __root__: Any
+class Item(RootModel[Any]):
+    root: Any
 
 
 class LabelValueFrom(BaseModel):
@@ -244,8 +244,8 @@ class NodeSynchronizationStatus(BaseModel):
     waiting: Optional[str] = None
 
 
-class NoneStrategy(BaseModel):
-    __root__: Any
+class NoneStrategy(RootModel[Any]):
+    root: Any
 
 
 class OAuth2EndpointParam(BaseModel):
@@ -258,12 +258,12 @@ class OSSLifecycleRule(BaseModel):
     mark_infrequent_access_after_days: Annotated[Optional[int], Field(alias="markInfrequentAccessAfterDays")] = None
 
 
-class ParallelSteps(BaseModel):
-    __root__: Any
+class ParallelSteps(RootModel[Any]):
+    root: Any
 
 
-class Plugin(BaseModel):
-    __root__: Any
+class Plugin(RootModel[Any]):
+    root: Any
 
 
 class Prometheus(BaseModel):
@@ -280,8 +280,8 @@ class RawArtifact(BaseModel):
     data: str
 
 
-class RetryNodeAntiAffinity(BaseModel):
-    __root__: Any
+class RetryNodeAntiAffinity(RootModel[Any]):
+    root: Any
 
 
 class S3EncryptionOptions(BaseModel):
@@ -315,8 +315,8 @@ class Sequence(BaseModel):
     start: Optional[intstr.IntOrString] = None
 
 
-class SuppliedValueFrom(BaseModel):
-    __root__: Any
+class SuppliedValueFrom(RootModel[Any]):
+    root: Any
 
 
 class SuspendTemplate(BaseModel):
@@ -388,8 +388,8 @@ class WorkflowTemplateRef(BaseModel):
     name: Optional[str] = None
 
 
-class ZipStrategy(BaseModel):
-    __root__: Any
+class ZipStrategy(RootModel[Any]):
+    root: Any
 
 
 class ArchiveStrategy(BaseModel):
@@ -1689,7 +1689,7 @@ class WorkflowStatus(BaseModel):
     ] = None
 
 
-CronWorkflow.update_forward_refs()
-CronWorkflowSpec.update_forward_refs()
-DAGTask.update_forward_refs()
-Workflow.update_forward_refs()
+CronWorkflow.model_rebuild()
+CronWorkflowSpec.model_rebuild()
+DAGTask.model_rebuild()
+Workflow.model_rebuild()

{pipekit_sdk-2.0.0 → pipekit_sdk-2.0.1}/src/pipekit_sdk/service.py

@@ -80,7 +80,7 @@ class PipekitService:
         if clusters_response.status_code // 100 != 2:
             raise Exception(f"List clusters failed: {clusters_response.text}")
 
-        clusters = [Cluster.parse_obj(cluster) for cluster in clusters_response.json()]
+        clusters = [Cluster.model_validate(cluster) for cluster in clusters_response.json()]
         return clusters
 
     def list_pipes(
@@ -107,7 +107,7 @@ class PipekitService:
             if pipes_response.status_code // 100 != 2:
                 raise Exception(f"List pipes failed: {pipes_response.text}")
 
-            pipes_page = [Pipe.parse_obj(item) for item in pipes_response.json()["items"]]
+            pipes_page = [Pipe.model_validate(item) for item in pipes_response.json()["items"]]
             if len(pipes_page) == 0:
                 break
 
@@ -135,7 +135,7 @@ class PipekitService:
         if pipe_response.status_code // 100 != 2:
             raise Exception(f"Get pipe failed: {pipe_response.text}")
 
-        return Pipe.parse_obj(pipe_response.json())
+        return Pipe.model_validate(pipe_response.json())
 
     def list_runs_by_cluster(
         self,
@@ -160,7 +160,7 @@ class PipekitService:
             if runs_response.status_code // 100 != 2:
                 raise Exception(f"List runs failed: {runs_response.text}")
 
-            runs_page = [PipeRun.parse_obj(item) for item in runs_response.json()["items"]]
+            runs_page = [PipeRun.model_validate(item) for item in runs_response.json()["items"]]
             if len(runs_page) == 0:
                 break
 
@@ -196,7 +196,7 @@ class PipekitService:
             if runs_response.status_code // 100 != 2:
                 raise Exception(f"List runs failed: {runs_response.text}")
 
-            runs_page = [PipeRun.parse_obj(item) for item in runs_response.json()["items"]]
+            runs_page = [PipeRun.model_validate(item) for item in runs_response.json()["items"]]
             if len(runs_page) == 0:
                 break
 
@@ -224,7 +224,7 @@ class PipekitService:
         if run_response.status_code // 100 != 2:
             raise Exception(f"Get run failed: {run_response.text}")
 
-        return PipeRun.parse_obj(run_response.json())
+        return PipeRun.model_validate(run_response.json())
 
     def get_cluster(
         self,
@@ -241,7 +241,7 @@ class PipekitService:
         if cluster_response.status_code // 100 != 2:
             raise Exception(f"Get cluster failed: {cluster_response.text}")
 
-        return Cluster.parse_obj(cluster_response.json())
+        return Cluster.model_validate(cluster_response.json())
 
     def get_cluster_by_name(
         self,
@@ -293,7 +293,7 @@ class PipekitService:
         if submit_response.status_code // 100 != 2:
             raise Exception(f"Submit failed: {submit_response.text}")
 
-        return PipeRun.parse_obj(submit_response.json())
+        return PipeRun.model_validate(submit_response.json())
 
     def create(self, cron_workflow: CronWorkflow, cluster: str, pipe_name: str = "") -> PipeRun:
         """Create a CronWorkflow on the cluster, under an existing or new pipe with the given name."""
@@ -328,7 +328,7 @@ class PipekitService:
         if create_response.status_code // 100 != 2:
             raise Exception(f"Create failed: {create_response.text}")
 
-        return PipeRun.parse_obj(create_response.json())
+        return PipeRun.model_validate(create_response.json())
 
     def __apply_run_action(
         self,
@@ -348,7 +348,7 @@ class PipekitService:
         if action_response.status_code // 100 != 2:
             raise Exception(f"Action failed: {action_response.text}")
 
-        return PipeRun.parse_obj(action_response.json())
+        return PipeRun.model_validate(action_response.json())
 
     def restart(
         self,
@@ -388,7 +388,7 @@ class PipekitService:
         if logs_response.status_code // 100 != 2:
             raise Exception(f"Get logs failed: {logs_response.text}")
 
-        return [Logs.parse_obj(log) for log in list(logs_response.json())]
+        return [Logs.model_validate(log) for log in list(logs_response.json())]
 
     def __get_logs_stream(
         self,
@@ -464,7 +464,7 @@ class PipekitService:
                         # bye
                         return
                     else:
-                        yield Logs.parse_obj(json.loads(line))
+                        yield Logs.model_validate(json.loads(line))
 
             except requests.exceptions.Timeout:
                 pass  # we'll ignore timeout errors and reconnect

pipekit_sdk-2.0.0/src/pipekit_sdk/models/_config.py

@@ -1,15 +0,0 @@
-from pydantic.v1 import BaseModel as PydanticBaseModel
-
-
-class BaseModel(PydanticBaseModel):
-    class Config:
-        """General config for generated models."""
-
-        allow_population_by_field_name = True
-        """support populating object by Field alias"""
-
-        allow_mutation = True
-        """allow mutation of objects post instantiation"""
-
-        use_enum_values = True
-        """supports using enums, which are then unpacked to obtain the actual `.value`"""