prefect-client 3.3.3.dev1__tar.gz → 3.3.4.dev1__tar.gz

{prefect_client-3.3.3.dev1 → prefect_client-3.3.4.dev1}/.gitignore

@@ -39,6 +39,7 @@ ipython_config.py
 # Environments
 .python-version
 .env
+.envrc
 .venv
 env/
 venv/
@@ -87,4 +88,4 @@ src/integrations/*/**/_version.py
 *.log
 
 # Pyright type analysis report
-prefect-analysis.json
+prefect-analysis.json

{prefect_client-3.3.3.dev1 → prefect_client-3.3.4.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prefect-client
-Version: 3.3.3.dev1
+Version: 3.3.4.dev1
 Summary: Workflow orchestration and management.
 Project-URL: Changelog, https://github.com/PrefectHQ/prefect/releases
 Project-URL: Documentation, https://docs.prefect.io

prefect_client-3.3.4.dev1/src/prefect/_build_info.py

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.3.4.dev1"
+__build_date__ = "2025-04-08 08:08:41.971107+00:00"
+__git_commit__ = "e7fa0e30c362e23a55ba2f2362b850e34954236f"
+__dirty__ = False

{prefect_client-3.3.3.dev1 → prefect_client-3.3.4.dev1}/src/prefect/client/orchestration/_deployments/client.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 from collections.abc import Iterable
 from typing import TYPE_CHECKING, Any, Union
+from uuid import UUID
 
 from httpx import HTTPStatusError, RequestError
 
@@ -10,7 +11,6 @@ from prefect.exceptions import ObjectNotFound
 
 if TYPE_CHECKING:
     import datetime
-    from uuid import UUID
 
     from prefect.client.schemas import FlowRun
     from prefect.client.schemas.actions import (
@@ -28,6 +28,7 @@ if TYPE_CHECKING:
     )
     from prefect.client.schemas.objects import (
         ConcurrencyOptions,
+        DeploymentBranchingOptions,
         DeploymentSchedule,
         VersionInfo,
     )
@@ -46,7 +47,7 @@ if TYPE_CHECKING:
 class DeploymentClient(BaseClient):
     def create_deployment(
         self,
-        flow_id: "UUID",
+        flow_id: UUID,
         name: str,
         version: str | None = None,
         version_info: "VersionInfo | None" = None,
@@ -58,10 +59,10 @@ class DeploymentClient(BaseClient):
         work_queue_name: str | None = None,
         work_pool_name: str | None = None,
         tags: list[str] | None = None,
-        storage_document_id: "UUID | None" = None,
+        storage_document_id: UUID | None = None,
         path: str | None = None,
         entrypoint: str | None = None,
-        infrastructure_document_id: "UUID | None" = None,
+        infrastructure_document_id: UUID | None = None,
         parameter_openapi_schema: dict[str, Any] | None = None,
         paused: bool | None = None,
         pull_steps: list[dict[str, Any]] | None = None,
@@ -70,7 +71,7 @@ class DeploymentClient(BaseClient):
         branch: str | None = None,
         base: UUID | None = None,
         root: UUID | None = None,
-    ) -> "UUID":
+    ) -> UUID:
         """
         Create a deployment.
 
@@ -94,7 +95,6 @@ class DeploymentClient(BaseClient):
         Returns:
             the ID of the deployment in the backend
         """
-        from uuid import UUID
 
         from prefect.client.schemas.actions import DeploymentCreate
 
@@ -166,7 +166,7 @@ class DeploymentClient(BaseClient):
 
         return UUID(deployment_id)
 
-    def set_deployment_paused_state(self, deployment_id: "UUID", paused: bool) -> None:
+    def set_deployment_paused_state(self, deployment_id: UUID, paused: bool) -> None:
         self.request(
             "PATCH",
             "/deployments/{id}",
@@ -176,7 +176,7 @@ class DeploymentClient(BaseClient):
 
     def update_deployment(
         self,
-        deployment_id: "UUID",
+        deployment_id: UUID,
         deployment: "DeploymentUpdate",
     ) -> None:
         self.request(
@@ -190,12 +190,10 @@ class DeploymentClient(BaseClient):
             ),
         )
 
-    def _create_deployment_from_schema(self, schema: "DeploymentCreate") -> "UUID":
+    def _create_deployment_from_schema(self, schema: "DeploymentCreate") -> UUID:
         """
         Create a deployment from a prepared `DeploymentCreate` schema.
         """
-        from uuid import UUID
-
         # TODO: We are likely to remove this method once we have considered the
         #       packaging interface for deployments further.
         response = self.request(
@@ -209,7 +207,7 @@ class DeploymentClient(BaseClient):
 
     def read_deployment(
         self,
-        deployment_id: Union["UUID", str],
+        deployment_id: Union[UUID, str],
     ) -> "DeploymentResponse":
         """
         Query the Prefect API for a deployment by id.
@@ -220,7 +218,6 @@ class DeploymentClient(BaseClient):
         Returns:
             a [Deployment model][prefect.client.schemas.objects.Deployment] representation of the deployment
         """
-        from uuid import UUID
 
         from prefect.client.schemas.responses import DeploymentResponse
 
@@ -346,7 +343,7 @@ class DeploymentClient(BaseClient):
 
     def delete_deployment(
         self,
-        deployment_id: "UUID",
+        deployment_id: UUID,
     ) -> None:
         """
         Delete deployment by id.
@@ -371,7 +368,7 @@ class DeploymentClient(BaseClient):
 
     def create_deployment_schedules(
         self,
-        deployment_id: "UUID",
+        deployment_id: UUID,
         schedules: list[tuple["SCHEDULE_TYPES", bool]],
     ) -> list["DeploymentSchedule"]:
         """
@@ -410,7 +407,7 @@ class DeploymentClient(BaseClient):
 
     def read_deployment_schedules(
         self,
-        deployment_id: "UUID",
+        deployment_id: UUID,
     ) -> list["DeploymentSchedule"]:
         """
         Query the Prefect API for a deployment's schedules.
@@ -438,8 +435,8 @@ class DeploymentClient(BaseClient):
 
     def update_deployment_schedule(
         self,
-        deployment_id: "UUID",
-        schedule_id: "UUID",
+        deployment_id: UUID,
+        schedule_id: UUID,
         active: bool | None = None,
         schedule: "SCHEDULE_TYPES | None" = None,
     ) -> None:
@@ -478,8 +475,8 @@ class DeploymentClient(BaseClient):
 
     def delete_deployment_schedule(
         self,
-        deployment_id: "UUID",
-        schedule_id: "UUID",
+        deployment_id: UUID,
+        schedule_id: UUID,
     ) -> None:
         """
         Delete a deployment schedule.
@@ -505,7 +502,7 @@ class DeploymentClient(BaseClient):
 
     def get_scheduled_flow_runs_for_deployments(
         self,
-        deployment_ids: list["UUID"],
+        deployment_ids: list[UUID],
         scheduled_before: "datetime.datetime | None" = None,
         limit: int | None = None,
     ) -> list["FlowRunResponse"]:
@@ -527,7 +524,7 @@ class DeploymentClient(BaseClient):
 
     def create_flow_run_from_deployment(
         self,
-        deployment_id: "UUID",
+        deployment_id: UUID,
         *,
         parameters: dict[str, Any] | None = None,
         context: dict[str, Any] | None = None,
@@ -535,7 +532,7 @@ class DeploymentClient(BaseClient):
         name: str | None = None,
         tags: Iterable[str] | None = None,
         idempotency_key: str | None = None,
-        parent_task_run_id: "UUID | None" = None,
+        parent_task_run_id: UUID | None = None,
         work_queue_name: str | None = None,
         job_variables: dict[str, Any] | None = None,
         labels: "KeyValueLabelsField | None" = None,
@@ -604,11 +601,33 @@ class DeploymentClient(BaseClient):
         )
         return FlowRun.model_validate(response.json())
 
+    def create_deployment_branch(
+        self,
+        deployment_id: UUID,
+        branch: str,
+        options: "DeploymentBranchingOptions | None" = None,
+        overrides: "DeploymentUpdate | None" = None,
+    ) -> UUID:
+        from prefect.client.schemas.actions import DeploymentBranch
+        from prefect.client.schemas.objects import DeploymentBranchingOptions
+
+        response = self.request(
+            "POST",
+            "/deployments/{id}/branch",
+            path_params={"id": deployment_id},
+            json=DeploymentBranch(
+                branch=branch,
+                options=options or DeploymentBranchingOptions(),
+                overrides=overrides,
+            ).model_dump(mode="json", exclude_unset=True),
+        )
+        return UUID(response.json().get("id"))
+
 
 class DeploymentAsyncClient(BaseAsyncClient):
     async def create_deployment(
         self,
-        flow_id: "UUID",
+        flow_id: UUID,
         name: str,
         version: str | None = None,
         version_info: "VersionInfo | None" = None,
@@ -620,10 +639,10 @@ class DeploymentAsyncClient(BaseAsyncClient):
         work_queue_name: str | None = None,
         work_pool_name: str | None = None,
         tags: list[str] | None = None,
-        storage_document_id: "UUID | None" = None,
+        storage_document_id: UUID | None = None,
         path: str | None = None,
         entrypoint: str | None = None,
-        infrastructure_document_id: "UUID | None" = None,
+        infrastructure_document_id: UUID | None = None,
         parameter_openapi_schema: dict[str, Any] | None = None,
         paused: bool | None = None,
         pull_steps: list[dict[str, Any]] | None = None,
@@ -632,7 +651,7 @@ class DeploymentAsyncClient(BaseAsyncClient):
         branch: str | None = None,
         base: UUID | None = None,
         root: UUID | None = None,
-    ) -> "UUID":
+    ) -> UUID:
         """
         Create a deployment.
 
@@ -656,7 +675,6 @@ class DeploymentAsyncClient(BaseAsyncClient):
         Returns:
             the ID of the deployment in the backend
         """
-        from uuid import UUID
 
         from prefect.client.schemas.actions import DeploymentCreate
 
@@ -729,7 +747,7 @@ class DeploymentAsyncClient(BaseAsyncClient):
         return UUID(deployment_id)
 
     async def set_deployment_paused_state(
-        self, deployment_id: "UUID", paused: bool
+        self, deployment_id: UUID, paused: bool
     ) -> None:
         await self.request(
             "PATCH",
@@ -740,7 +758,7 @@ class DeploymentAsyncClient(BaseAsyncClient):
 
     async def update_deployment(
         self,
-        deployment_id: "UUID",
+        deployment_id: UUID,
         deployment: "DeploymentUpdate",
     ) -> None:
         await self.request(
@@ -754,13 +772,10 @@ class DeploymentAsyncClient(BaseAsyncClient):
             ),
         )
 
-    async def _create_deployment_from_schema(
-        self, schema: "DeploymentCreate"
-    ) -> "UUID":
+    async def _create_deployment_from_schema(self, schema: "DeploymentCreate") -> UUID:
         """
         Create a deployment from a prepared `DeploymentCreate` schema.
         """
-        from uuid import UUID
 
         # TODO: We are likely to remove this method once we have considered the
         #       packaging interface for deployments further.
@@ -775,7 +790,7 @@ class DeploymentAsyncClient(BaseAsyncClient):
 
     async def read_deployment(
         self,
-        deployment_id: Union["UUID", str],
+        deployment_id: Union[UUID, str],
     ) -> "DeploymentResponse":
         """
         Query the Prefect API for a deployment by id.
@@ -786,7 +801,6 @@ class DeploymentAsyncClient(BaseAsyncClient):
         Returns:
             a [Deployment model][prefect.client.schemas.objects.Deployment] representation of the deployment
         """
-        from uuid import UUID
 
         from prefect.client.schemas.responses import DeploymentResponse
 
@@ -912,7 +926,7 @@ class DeploymentAsyncClient(BaseAsyncClient):
 
     async def delete_deployment(
         self,
-        deployment_id: "UUID",
+        deployment_id: UUID,
     ) -> None:
         """
         Delete deployment by id.
@@ -937,7 +951,7 @@ class DeploymentAsyncClient(BaseAsyncClient):
 
     async def create_deployment_schedules(
         self,
-        deployment_id: "UUID",
+        deployment_id: UUID,
         schedules: list[tuple["SCHEDULE_TYPES", bool]],
     ) -> list["DeploymentSchedule"]:
         """
@@ -976,7 +990,7 @@ class DeploymentAsyncClient(BaseAsyncClient):
 
     async def read_deployment_schedules(
         self,
-        deployment_id: "UUID",
+        deployment_id: UUID,
     ) -> list["DeploymentSchedule"]:
         """
         Query the Prefect API for a deployment's schedules.
@@ -1004,8 +1018,8 @@ class DeploymentAsyncClient(BaseAsyncClient):
 
     async def update_deployment_schedule(
         self,
-        deployment_id: "UUID",
-        schedule_id: "UUID",
+        deployment_id: UUID,
+        schedule_id: UUID,
         active: bool | None = None,
         schedule: "SCHEDULE_TYPES | None" = None,
     ) -> None:
@@ -1044,8 +1058,8 @@ class DeploymentAsyncClient(BaseAsyncClient):
 
     async def delete_deployment_schedule(
         self,
-        deployment_id: "UUID",
-        schedule_id: "UUID",
+        deployment_id: UUID,
+        schedule_id: UUID,
     ) -> None:
         """
         Delete a deployment schedule.
@@ -1071,7 +1085,7 @@ class DeploymentAsyncClient(BaseAsyncClient):
 
     async def get_scheduled_flow_runs_for_deployments(
         self,
-        deployment_ids: list["UUID"],
+        deployment_ids: list[UUID],
         scheduled_before: "datetime.datetime | None" = None,
         limit: int | None = None,
     ) -> list["FlowRun"]:
@@ -1093,7 +1107,7 @@ class DeploymentAsyncClient(BaseAsyncClient):
 
     async def create_flow_run_from_deployment(
         self,
-        deployment_id: "UUID",
+        deployment_id: UUID,
         *,
         parameters: dict[str, Any] | None = None,
         context: dict[str, Any] | None = None,
@@ -1101,7 +1115,7 @@ class DeploymentAsyncClient(BaseAsyncClient):
         name: str | None = None,
         tags: Iterable[str] | None = None,
         idempotency_key: str | None = None,
-        parent_task_run_id: "UUID | None" = None,
+        parent_task_run_id: UUID | None = None,
         work_queue_name: str | None = None,
         job_variables: dict[str, Any] | None = None,
         labels: "KeyValueLabelsField | None" = None,
@@ -1169,3 +1183,25 @@ class DeploymentAsyncClient(BaseAsyncClient):
             json=flow_run_create.model_dump(mode="json", exclude_unset=True),
         )
         return FlowRun.model_validate(response.json())
+
+    async def create_deployment_branch(
+        self,
+        deployment_id: UUID,
+        branch: str,
+        options: "DeploymentBranchingOptions | None" = None,
+        overrides: "DeploymentUpdate | None" = None,
+    ) -> UUID:
+        from prefect.client.schemas.actions import DeploymentBranch
+        from prefect.client.schemas.objects import DeploymentBranchingOptions
+
+        response = await self.request(
+            "POST",
+            "/deployments/{id}/branch",
+            path_params={"id": deployment_id},
+            json=DeploymentBranch(
+                branch=branch,
+                options=options or DeploymentBranchingOptions(),
+                overrides=overrides,
+            ).model_dump(mode="json", exclude_unset=True),
+        )
+        return UUID(response.json().get("id"))

{prefect_client-3.3.3.dev1 → prefect_client-3.3.4.dev1}/src/prefect/client/orchestration/routes.py

@@ -46,6 +46,7 @@ ServerRoutes = Literal[
     "/csrf-token",
     "/deployments/",
     "/deployments/{id}",
+    "/deployments/{id}/branch",
     "/deployments/{id}/create_flow_run",
     "/deployments/{id}/pause_deployment",
     "/deployments/{id}/resume_deployment",

{prefect_client-3.3.3.dev1 → prefect_client-3.3.4.dev1}/src/prefect/client/schemas/actions.py

@@ -373,6 +373,25 @@ class DeploymentUpdate(ActionBaseModel):
             jsonschema.validate(self.job_variables, variables_schema)
 
 
+class DeploymentBranch(ActionBaseModel):
+    branch: str = Field(..., description="Name of the branch to create")
+    options: objects.DeploymentBranchingOptions = Field(
+        default_factory=objects.DeploymentBranchingOptions,
+        description="Configuration options for how the deployment should be branched",
+    )
+    overrides: Optional[DeploymentUpdate] = Field(
+        default=None,
+        description="Optional values to override in the branched deployment",
+    )
+
+    @field_validator("branch")
+    @classmethod
+    def validate_branch_length(cls, v: str) -> str:
+        if len(v.strip()) < 1:
+            raise ValueError("Branch name cannot be empty or contain only whitespace")
+        return v
+
+
 class FlowRunUpdate(ActionBaseModel):
     """Data used by the Prefect REST API to update a flow run."""
 

{prefect_client-3.3.3.dev1 → prefect_client-3.3.4.dev1}/src/prefect/client/schemas/objects.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 import datetime
 import warnings
 from collections.abc import Callable, Mapping
+from enum import Enum
 from functools import partial
 from typing import (
     TYPE_CHECKING,
@@ -1107,6 +1108,19 @@ class VersionInfo(PrefectBaseModel, extra="allow"):
     version: str = Field(default=..., description="The version of the deployment.")
 
 
+class BranchingScheduleHandling(str, Enum):
+    KEEP = "keep"
+    REMOVE = "remove"
+    INACTIVE = "inactive"
+
+
+class DeploymentBranchingOptions(ObjectBaseModel):
+    schedule_handling: BranchingScheduleHandling = Field(
+        default=BranchingScheduleHandling.REMOVE,
+        description="Whether to keep, remove, or set inactive the existing schedules when branching",
+    )
+
+
 class Deployment(ObjectBaseModel):
     """An ORM representation of deployment data."""
 

{prefect_client-3.3.3.dev1 → prefect_client-3.3.4.dev1}/src/prefect/client/schemas/responses.py

@@ -308,6 +308,8 @@ class FlowRunResponse(ObjectBaseModel):
 
 class DeploymentResponse(ObjectBaseModel):
     name: str = Field(default=..., description="The name of the deployment.")
+
+    # Versionining
     version: Optional[str] = Field(
         default=None, description="An optional version for the deployment."
     )
@@ -317,6 +319,18 @@ class DeploymentResponse(ObjectBaseModel):
     version_info: Optional[objects.VersionInfo] = Field(
         default=None, description="A description of this version of the deployment."
     )
+
+    # Branching
+    branch: Optional[str] = Field(
+        default=None, description="The branch of the deployment."
+    )
+    base: Optional[UUID] = Field(
+        default=None, description="The base deployment of the deployment."
+    )
+    root: Optional[UUID] = Field(
+        default=None, description="The root deployment of the deployment."
+    )
+
     description: Optional[str] = Field(
         default=None, description="A description for the deployment."
     )
@@ -438,6 +452,15 @@ class DeploymentResponse(ObjectBaseModel):
             "prefect.resource.name": self.name,
         }
 
+        if self.branch:
+            labels["prefect.deployment.branch"] = self.branch
+
+        if self.base:
+            labels["prefect.deployment.base"] = f"prefect.deployment.{self.base}"
+
+        if self.root:
+            labels["prefect.deployment.root"] = f"prefect.deployment.{self.root}"
+
         if self.version_id and self.version_info:
             labels["prefect.deployment.version-id"] = str(self.version_id)
             labels["prefect.deployment.version-type"] = self.version_info.type

{prefect_client-3.3.3.dev1 → prefect_client-3.3.4.dev1}/src/prefect/server/api/flow_runs.py

@@ -48,7 +48,7 @@ from prefect.server.schemas.responses import (
 )
 from prefect.server.utilities.server import PrefectRouter
 from prefect.types import DateTime
-from prefect.types._datetime import now
+from prefect.types._datetime import earliest_possible_datetime, now
 from prefect.utilities import schema_tools
 
 if TYPE_CHECKING:
@@ -353,7 +353,7 @@ async def read_flow_run_graph_v1(
 async def read_flow_run_graph_v2(
     flow_run_id: UUID = Path(..., description="The flow run id", alias="id"),
     since: datetime.datetime = Query(
-        default=jsonable_encoder(datetime.datetime.min),
+        default=jsonable_encoder(earliest_possible_datetime()),
         description="Only include runs that start or end after this time.",
     ),
     db: PrefectDBInterface = Depends(provide_database_interface),

{prefect_client-3.3.3.dev1 → prefect_client-3.3.4.dev1}/src/prefect/settings/models/server/events.py

@@ -50,7 +50,7 @@ class ServerEventsSettings(PrefectBaseSettings):
     )
 
     maximum_related_resources: int = Field(
-        default=500,
+        default=100,
         description="The maximum number of related resources an Event may have.",
         validation_alias=AliasChoices(
             AliasPath("maximum_related_resources"),

prefect_client-3.3.3.dev1/src/prefect/_build_info.py

@@ -1,5 +0,0 @@
-# Generated by versioningit
-__version__ = "3.3.3.dev1"
-__build_date__ = "2025-04-04 08:08:08.629815+00:00"
-__git_commit__ = "cf8d1ca6f0eb2e922c48162b40454fd5b1cfb192"
-__dirty__ = False