prefect 3.1.7.dev2__tar.gz → 3.1.7.dev3__tar.gz

{prefect-3.1.7.dev2/src/prefect.egg-info → prefect-3.1.7.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prefect
-Version: 3.1.7.dev2
+Version: 3.1.7.dev3
 Summary: Workflow orchestration and management.
 Home-page: https://www.prefect.io
 Author: Prefect Technologies, Inc.

{prefect-3.1.7.dev2 → prefect-3.1.7.dev3}/src/prefect/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2024-12-12T19:33:15-0500",
+ "date": "2024-12-13T20:55:38-0600",
  "dirty": false,
  "error": null,
- "full-revisionid": "40590a5fb0c5dca63aa96f3b706d7b74ce08dc0a",
- "version": "3.1.7.dev2"
+ "full-revisionid": "df9d64f2d2f08597d7ff00700b6bfd233634b273",
+ "version": "3.1.7.dev3"
 }
 '''  # END VERSION_JSON
 

{prefect-3.1.7.dev2 → prefect-3.1.7.dev3}/src/prefect/cli/global_concurrency_limit.py

@@ -150,14 +150,16 @@ async def delete_global_concurrency_limit(
     Arguments:
         name (str): The name of the global concurrency limit to delete.
     """
-    if is_interactive() and not typer.confirm(
-        f"Are you sure you want to delete global concurrency limit with name {name!r}?",
-        default=False,
-    ):
-        exit_with_error("Deletion aborted.")
-
     async with get_client() as client:
         try:
+            gcl_limit = await client.read_global_concurrency_limit_by_name(name=name)
+
+            if is_interactive() and not typer.confirm(
+                f"Are you sure you want to delete global concurrency limit with name {gcl_limit.name!r}?",
+                default=False,
+            ):
+                exit_with_error("Deletion aborted.")
+
             await client.delete_global_concurrency_limit_by_name(name=name)
         except ObjectNotFound:
             exit_with_error(f"Global concurrency limit {name!r} not found.")

prefect-3.1.7.dev3/src/prefect/client/_adapters.py

@@ -0,0 +1,123 @@
+"""Creating type adapters to avoid recreation of the pydantic core schemas when possible."""
+
+from typing import TYPE_CHECKING
+
+from pydantic import ConfigDict, TypeAdapter
+
+if TYPE_CHECKING:
+    from prefect.client.schemas.objects import (
+        Artifact,
+        ArtifactCollection,
+        BlockDocument,
+        BlockSchema,
+        BlockType,
+        ConcurrencyLimit,
+        DeploymentSchedule,
+        Flow,
+        FlowRun,
+        FlowRunInput,
+        FlowRunNotificationPolicy,
+        GlobalConcurrencyLimit,
+        Log,
+        State,
+        TaskRun,
+        Variable,
+        Worker,
+        WorkPool,
+        WorkQueue,
+    )
+    from prefect.client.schemas.responses import (
+        DeploymentResponse,
+        FlowRunResponse,
+        GlobalConcurrencyLimitResponse,
+        WorkerFlowRunResponse,
+    )
+    from prefect.events.schemas.automations import Automation
+
+defer_build_cfg = ConfigDict(defer_build=True)
+
+# Create the adapters with forward refs
+BlockTypeAdapter: TypeAdapter["BlockType"] = TypeAdapter(
+    "BlockType", config=defer_build_cfg
+)
+BlockSchemaAdapter: TypeAdapter[list["BlockSchema"]] = TypeAdapter(
+    list["BlockSchema"], config=defer_build_cfg
+)
+ConcurrencyLimitAdapter: TypeAdapter["ConcurrencyLimit"] = TypeAdapter(
+    "ConcurrencyLimit", config=defer_build_cfg
+)
+ConcurrencyLimitListAdapter: TypeAdapter[list["ConcurrencyLimit"]] = TypeAdapter(
+    list["ConcurrencyLimit"], config=defer_build_cfg
+)
+FlowAdapter: TypeAdapter["Flow"] = TypeAdapter("Flow", config=defer_build_cfg)
+FlowRunAdapter: TypeAdapter["FlowRun"] = TypeAdapter("FlowRun", config=defer_build_cfg)
+ArtifactListAdapter: TypeAdapter[list["Artifact"]] = TypeAdapter(
+    list["Artifact"], config=defer_build_cfg
+)
+ArtifactCollectionListAdapter: TypeAdapter[list["ArtifactCollection"]] = TypeAdapter(
+    list["ArtifactCollection"], config=defer_build_cfg
+)
+AutomationListAdapter: TypeAdapter[list["Automation"]] = TypeAdapter(
+    list["Automation"], config=defer_build_cfg
+)
+BlockDocumentListAdapter: TypeAdapter[list["BlockDocument"]] = TypeAdapter(
+    list["BlockDocument"], config=defer_build_cfg
+)
+BlockSchemaListAdapter: TypeAdapter[list["BlockSchema"]] = TypeAdapter(
+    list["BlockSchema"], config=defer_build_cfg
+)
+BlockTypeListAdapter: TypeAdapter[list["BlockType"]] = TypeAdapter(
+    list["BlockType"], config=defer_build_cfg
+)
+DeploymentResponseListAdapter: TypeAdapter[list["DeploymentResponse"]] = TypeAdapter(
+    list["DeploymentResponse"], config=defer_build_cfg
+)
+DeploymentScheduleListAdapter: TypeAdapter[list["DeploymentSchedule"]] = TypeAdapter(
+    list["DeploymentSchedule"], config=defer_build_cfg
+)
+FlowListAdapter: TypeAdapter[list["Flow"]] = TypeAdapter(
+    list["Flow"], config=defer_build_cfg
+)
+FlowRunListAdapter: TypeAdapter[list["FlowRun"]] = TypeAdapter(
+    list["FlowRun"], config=defer_build_cfg
+)
+FlowRunInputListAdapter: TypeAdapter[list["FlowRunInput"]] = TypeAdapter(
+    list["FlowRunInput"], config=defer_build_cfg
+)
+FlowRunNotificationPolicyListAdapter: TypeAdapter[
+    list["FlowRunNotificationPolicy"]
+] = TypeAdapter(list["FlowRunNotificationPolicy"], config=defer_build_cfg)
+FlowRunResponseListAdapter: TypeAdapter[list["FlowRunResponse"]] = TypeAdapter(
+    list["FlowRunResponse"], config=defer_build_cfg
+)
+GlobalConcurrencyLimitListAdapter = TypeAdapter(
+    list["GlobalConcurrencyLimit"], config=defer_build_cfg
+)
+GlobalConcurrencyLimitResponseListAdapter: TypeAdapter[
+    list["GlobalConcurrencyLimitResponse"]
+] = TypeAdapter(list["GlobalConcurrencyLimitResponse"], config=defer_build_cfg)
+LogListAdapter: TypeAdapter[list["Log"]] = TypeAdapter(
+    list["Log"], config=defer_build_cfg
+)
+StateListAdapter: TypeAdapter[list["State"]] = TypeAdapter(
+    list["State"], config=defer_build_cfg
+)
+TaskRunListAdapter: TypeAdapter[list["TaskRun"]] = TypeAdapter(
+    list["TaskRun"], config=defer_build_cfg
+)
+WorkerListAdapter: TypeAdapter[list["Worker"]] = TypeAdapter(
+    list["Worker"], config=defer_build_cfg
+)
+WorkPoolListAdapter: TypeAdapter[list["WorkPool"]] = TypeAdapter(
+    list["WorkPool"], config=defer_build_cfg
+)
+WorkQueueListAdapter: TypeAdapter[list["WorkQueue"]] = TypeAdapter(
+    list["WorkQueue"], config=defer_build_cfg
+)
+WorkerFlowRunResponseListAdapter: TypeAdapter[
+    list["WorkerFlowRunResponse"]
+] = TypeAdapter(list["WorkerFlowRunResponse"], config=defer_build_cfg)
+VariableListAdapter: TypeAdapter[list["Variable"]] = TypeAdapter(
+    list["Variable"], config=defer_build_cfg
+)
+StateAdapter: TypeAdapter["State"] = TypeAdapter("State", config=defer_build_cfg)

{prefect-3.1.7.dev2 → prefect-3.1.7.dev3}/src/prefect/client/orchestration.py

@@ -12,7 +12,6 @@ import certifi
 import httpcore
 import httpx
 import pendulum
-import pydantic
 from asgi_lifespan import LifespanManager
 from packaging import version
 from starlette import status
@@ -22,6 +21,34 @@ import prefect
 import prefect.exceptions
 import prefect.settings
 import prefect.states
+from prefect.client._adapters import (
+    ArtifactCollectionListAdapter,
+    ArtifactListAdapter,
+    AutomationListAdapter,
+    BlockDocumentListAdapter,
+    BlockSchemaListAdapter,
+    BlockTypeListAdapter,
+    ConcurrencyLimitAdapter,
+    ConcurrencyLimitListAdapter,
+    DeploymentResponseListAdapter,
+    DeploymentScheduleListAdapter,
+    FlowAdapter,
+    FlowListAdapter,
+    FlowRunAdapter,
+    FlowRunInputListAdapter,
+    FlowRunListAdapter,
+    FlowRunNotificationPolicyListAdapter,
+    FlowRunResponseListAdapter,
+    GlobalConcurrencyLimitResponseListAdapter,
+    LogListAdapter,
+    StateListAdapter,
+    TaskRunListAdapter,
+    VariableListAdapter,
+    WorkerFlowRunResponseListAdapter,
+    WorkerListAdapter,
+    WorkPoolListAdapter,
+    WorkQueueListAdapter,
+)
 from prefect.client.constants import SERVER_API_VERSION
 from prefect.client.schemas import FlowRun, OrchestrationResult, TaskRun, sorting
 from prefect.client.schemas.actions import (
@@ -85,6 +112,7 @@ from prefect.client.schemas.objects import (
     FlowRunPolicy,
     Log,
     Parameter,
+    State,
     TaskRunPolicy,
     TaskRunResult,
     Variable,
@@ -113,6 +141,7 @@ from prefect.client.schemas.sorting import (
 from prefect.events import filters
 from prefect.events.schemas.automations import Automation, AutomationCore
 from prefect.logging import get_logger
+from prefect.results import BaseResult, ResultRecordMetadata
 from prefect.settings import (
     PREFECT_API_DATABASE_CONNECTION_URL,
     PREFECT_API_ENABLE_HTTP2,
@@ -146,6 +175,12 @@ R = TypeVar("R", infer_variance=True)
 T = TypeVar("T")
 
 
+BaseResult.model_rebuild()
+ResultRecordMetadata.model_rebuild()
+
+State.model_rebuild()
+
+
 @overload
 def get_client(
     *,
@@ -499,7 +534,7 @@ class PrefectClient:
             a [Flow model][prefect.client.schemas.objects.Flow] representation of the flow
         """
         response = await self._client.get(f"/flows/{flow_id}")
-        return Flow.model_validate(response.json())
+        return FlowAdapter.validate_python(response.json())
 
     async def delete_flow(self, flow_id: UUID) -> None:
         """
@@ -575,7 +610,7 @@ class PrefectClient:
         }
 
         response = await self._client.post("/flows/filter", json=body)
-        return pydantic.TypeAdapter(list[Flow]).validate_python(response.json())
+        return FlowListAdapter.validate_python(response.json())
 
     async def read_flow_by_name(
         self,
@@ -713,13 +748,13 @@ class PrefectClient:
             state=state.to_state_create(),
             empirical_policy=FlowRunPolicy(
                 retries=flow.retries,
-                retry_delay=int(flow.retry_delay_seconds or 0),
+                retry_delay=flow.retry_delay_seconds,  # type: ignore
             ),
         )
 
         flow_run_create_json = flow_run_create.model_dump(mode="json")
         response = await self._client.post("/flow_runs/", json=flow_run_create_json)
-        flow_run = FlowRun.model_validate(response.json())
+        flow_run = FlowRunAdapter.validate_python(response.json())
 
         # Restore the parameters to the local objects to retain expectations about
         # Python objects
@@ -869,8 +904,7 @@ class PrefectClient:
         if not concurrency_limit_id:
             raise httpx.RequestError(f"Malformed response: {response}")
 
-        concurrency_limit = ConcurrencyLimit.model_validate(response.json())
-        return concurrency_limit
+        return ConcurrencyLimitAdapter.validate_python(response.json())
 
     async def read_concurrency_limits(
         self,
@@ -894,9 +928,7 @@ class PrefectClient:
         }
 
         response = await self._client.post("/concurrency_limits/filter", json=body)
-        return pydantic.TypeAdapter(list[ConcurrencyLimit]).validate_python(
-            response.json()
-        )
+        return ConcurrencyLimitListAdapter.validate_python(response.json())
 
     async def reset_concurrency_limit_by_tag(
         self,
@@ -1161,7 +1193,7 @@ class PrefectClient:
                 raise prefect.exceptions.ObjectNotFound(http_exc=e) from e
             else:
                 raise
-        return pydantic.TypeAdapter(list[FlowRun]).validate_python(response.json())
+        return FlowRunListAdapter.validate_python(response.json())
 
     async def read_work_queue(
         self,
@@ -1464,7 +1496,7 @@ class PrefectClient:
             List of BlockTypes.
         """
         response = await self._client.post("/block_types/filter", json={})
-        return pydantic.TypeAdapter(list[BlockType]).validate_python(response.json())
+        return BlockTypeListAdapter.validate_python(response.json())
 
     async def read_block_schemas(self) -> list[BlockSchema]:
         """
@@ -1476,7 +1508,7 @@ class PrefectClient:
             A BlockSchema.
         """
         response = await self._client.post("/block_schemas/filter", json={})
-        return pydantic.TypeAdapter(list[BlockSchema]).validate_python(response.json())
+        return BlockSchemaListAdapter.validate_python(response.json())
 
     async def get_most_recent_block_schema_for_block_type(
         self,
@@ -1617,9 +1649,7 @@ class PrefectClient:
                 include_secrets=include_secrets,
             ),
         )
-        return pydantic.TypeAdapter(list[BlockDocument]).validate_python(
-            response.json()
-        )
+        return BlockDocumentListAdapter.validate_python(response.json())
 
     async def read_block_documents_by_type(
         self,
@@ -1648,9 +1678,7 @@ class PrefectClient:
             ),
         )
 
-        return pydantic.TypeAdapter(list[BlockDocument]).validate_python(
-            response.json()
-        )
+        return BlockDocumentListAdapter.validate_python(response.json())
 
     async def create_deployment(
         self,
@@ -1924,9 +1952,7 @@ class PrefectClient:
         }
 
         response = await self._client.post("/deployments/filter", json=body)
-        return pydantic.TypeAdapter(list[DeploymentResponse]).validate_python(
-            response.json()
-        )
+        return DeploymentResponseListAdapter.validate_python(response.json())
 
     async def delete_deployment(
         self,
@@ -1980,9 +2006,7 @@ class PrefectClient:
         response = await self._client.post(
             f"/deployments/{deployment_id}/schedules", json=json
         )
-        return pydantic.TypeAdapter(list[DeploymentSchedule]).validate_python(
-            response.json()
-        )
+        return DeploymentScheduleListAdapter.validate_python(response.json())
 
     async def read_deployment_schedules(
         self,
@@ -2004,9 +2028,7 @@ class PrefectClient:
                 raise prefect.exceptions.ObjectNotFound(http_exc=e) from e
             else:
                 raise
-        return pydantic.TypeAdapter(list[DeploymentSchedule]).validate_python(
-            response.json()
-        )
+        return DeploymentScheduleListAdapter.validate_python(response.json())
 
     async def update_deployment_schedule(
         self,
@@ -2085,7 +2107,7 @@ class PrefectClient:
                 raise prefect.exceptions.ObjectNotFound(http_exc=e) from e
             else:
                 raise
-        return FlowRun.model_validate(response.json())
+        return FlowRunAdapter.validate_python(response.json())
 
     async def resume_flow_run(
         self, flow_run_id: UUID, run_input: Optional[dict[str, Any]] = None
@@ -2169,7 +2191,7 @@ class PrefectClient:
         }
 
         response = await self._client.post("/flow_runs/filter", json=body)
-        return pydantic.TypeAdapter(list[FlowRun]).validate_python(response.json())
+        return FlowRunListAdapter.validate_python(response.json())
 
     async def set_flow_run_state(
         self,
@@ -2230,9 +2252,8 @@ class PrefectClient:
         response = await self._client.get(
             "/flow_run_states/", params=dict(flow_run_id=str(flow_run_id))
         )
-        return pydantic.TypeAdapter(list[prefect.states.State]).validate_python(
-            response.json()
-        )
+        StateListAdapter.rebuild()
+        return StateListAdapter.validate_python(response.json())
 
     async def set_flow_run_name(self, flow_run_id: UUID, name: str) -> httpx.Response:
         flow_run_data = FlowRunUpdate(name=name)
@@ -2310,7 +2331,7 @@ class PrefectClient:
             task_version=task.version,
             empirical_policy=TaskRunPolicy(
                 retries=task.retries,
-                retry_delay=retry_delay,
+                retry_delay=task.retry_delay_seconds,  # type: ignore
                 retry_jitter_factor=task.retry_jitter_factor,
             ),
             state=state.to_state_create(),
@@ -2386,7 +2407,7 @@ class PrefectClient:
             "offset": offset,
         }
         response = await self._client.post("/task_runs/filter", json=body)
-        return pydantic.TypeAdapter(list[TaskRun]).validate_python(response.json())
+        return TaskRunListAdapter.validate_python(response.json())
 
     async def delete_task_run(self, task_run_id: UUID) -> None:
         """
@@ -2450,9 +2471,7 @@ class PrefectClient:
         response = await self._client.get(
             "/task_run_states/", params=dict(task_run_id=str(task_run_id))
         )
-        return pydantic.TypeAdapter(list[prefect.states.State]).validate_python(
-            response.json()
-        )
+        return StateListAdapter.validate_python(response.json())
 
     async def create_logs(
         self, logs: Iterable[Union[LogCreate, dict[str, Any]]]
@@ -2610,9 +2629,7 @@ class PrefectClient:
         response = await self._client.post(
             "/flow_run_notification_policies/filter", json=body
         )
-        return pydantic.TypeAdapter(list[FlowRunNotificationPolicy]).validate_python(
-            response.json()
-        )
+        return FlowRunNotificationPolicyListAdapter.validate_python(response.json())
 
     async def read_logs(
         self,
@@ -2632,7 +2649,7 @@ class PrefectClient:
         }
 
         response = await self._client.post("/logs/filter", json=body)
-        return pydantic.TypeAdapter(list[Log]).validate_python(response.json())
+        return LogListAdapter.validate_python(response.json())
 
     async def send_worker_heartbeat(
         self,
@@ -2707,7 +2724,7 @@ class PrefectClient:
             },
         )
 
-        return pydantic.TypeAdapter(list[Worker]).validate_python(response.json())
+        return WorkerListAdapter.validate_python(response.json())
 
     async def read_work_pool(self, work_pool_name: str) -> WorkPool:
         """
@@ -2755,7 +2772,7 @@ class PrefectClient:
             ),
         }
         response = await self._client.post("/work_pools/filter", json=body)
-        return pydantic.TypeAdapter(list[WorkPool]).validate_python(response.json())
+        return WorkPoolListAdapter.validate_python(response.json())
 
     async def create_work_pool(
         self,
@@ -2885,7 +2902,7 @@ class PrefectClient:
         else:
             response = await self._client.post("/work_queues/filter", json=json)
 
-        return pydantic.TypeAdapter(list[WorkQueue]).validate_python(response.json())
+        return WorkQueueListAdapter.validate_python(response.json())
 
     async def get_scheduled_flow_runs_for_deployments(
         self,
@@ -2904,9 +2921,7 @@ class PrefectClient:
             json=body,
         )
 
-        return pydantic.TypeAdapter(list[FlowRunResponse]).validate_python(
-            response.json()
-        )
+        return FlowRunResponseListAdapter.validate_python(response.json())
 
     async def get_scheduled_flow_runs_for_work_pool(
         self,
@@ -2939,9 +2954,7 @@ class PrefectClient:
             f"/work_pools/{work_pool_name}/get_scheduled_flow_runs",
             json=body,
         )
-        return pydantic.TypeAdapter(list[WorkerFlowRunResponse]).validate_python(
-            response.json()
-        )
+        return WorkerFlowRunResponseListAdapter.validate_python(response.json())
 
     async def create_artifact(
         self,
@@ -3020,7 +3033,7 @@ class PrefectClient:
             "offset": offset,
         }
         response = await self._client.post("/artifacts/filter", json=body)
-        return pydantic.TypeAdapter(list[Artifact]).validate_python(response.json())
+        return ArtifactListAdapter.validate_python(response.json())
 
     async def read_latest_artifacts(
         self,
@@ -3060,9 +3073,7 @@ class PrefectClient:
             "offset": offset,
         }
         response = await self._client.post("/artifacts/latest/filter", json=body)
-        return pydantic.TypeAdapter(list[ArtifactCollection]).validate_python(
-            response.json()
-        )
+        return ArtifactCollectionListAdapter.validate_python(response.json())
 
     async def delete_artifact(self, artifact_id: UUID) -> None:
         """
@@ -3132,7 +3143,7 @@ class PrefectClient:
     async def read_variables(self, limit: Optional[int] = None) -> list[Variable]:
         """Reads all variables."""
         response = await self._client.post("/variables/filter", json={"limit": limit})
-        return pydantic.TypeAdapter(list[Variable]).validate_python(response.json())
+        return VariableListAdapter.validate_python(response.json())
 
     async def read_worker_metadata(self) -> dict[str, Any]:
         """Reads worker metadata stored in Prefect collection registry."""
@@ -3266,9 +3277,9 @@ class PrefectClient:
                 "offset": offset,
             },
         )
-        return pydantic.TypeAdapter(
-            list[GlobalConcurrencyLimitResponse]
-        ).validate_python(response.json())
+        return GlobalConcurrencyLimitResponseListAdapter.validate_python(
+            response.json()
+        )
 
     async def create_flow_run_input(
         self, flow_run_id: UUID, key: str, value: str, sender: Optional[str] = None
@@ -3304,7 +3315,7 @@ class PrefectClient:
             },
         )
         response.raise_for_status()
-        return pydantic.TypeAdapter(list[FlowRunInput]).validate_python(response.json())
+        return FlowRunInputListAdapter.validate_python(response.json())
 
     async def read_flow_run_input(self, flow_run_id: UUID, key: str) -> str:
         """
@@ -3351,7 +3362,7 @@ class PrefectClient:
     async def read_automations(self) -> list[Automation]:
         response = await self._client.post("/automations/filter")
         response.raise_for_status()
-        return pydantic.TypeAdapter(list[Automation]).validate_python(response.json())
+        return AutomationListAdapter.validate_python(response.json())
 
     async def find_automation(
         self, id_or_name: Union[str, UUID]
@@ -3423,7 +3434,7 @@ class PrefectClient:
 
         response.raise_for_status()
 
-        return pydantic.TypeAdapter(list[Automation]).validate_python(response.json())
+        return AutomationListAdapter.validate_python(response.json())
 
     async def pause_automation(self, automation_id: UUID) -> None:
         response = await self._client.patch(
@@ -3449,7 +3460,7 @@ class PrefectClient:
     ) -> list[Automation]:
         response = await self._client.get(f"/automations/related-to/{resource_id}")
         response.raise_for_status()
-        return pydantic.TypeAdapter(list[Automation]).validate_python(response.json())
+        return AutomationListAdapter.validate_python(response.json())
 
     async def delete_resource_owned_automations(self, resource_id: str) -> None:
         await self._client.delete(f"/automations/owned-by/{resource_id}")
@@ -3899,13 +3910,13 @@ class SyncPrefectClient:
             state=state.to_state_create(),
             empirical_policy=FlowRunPolicy(
                 retries=flow.retries,
-                retry_delay=int(flow.retry_delay_seconds or 0),
+                retry_delay=flow.retry_delay_seconds,  # type: ignore
             ),
         )
 
         flow_run_create_json = flow_run_create.model_dump(mode="json")
         response = self._client.post("/flow_runs/", json=flow_run_create_json)
-        flow_run = FlowRun.model_validate(response.json())
+        flow_run = FlowRunAdapter.validate_python(response.json())
 
         # Restore the parameters to the local objects to retain expectations about
         # Python objects
@@ -3985,7 +3996,7 @@ class SyncPrefectClient:
                 raise prefect.exceptions.ObjectNotFound(http_exc=e) from e
             else:
                 raise
-        return FlowRun.model_validate(response.json())
+        return FlowRunAdapter.validate_python(response.json())
 
     def read_flow_runs(
         self,
@@ -4044,7 +4055,7 @@ class SyncPrefectClient:
         }
 
         response = self._client.post("/flow_runs/filter", json=body)
-        return pydantic.TypeAdapter(list[FlowRun]).validate_python(response.json())
+        return FlowRunListAdapter.validate_python(response.json())
 
     def set_flow_run_state(
         self,
@@ -4162,7 +4173,7 @@ class SyncPrefectClient:
             task_version=task.version,
             empirical_policy=TaskRunPolicy(
                 retries=task.retries,
-                retry_delay=retry_delay,
+                retry_delay=task.retry_delay_seconds,  # type: ignore
                 retry_jitter_factor=task.retry_jitter_factor,
             ),
             state=state.to_state_create(),
@@ -4239,7 +4250,7 @@ class SyncPrefectClient:
             "offset": offset,
         }
         response = self._client.post("/task_runs/filter", json=body)
-        return pydantic.TypeAdapter(list[TaskRun]).validate_python(response.json())
+        return TaskRunListAdapter.validate_python(response.json())
 
     def set_task_run_state(
         self,
@@ -4283,9 +4294,7 @@ class SyncPrefectClient:
         response = self._client.get(
             "/task_run_states/", params=dict(task_run_id=str(task_run_id))
         )
-        return pydantic.TypeAdapter(list[prefect.states.State]).validate_python(
-            response.json()
-        )
+        return StateListAdapter.validate_python(response.json())
 
     def read_deployment(
         self,

{prefect-3.1.7.dev2 → prefect-3.1.7.dev3}/src/prefect/client/schemas/actions.py

@@ -393,10 +393,10 @@ class DeploymentFlowRunCreate(ActionBaseModel):
         default_factory=objects.FlowRunPolicy
     )
     tags: list[str] = Field(default_factory=list)
-    idempotency_key: Optional[str] = Field(None)
-    parent_task_run_id: Optional[UUID] = Field(None)
-    work_queue_name: Optional[str] = Field(None)
-    job_variables: Optional[dict] = Field(None)
+    idempotency_key: Optional[str] = Field(default=None)
+    parent_task_run_id: Optional[UUID] = Field(default=None)
+    work_queue_name: Optional[str] = Field(default=None)
+    job_variables: Optional[dict[str, Any]] = Field(default=None)
     labels: KeyValueLabelsField = Field(default_factory=dict)
 
 

{prefect-3.1.7.dev2 → prefect-3.1.7.dev3}/src/prefect/client/schemas/objects.py

@@ -207,7 +207,7 @@ class State(ObjectBaseModel, Generic[R]):
 
     type: StateType
     name: Optional[str] = Field(default=None)
-    timestamp: DateTime = Field(default_factory=lambda: pendulum.now("UTC"))
+    timestamp: DateTime = Field(default_factory=lambda: DateTime.now("UTC"))
     message: Optional[str] = Field(default=None, examples=["Run started"])
     state_details: StateDetails = Field(default_factory=StateDetails)
     data: Annotated[
@@ -429,9 +429,9 @@ class State(ObjectBaseModel, Generic[R]):
         return self.model_copy(
             update={
                 "id": uuid4(),
-                "created": pendulum.now("utc"),
-                "updated": pendulum.now("utc"),
-                "timestamp": pendulum.now("utc"),
+                "created": DateTime.now("utc"),
+                "updated": DateTime.now("utc"),
+                "timestamp": DateTime.now("utc"),
             },
             **kwargs,
         )