prefect-client 3.2.10__tar.gz → 3.2.12__tar.gz

{prefect_client-3.2.10 → prefect_client-3.2.12}/.gitignore

@@ -7,8 +7,9 @@ __pycache__/
 build/
 dist/
 sdist/
-# We likely ought to eventually include uv.lock
-uv.lock
+
+# uv (consider committing later)
+src/integrations/**/uv.lock
 
 # Build metadata
 src/prefect/_build_info.py

{prefect_client-3.2.10 → prefect_client-3.2.12}/PKG-INFO

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

prefect_client-3.2.12/src/prefect/_build_info.py

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.2.12"
+__build_date__ = "2025-03-10 16:35:36.997425+00:00"
+__git_commit__ = "826eb1a7423831a2eeb6541909ef35ecca17fde7"
+__dirty__ = False

{prefect_client-3.2.10 → prefect_client-3.2.12}/src/prefect/client/schemas/actions.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from copy import deepcopy
 from typing import TYPE_CHECKING, Any, Callable, Optional, TypeVar, Union
 from uuid import UUID, uuid4

{prefect_client-3.2.10 → prefect_client-3.2.12}/src/prefect/deployments/flow_runs.py

@@ -4,7 +4,6 @@ from uuid import UUID
 
 import anyio
 from opentelemetry import trace
-from opentelemetry.instrumentation.utils import is_instrumentation_enabled
 
 import prefect
 from prefect._result_records import ResultRecordMetadata
@@ -19,6 +18,16 @@ from prefect.types._datetime import now
 from prefect.utilities.asyncutils import sync_compatible
 from prefect.utilities.slugify import slugify
 
+
+def _is_instrumentation_enabled() -> bool:
+    try:
+        from opentelemetry.instrumentation.utils import is_instrumentation_enabled
+
+        return is_instrumentation_enabled()
+    except (ImportError, ModuleNotFoundError):
+        return False
+
+
 if TYPE_CHECKING:
     from prefect.client.orchestration import PrefectClient
     from prefect.client.schemas.objects import FlowRun
@@ -164,7 +173,7 @@ async def run_deployment(
 
     if flow_run_ctx and flow_run_ctx.flow_run:
         traceparent = flow_run_ctx.flow_run.labels.get(LABELS_TRACEPARENT_KEY)
-    elif is_instrumentation_enabled():
+    elif _is_instrumentation_enabled():
         traceparent = RunTelemetry.traceparent_from_span(span=trace.get_current_span())
     else:
         traceparent = None

{prefect_client-3.2.10 → prefect_client-3.2.12}/src/prefect/deployments/runner.py

@@ -47,6 +47,7 @@ from pydantic import (
 from rich.console import Console
 from rich.progress import Progress, SpinnerColumn, TextColumn, track
 from rich.table import Table
+from typing_extensions import Self
 
 from prefect._experimental.sla.objects import SlaTypes
 from prefect._internal.compatibility.async_dispatch import async_dispatch
@@ -84,6 +85,7 @@ from prefect.settings import (
 )
 from prefect.types import ListOfNonEmptyStrings
 from prefect.types.entrypoint import EntrypointType
+from prefect.utilities.annotations import freeze
 from prefect.utilities.asyncutils import run_coro_as_sync, sync_compatible
 from prefect.utilities.callables import ParameterSchema, parameter_schema
 from prefect.utilities.collections import get_from_dict, isiterable
@@ -135,7 +137,9 @@ class RunnerDeployment(BaseModel):
         _sla: (Experimental) SLA configuration for the deployment. May be removed or modified at any time. Currently only supported on Prefect Cloud.
     """
 
-    model_config: ClassVar[ConfigDict] = ConfigDict(arbitrary_types_allowed=True)
+    model_config: ClassVar[ConfigDict] = ConfigDict(
+        arbitrary_types_allowed=True, validate_assignment=True
+    )
 
     name: str = Field(..., description="The name of the deployment.")
     flow_name: Optional[str] = Field(
@@ -250,14 +254,30 @@ class RunnerDeployment(BaseModel):
                 trigger.name = f"{self.name}__automation_{i}"
         return self
 
+    @model_validator(mode="after")
+    def validate_deployment_parameters(self) -> Self:
+        """Update the parameter schema to mark frozen parameters as readonly."""
+        if not self.parameters:
+            return self
+
+        for key, value in self.parameters.items():
+            if isinstance(value, freeze):
+                raw_value = value.unfreeze()
+                if key in self._parameter_openapi_schema.properties:
+                    self._parameter_openapi_schema.properties[key]["readOnly"] = True
+                    self._parameter_openapi_schema.properties[key]["enum"] = [raw_value]
+                    self.parameters[key] = raw_value
+
+        return self
+
     @model_validator(mode="before")
     @classmethod
-    def reconcile_paused(cls, values):
+    def reconcile_paused(cls, values: dict[str, Any]) -> dict[str, Any]:
         return reconcile_paused_deployment(values)
 
     @model_validator(mode="before")
     @classmethod
-    def reconcile_schedules(cls, values):
+    def reconcile_schedules(cls, values: dict[str, Any]) -> dict[str, Any]:
         return reconcile_schedules_runner(values)
 
     async def _create(
@@ -367,7 +387,8 @@ class RunnerDeployment(BaseModel):
         await client.update_deployment(
             deployment_id,
             deployment=DeploymentUpdate(
-                **update_payload, parameter_openapi_schema=parameter_openapi_schema
+                **update_payload,
+                parameter_openapi_schema=parameter_openapi_schema,
             ),
         )
 

{prefect_client-3.2.10 → prefect_client-3.2.12}/src/prefect/flows.py

@@ -1490,10 +1490,10 @@ class Flow(Generic[P, R]):
             _sla=_sla,
         )
 
-        if TYPE_CHECKING:
-            assert inspect.isawaitable(to_deployment_coro)
-
-        deployment = await to_deployment_coro
+        if inspect.isawaitable(to_deployment_coro):
+            deployment = await to_deployment_coro
+        else:
+            deployment = to_deployment_coro
 
         from prefect.deployments.runner import deploy
 

{prefect_client-3.2.10 → prefect_client-3.2.12}/src/prefect/server/api/flow_runs.py

@@ -313,7 +313,7 @@ async def read_flow_run(
         )
 
 
-@router.get("/{id}/graph")
+@router.get("/{id}/graph", tags=["Flow Run Graph"])
 async def read_flow_run_graph_v1(
     flow_run_id: UUID = Path(..., description="The flow run id", alias="id"),
     db: PrefectDBInterface = Depends(provide_database_interface),
@@ -327,7 +327,7 @@ async def read_flow_run_graph_v1(
         )
 
 
-@router.get("/{id:uuid}/graph-v2")
+@router.get("/{id:uuid}/graph-v2", tags=["Flow Run Graph"])
 async def read_flow_run_graph_v2(
     flow_run_id: UUID = Path(..., description="The flow run id", alias="id"),
     since: DateTime = Query(

{prefect_client-3.2.10 → prefect_client-3.2.12}/src/prefect/server/api/ui/schemas.py

@@ -29,6 +29,7 @@ async def validate_obj(
         ...,
         embed=True,
         alias="schema",
+        validation_alias="schema",
         json_schema_extra={"additionalProperties": True},
     ),
     values: dict[str, Any] = Body(

{prefect_client-3.2.10 → prefect_client-3.2.12}/src/prefect/utilities/annotations.py

@@ -2,6 +2,8 @@ import warnings
 from operator import itemgetter
 from typing import Any, cast
 
+from pydantic import GetCoreSchemaHandler
+from pydantic_core import core_schema, to_json
 from typing_extensions import Self, TypeVar
 
 T = TypeVar("T", infer_variance=True)
@@ -108,3 +110,44 @@ class NotSet:
     """
     Singleton to distinguish `None` from a value that is not provided by the user.
     """
+
+
+class freeze(BaseAnnotation[T]):
+    """
+    Wrapper for parameters in deployments.
+
+    Indicates that this parameter should be frozen in the UI and not editable
+    when creating flow runs from this deployment.
+
+    Example:
+    ```python
+    @flow
+    def my_flow(customer_id: str):
+        # flow logic
+
+    deployment = my_flow.deploy(parameters={"customer_id": freeze("customer123")})
+    ```
+    """
+
+    def __new__(cls, value: T) -> Self:
+        try:
+            to_json(value)
+        except Exception:
+            raise ValueError("Value must be JSON serializable")
+        return super().__new__(cls, value)
+
+    def unfreeze(self) -> T:
+        """Return the unwrapped value."""
+        return self.unwrap()
+
+    @classmethod
+    def __get_pydantic_core_schema__(
+        cls, source: type[Any], handler: GetCoreSchemaHandler
+    ) -> core_schema.CoreSchema:
+        return core_schema.no_info_after_validator_function(
+            cls,  # Use the class itself as the validator
+            core_schema.any_schema(),
+            serialization=core_schema.plain_serializer_function_ser_schema(
+                lambda x: x.unfreeze()  # Serialize by unwrapping the value
+            ),
+        )

prefect_client-3.2.10/src/prefect/_build_info.py

@@ -1,5 +0,0 @@
-# Generated by versioningit
-__version__ = "3.2.10"
-__build_date__ = "2025-03-05 18:07:05.465362+00:00"
-__git_commit__ = "3a61a72641ef7ba00d874f725351b1017927a247"
-__dirty__ = False