prefect-client 3.3.2.dev2__py3-none-any.whl → 3.3.3__py3-none-any.whl

prefect/_build_info.py

@@ -1,5 +1,5 @@
 # Generated by versioningit
-__version__ = "3.3.2.dev2"
-__build_date__ = "2025-04-03 08:08:10.651077+00:00"
-__git_commit__ = "6d3d12cfe81fa08c1b14f99867e2e621816d1a89"
+__version__ = "3.3.3"
+__build_date__ = "2025-04-05 01:43:07.772891+00:00"
+__git_commit__ = "4100d4ea3c8d2559e48d9860a5b241d8cd29d47e"
 __dirty__ = False

prefect/client/orchestration/__init__.py

@@ -1,3 +1,4 @@
+from __future__ import annotations
 import asyncio
 import base64
 import datetime
@@ -1164,6 +1165,10 @@ class PrefectClient(
     def client_version(self) -> str:
         return prefect.__version__
 
+    @property
+    def loop(self) -> asyncio.AbstractEventLoop | None:
+        return self._loop
+
     async def raise_for_api_version_mismatch(self) -> None:
         # Cloud is always compatible as a server
         if self.server_type == ServerType.CLOUD:

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,7 +28,9 @@ if TYPE_CHECKING:
     )
     from prefect.client.schemas.objects import (
         ConcurrencyOptions,
+        DeploymentBranchingOptions,
         DeploymentSchedule,
+        VersionInfo,
     )
     from prefect.client.schemas.responses import (
         DeploymentResponse,
@@ -45,9 +47,10 @@ 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,
         schedules: list["DeploymentScheduleCreate"] | None = None,
         concurrency_limit: int | None = None,
         concurrency_options: "ConcurrencyOptions | None" = None,
@@ -56,16 +59,19 @@ 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,
         enforce_parameter_schema: bool | None = None,
         job_variables: dict[str, Any] | None = None,
-    ) -> "UUID":
+        branch: str | None = None,
+        base: UUID | None = None,
+        root: UUID | None = None,
+    ) -> UUID:
         """
         Create a deployment.
 
@@ -89,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
 
@@ -99,6 +104,7 @@ class DeploymentClient(BaseClient):
             flow_id=flow_id,
             name=name,
             version=version,
+            version_info=version_info,
             parameters=dict(parameters or {}),
             tags=list(tags or []),
             work_queue_name=work_queue_name,
@@ -115,6 +121,9 @@ class DeploymentClient(BaseClient):
             concurrency_options=concurrency_options,
             pull_steps=pull_steps,
             enforce_parameter_schema=enforce_parameter_schema,
+            branch=branch,
+            base=base,
+            root=root,
         )
 
         if work_pool_name is not None:
@@ -123,20 +132,29 @@ class DeploymentClient(BaseClient):
         # Exclude newer fields that are not set to avoid compatibility issues
         exclude = {
             field
-            for field in ["work_pool_name", "work_queue_name"]
+            for field in [
+                "work_pool_name",
+                "work_queue_name",
+            ]
             if field not in deployment_create.model_fields_set
         }
 
-        if deployment_create.paused is None:
-            exclude.add("paused")
-
-        if deployment_create.pull_steps is None:
-            exclude.add("pull_steps")
+        exclude_if_none = [
+            "paused",
+            "pull_steps",
+            "enforce_parameter_schema",
+            "version_info",
+            "branch",
+            "base",
+            "root",
+        ]
 
-        if deployment_create.enforce_parameter_schema is None:
-            exclude.add("enforce_parameter_schema")
+        for field in exclude_if_none:
+            if getattr(deployment_create, field) is None:
+                exclude.add(field)
 
         json = deployment_create.model_dump(mode="json", exclude=exclude)
+
         response = self.request(
             "POST",
             "/deployments/",
@@ -148,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}",
@@ -158,7 +176,7 @@ class DeploymentClient(BaseClient):
 
     def update_deployment(
         self,
-        deployment_id: "UUID",
+        deployment_id: UUID,
         deployment: "DeploymentUpdate",
     ) -> None:
         self.request(
@@ -172,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(
@@ -191,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.
@@ -202,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
 
@@ -328,7 +343,7 @@ class DeploymentClient(BaseClient):
 
     def delete_deployment(
         self,
-        deployment_id: "UUID",
+        deployment_id: UUID,
     ) -> None:
         """
         Delete deployment by id.
@@ -353,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"]:
         """
@@ -392,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.
@@ -420,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:
@@ -460,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.
@@ -487,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"]:
@@ -509,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,
@@ -517,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,
@@ -586,13 +601,36 @@ 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,
         schedules: list["DeploymentScheduleCreate"] | None = None,
         concurrency_limit: int | None = None,
         concurrency_options: "ConcurrencyOptions | None" = None,
@@ -601,16 +639,19 @@ 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,
         enforce_parameter_schema: bool | None = None,
         job_variables: dict[str, Any] | None = None,
-    ) -> "UUID":
+        branch: str | None = None,
+        base: UUID | None = None,
+        root: UUID | None = None,
+    ) -> UUID:
         """
         Create a deployment.
 
@@ -634,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
 
@@ -644,6 +684,7 @@ class DeploymentAsyncClient(BaseAsyncClient):
             flow_id=flow_id,
             name=name,
             version=version,
+            version_info=version_info,
             parameters=dict(parameters or {}),
             tags=list(tags or []),
             work_queue_name=work_queue_name,
@@ -660,6 +701,9 @@ class DeploymentAsyncClient(BaseAsyncClient):
             concurrency_options=concurrency_options,
             pull_steps=pull_steps,
             enforce_parameter_schema=enforce_parameter_schema,
+            branch=branch,
+            base=base,
+            root=root,
         )
 
         if work_pool_name is not None:
@@ -668,20 +712,29 @@ class DeploymentAsyncClient(BaseAsyncClient):
         # Exclude newer fields that are not set to avoid compatibility issues
         exclude = {
             field
-            for field in ["work_pool_name", "work_queue_name"]
+            for field in [
+                "work_pool_name",
+                "work_queue_name",
+            ]
             if field not in deployment_create.model_fields_set
         }
 
-        if deployment_create.paused is None:
-            exclude.add("paused")
-
-        if deployment_create.pull_steps is None:
-            exclude.add("pull_steps")
+        exclude_if_none = [
+            "paused",
+            "pull_steps",
+            "enforce_parameter_schema",
+            "version_info",
+            "branch",
+            "base",
+            "root",
+        ]
 
-        if deployment_create.enforce_parameter_schema is None:
-            exclude.add("enforce_parameter_schema")
+        for field in exclude_if_none:
+            if getattr(deployment_create, field) is None:
+                exclude.add(field)
 
         json = deployment_create.model_dump(mode="json", exclude=exclude)
+
         response = await self.request(
             "POST",
             "/deployments/",
@@ -694,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",
@@ -705,7 +758,7 @@ class DeploymentAsyncClient(BaseAsyncClient):
 
     async def update_deployment(
         self,
-        deployment_id: "UUID",
+        deployment_id: UUID,
         deployment: "DeploymentUpdate",
     ) -> None:
         await self.request(
@@ -719,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.
@@ -740,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.
@@ -751,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
 
@@ -877,7 +926,7 @@ class DeploymentAsyncClient(BaseAsyncClient):
 
     async def delete_deployment(
         self,
-        deployment_id: "UUID",
+        deployment_id: UUID,
     ) -> None:
         """
         Delete deployment by id.
@@ -902,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"]:
         """
@@ -941,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.
@@ -969,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:
@@ -1009,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.
@@ -1036,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"]:
@@ -1058,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,
@@ -1066,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,
@@ -1134,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/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/schemas/actions.py

@@ -253,13 +253,29 @@ class DeploymentCreate(ActionBaseModel):
     infrastructure_document_id: Optional[UUID] = Field(default=None)
     description: Optional[str] = Field(default=None)
     path: Optional[str] = Field(default=None)
-    version: Optional[str] = Field(default=None)
     entrypoint: Optional[str] = Field(default=None)
     job_variables: dict[str, Any] = Field(
         default_factory=dict,
         description="Overrides to apply to flow run infrastructure at runtime.",
     )
 
+    # Versionining
+    version: Optional[str] = Field(default=None)
+    version_info: Optional[objects.VersionInfo] = Field(
+        default=None, description="Version information for 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."
+    )
+
     def check_valid_configuration(self, base_job_template: dict[str, Any]) -> None:
         """Check that the combination of base_job_template defaults
         and job_variables conforms to the specified schema.
@@ -289,6 +305,9 @@ class DeploymentUpdate(ActionBaseModel):
         return remove_old_deployment_fields(values)
 
     version: Optional[str] = Field(default=None)
+    version_info: Optional[objects.VersionInfo] = Field(
+        default=None, description="Version information for the deployment."
+    )
     description: Optional[str] = Field(default=None)
     parameters: Optional[dict[str, Any]] = Field(
         default=None,
@@ -354,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/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/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/context.py

@@ -6,6 +6,7 @@ These contexts should never be directly mutated by the user.
 For more user-accessible information about the current run, see [`prefect.runtime`](../runtime/flow_run).
 """
 
+import asyncio
 import os
 import sys
 import warnings
@@ -294,7 +295,7 @@ class AsyncClientContext(ContextModel):
     @asynccontextmanager
     async def get_or_create(cls) -> AsyncGenerator[Self, None]:
         ctx = cls.get()
-        if ctx:
+        if ctx and asyncio.get_running_loop() is ctx.client.loop:
             yield ctx
         else:
             async with cls() as ctx:

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/server/api/ui/task_runs.py

@@ -1,8 +1,9 @@
 from datetime import datetime
 from typing import TYPE_CHECKING, List, Optional
+from uuid import UUID
 
 import sqlalchemy as sa
-from fastapi import Depends, HTTPException, status
+from fastapi import Depends, HTTPException, Path, status
 from pydantic import Field, model_serializer
 
 import prefect.server.schemas as schemas
@@ -173,3 +174,22 @@ async def read_task_run_counts_by_state(
             task_run_filter=task_runs,
             deployment_filter=deployments,
         )
+
+
+@router.get("/{id}")
+async def read_task_run_with_flow_run_name(
+    task_run_id: UUID = Path(..., description="The task run id", alias="id"),
+    db: PrefectDBInterface = Depends(provide_database_interface),
+) -> schemas.ui.UITaskRun:
+    """
+    Get a task run by id.
+    """
+    async with db.session_context() as session:
+        task_run = await models.task_runs.read_task_run_with_flow_run_name(
+            session=session, task_run_id=task_run_id
+        )
+
+    if not task_run:
+        raise HTTPException(status.HTTP_404_NOT_FOUND, detail="Task not found")
+
+    return schemas.ui.UITaskRun.model_validate(task_run)

{prefect_client-3.3.2.dev2.dist-info → prefect_client-3.3.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prefect-client
-Version: 3.3.2.dev2
+Version: 3.3.3
 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.2.dev2.dist-info → prefect_client-3.3.3.dist-info}/RECORD

@@ -1,14 +1,14 @@
 prefect/.prefectignore,sha256=awSprvKT0vI8a64mEOLrMxhxqcO-b0ERQeYpA2rNKVQ,390
 prefect/__init__.py,sha256=iCdcC5ZmeewikCdnPEP6YBAjPNV5dvfxpYCTpw30Hkw,3685
 prefect/__main__.py,sha256=WFjw3kaYJY6pOTA7WDOgqjsz8zUEUZHCcj3P5wyVa-g,66
-prefect/_build_info.py,sha256=8Nc8NcbIDZ9djzkeEomk1WOEdmQddb0Ae-XmePNGZQo,185
+prefect/_build_info.py,sha256=rkcwN5IlOT8GJmmraH2Q4ljo2fZu5atasia8KZm29DQ,180
 prefect/_result_records.py,sha256=S6QmsODkehGVSzbMm6ig022PYbI6gNKz671p_8kBYx4,7789
 prefect/_waiters.py,sha256=Ia2ITaXdHzevtyWIgJoOg95lrEXQqNEOquHvw3T33UQ,9026
 prefect/agent.py,sha256=dPvG1jDGD5HSH7aM2utwtk6RaJ9qg13XjkA0lAIgQmY,287
 prefect/artifacts.py,sha256=dMBUOAWnUamzjb5HSqwB5-GR2Qb-Gxee26XG5NDCUuw,22720
 prefect/automations.py,sha256=ZzPxn2tINdlXTQo805V4rIlbXuNWxd7cdb3gTJxZIeY,12567
 prefect/cache_policies.py,sha256=Kwdei4JjitNfx42OepKpDNxwPtEwRgUUAn_soxsnNzI,12699
-prefect/context.py,sha256=u1IMAe-X2__pJceNrELI_9aoAam0og9hz47QEyM6mUY,23733
+prefect/context.py,sha256=LYEOlz7ZkuSDj7TmrE4mByy3N3TquFkIE2hEy0WHW1Y,23798
 prefect/engine.py,sha256=uB5JN4l045i5JTlRQNT1x7MwlSiGQ5Bop2Q6jHHOgxY,3699
 prefect/exceptions.py,sha256=wZLQQMRB_DyiYkeEdIC5OKwbba5A94Dlnics-lrWI7A,11581
 prefect/filesystems.py,sha256=v5YqGB4uXf9Ew2VuB9VCSkawvYMMVvEtZf7w1VmAmr8,18036
@@ -82,9 +82,9 @@ prefect/client/collections.py,sha256=t9XkVU_onQMZ871L21F1oZnAiPSQeeVfd_MuDEBS3iM
 prefect/client/constants.py,sha256=Z_GG8KF70vbbXxpJuqW5pLnwzujTVeHbcYYRikNmGH0,29
 prefect/client/subscriptions.py,sha256=b2gjoWjrTjshnv_s6zlPN3t0JIe2EKAdMqEzj3-kc6w,3879
 prefect/client/utilities.py,sha256=UEJD6nwYg2mD8-GSmru-E2ofXaBlmSFZ2-8T_5rIK6c,3472
-prefect/client/orchestration/__init__.py,sha256=_TYB6Dm_rrBJbOXAkxI4uby63Z6rW-YrQOXDVrfLfMs,60675
+prefect/client/orchestration/__init__.py,sha256=1GHRA9-JWPBX6oSVNytQ1qL6epFeslRT2oBgZdFDJ68,60807
 prefect/client/orchestration/base.py,sha256=HM6ryHBZSzuHoCFQM9u5qR5k1dN9Bbr_ah6z1UPNbZQ,1542
-prefect/client/orchestration/routes.py,sha256=JFG1OWUBfrxPKW8Q7XWItlhOrSZ67IOySSoFZ6mxzm0,4364
+prefect/client/orchestration/routes.py,sha256=FHG9xyop_eXp6wtg0Fp0G14J6lO8UVsdnhGY5qvnJ9k,4396
 prefect/client/orchestration/_artifacts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prefect/client/orchestration/_artifacts/client.py,sha256=0GEM4rJWeedKR2xVgWQcX6DpLyn0zKFJF9nfRCQ4tpM,8855
 prefect/client/orchestration/_automations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -98,7 +98,7 @@ prefect/client/orchestration/_blocks_types/client.py,sha256=alA4xD-yp3mycAbzMyRu
 prefect/client/orchestration/_concurrency_limits/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prefect/client/orchestration/_concurrency_limits/client.py,sha256=r_oyY7hQbgyG1rntwe7WWcsraQHBKhk6MOPFUAHWiVc,23678
 prefect/client/orchestration/_deployments/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-prefect/client/orchestration/_deployments/client.py,sha256=DfL6eiMaPkYBwCmkMcjKnyVIvE1qC8Xo14h2YqJhFsI,40769
+prefect/client/orchestration/_deployments/client.py,sha256=I2-RhU15lbEI1_JX_-WJErfp3lkvTCJHja-p4y8hzLA,42923
 prefect/client/orchestration/_flow_runs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prefect/client/orchestration/_flow_runs/client.py,sha256=fjh5J-LG8tsny7BGYEvynbuGuHDAudYHpx-PamL0GYQ,32220
 prefect/client/orchestration/_flows/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -110,10 +110,10 @@ prefect/client/orchestration/_variables/client.py,sha256=wKBbZBLGgs5feDCil-xxKt3
 prefect/client/orchestration/_work_pools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prefect/client/orchestration/_work_pools/client.py,sha256=s1DfUQQBgB2sLiVVPhLNTlkueUDE6uFsh4mAzcSA1OE,19881
 prefect/client/schemas/__init__.py,sha256=InZcDzdeWA2oaV0TlyvoMcyLcbi_aaqU1U9D6Gx-eoU,2747
-prefect/client/schemas/actions.py,sha256=I8LGTyDBrV4eXI_VSq8nQz5jUuOUAn1A0Q5JsPgCa2A,33032
+prefect/client/schemas/actions.py,sha256=e4I21SQZt0zSIxdHgPox269eUUOPHzcv0xytgsmyKqY,34403
 prefect/client/schemas/filters.py,sha256=zaiDkalrIpKjd38V4aP1GHlqD24KTPCZiKtPyX69ZWE,36607
-prefect/client/schemas/objects.py,sha256=qEV2lIdK6JGGhNsAiUDsvtPbjdIZZABqdjnA8cMsnak,58999
-prefect/client/schemas/responses.py,sha256=9x8gP2O52xEf40EbnSKaKwxS1vIg347t8CqgHGMYCNg,16664
+prefect/client/schemas/objects.py,sha256=nyaodVdbZIl-p5e7BvF5olPoiHZkymXkn7nFkFnXqXA,59402
+prefect/client/schemas/responses.py,sha256=Zdcx7jlIaluEa2uYIOE5mK1HsJvWPErRAcaWM20oY_I,17336
 prefect/client/schemas/schedules.py,sha256=sxLFk0SmFY7X1Y9R9HyGDqOS3U5NINBWTciUU7vTTic,14836
 prefect/client/schemas/sorting.py,sha256=L-2Mx-igZPtsUoRUguTcG3nIEstMEMPD97NwPM2Ox5s,2579
 prefect/client/types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -209,7 +209,7 @@ prefect/server/api/deployments.py,sha256=d-GAbVP3T0RVkkz6VN5nsQN7XU7fTjQg-vzYxuo
 prefect/server/api/events.py,sha256=3-Qdt6ORxFv3nLoogQqvd72zEulJSoAmcqZto2OULuk,9907
 prefect/server/api/flow_run_notification_policies.py,sha256=F8xNm6bgZTC3nFe9xCUJS4NlU9tLXZ8fShtJqmhT2m4,4828
 prefect/server/api/flow_run_states.py,sha256=lIdxVE9CqLgtDCuH9bTaKkzHNL81FPrr11liPzvONrw,1661
-prefect/server/api/flow_runs.py,sha256=W_9S7MGrTnUnY5NJdjn13LKdRWeUcyzj8lKx0i1BfH0,33709
+prefect/server/api/flow_runs.py,sha256=Lmb165fLbN4DioxjxgDYaAJ5Qxj771iRYaqn-hYq9KM,33744
 prefect/server/api/flows.py,sha256=Bz0ISh-9oY0W1X3mqA631_8678pQ6tuRGMpSgWAfxOc,7018
 prefect/server/api/logs.py,sha256=0z78tM2B5sRgJWYRWJn5lHhRoLtZB_OU3C-uALV8tOs,1571
 prefect/server/api/middleware.py,sha256=WkyuyeJIfo9Q0GAIVU5gO6yIGNVwoHwuBah5AB5oUyw,2733
@@ -231,7 +231,7 @@ prefect/server/api/ui/__init__.py,sha256=TCXO4ZUZCqCbm2QoNvWNTErkzWiX2nSACuO-0Ti
 prefect/server/api/ui/flow_runs.py,sha256=ALmUFY4WrJggN1ha0z-tqXeddG2GptswbPnB7iYixUM,4172
 prefect/server/api/ui/flows.py,sha256=W4kwqOCJ_2vROmMCmemH2Mq3uWbWZyu5q5uTZPBdYwk,5902
 prefect/server/api/ui/schemas.py,sha256=NVWA1RFnHW-MMU1s6WbNmp_S5mhbrN-_P41I4O2XtMg,2085
-prefect/server/api/ui/task_runs.py,sha256=dKVNe4EjmMrXH-VtsRoFTKJEBbl35upGHAW2C_AhKKM,6664
+prefect/server/api/ui/task_runs.py,sha256=6CMrHmY-ybJGHXz7YlVVP2ZTmvq7w-XA9GUHqCcw_7o,7319
 prefect/settings/__init__.py,sha256=3jDLzExmq9HsRWo1kTSE16BO_3B3JlVsk5pR0s4PWEQ,2136
 prefect/settings/base.py,sha256=HGukXOXOokfqmrVirgejNskKtf1x2QheZ-ldRakxPJA,9701
 prefect/settings/constants.py,sha256=5NjVLG1Km9J9I-a6wrq-qmi_dTkPdwEk3IrY9bSxWvw,281
@@ -316,7 +316,7 @@ prefect/workers/cloud.py,sha256=dPvG1jDGD5HSH7aM2utwtk6RaJ9qg13XjkA0lAIgQmY,287
 prefect/workers/process.py,sha256=uxOwcqA2Ps-V-W6WeSdKCQMINrCxBEVx1K1Un8pb7vs,8973
 prefect/workers/server.py,sha256=2pmVeJZiVbEK02SO6BEZaBIvHMsn6G8LzjW8BXyiTtk,1952
 prefect/workers/utilities.py,sha256=VfPfAlGtTuDj0-Kb8WlMgAuOfgXCdrGAnKMapPSBrwc,2483
-prefect_client-3.3.2.dev2.dist-info/METADATA,sha256=cARpEVGuMuSP0iE0-J9_UfPSIhf2guyrHfjIMtCY0XY,7456
-prefect_client-3.3.2.dev2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-prefect_client-3.3.2.dev2.dist-info/licenses/LICENSE,sha256=MCxsn8osAkzfxKC4CC_dLcUkU8DZLkyihZ8mGs3Ah3Q,11357
-prefect_client-3.3.2.dev2.dist-info/RECORD,,
+prefect_client-3.3.3.dist-info/METADATA,sha256=iOCVZHQAkXFFvPYp6Z-LNCAfswoTz_HQDvEfYcZ7i5E,7451
+prefect_client-3.3.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+prefect_client-3.3.3.dist-info/licenses/LICENSE,sha256=MCxsn8osAkzfxKC4CC_dLcUkU8DZLkyihZ8mGs3Ah3Q,11357
+prefect_client-3.3.3.dist-info/RECORD,,