prefect-client 3.3.8.dev2__tar.gz → 3.3.8.dev4__tar.gz

{prefect_client-3.3.8.dev2 → prefect_client-3.3.8.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prefect-client
-Version: 3.3.8.dev2
+Version: 3.3.8.dev4
 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.8.dev4/src/prefect/_build_info.py

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.3.8.dev4"
+__build_date__ = "2025-05-02 08:08:48.097027+00:00"
+__git_commit__ = "5d6e0c61baba4513aa9744c942297d645fc4db2c"
+__dirty__ = False

{prefect_client-3.3.8.dev2 → prefect_client-3.3.8.dev4}/src/prefect/_internal/schemas/validators.py

@@ -485,25 +485,6 @@ def validate_not_negative(v: Optional[float]) -> Optional[float]:
     return v
 
 
-@overload
-def validate_message_template_variables(v: str) -> str: ...
-
-
-@overload
-def validate_message_template_variables(v: None) -> None: ...
-
-
-def validate_message_template_variables(v: Optional[str]) -> Optional[str]:
-    from prefect.client.schemas.objects import FLOW_RUN_NOTIFICATION_TEMPLATE_KWARGS
-
-    if v is not None:
-        try:
-            v.format(**{k: "test" for k in FLOW_RUN_NOTIFICATION_TEMPLATE_KWARGS})
-        except KeyError as exc:
-            raise ValueError(f"Invalid template variable provided: '{exc.args[0]}'")
-    return v
-
-
 def validate_default_queue_id_not_none(v: Optional[UUID]) -> UUID:
     if v is None:
         raise ValueError(

{prefect_client-3.3.8.dev2 → prefect_client-3.3.8.dev4}/src/prefect/blocks/core.py

@@ -374,15 +374,15 @@ class Block(BaseModel, ABC):
                     for field_name in type(self).model_fields
                 }
             )
-        if extra_fields := {
+        extra_fields = {
             "block_type_slug": self.get_block_type_slug(),
             "_block_document_id": self._block_document_id,
             "_block_document_name": self._block_document_name,
             "_is_anonymous": self._is_anonymous,
-        }:
-            jsonable_self |= {
-                key: value for key, value in extra_fields.items() if value is not None
-            }
+        }
+        jsonable_self |= {
+            key: value for key, value in extra_fields.items() if value is not None
+        }
         return jsonable_self
 
     @classmethod

{prefect_client-3.3.8.dev2 → prefect_client-3.3.8.dev4}/src/prefect/client/orchestration/__init__.py

@@ -87,8 +87,6 @@ import prefect.states
 from prefect.client.constants import SERVER_API_VERSION
 from prefect.client.schemas import FlowRun, OrchestrationResult, TaskRun
 from prefect.client.schemas.actions import (
-    FlowRunNotificationPolicyCreate,
-    FlowRunNotificationPolicyUpdate,
     TaskRunCreate,
     TaskRunUpdate,
     WorkQueueCreate,
@@ -98,14 +96,12 @@ from prefect.client.schemas.filters import (
     DeploymentFilter,
     FlowFilter,
     FlowRunFilter,
-    FlowRunNotificationPolicyFilter,
     TaskRunFilter,
     WorkQueueFilter,
     WorkQueueFilterName,
 )
 from prefect.client.schemas.objects import (
     Constant,
-    FlowRunNotificationPolicy,
     Parameter,
     TaskRunPolicy,
     TaskRunResult,
@@ -962,151 +958,6 @@ class PrefectClient(
             response.json()
         )
 
-    async def create_flow_run_notification_policy(
-        self,
-        block_document_id: UUID,
-        is_active: bool = True,
-        tags: Optional[list[str]] = None,
-        state_names: Optional[list[str]] = None,
-        message_template: Optional[str] = None,
-    ) -> UUID:
-        """
-        Create a notification policy for flow runs
-
-        Args:
-            block_document_id: The block document UUID
-            is_active: Whether the notification policy is active
-            tags: List of flow tags
-            state_names: List of state names
-            message_template: Notification message template
-        """
-        if tags is None:
-            tags = []
-        if state_names is None:
-            state_names = []
-
-        policy = FlowRunNotificationPolicyCreate(
-            block_document_id=block_document_id,
-            is_active=is_active,
-            tags=tags,
-            state_names=state_names,
-            message_template=message_template,
-        )
-        response = await self._client.post(
-            "/flow_run_notification_policies/",
-            json=policy.model_dump(mode="json"),
-        )
-
-        policy_id = response.json().get("id")
-        if not policy_id:
-            raise httpx.RequestError(f"Malformed response: {response}")
-
-        return UUID(policy_id)
-
-    async def delete_flow_run_notification_policy(
-        self,
-        id: UUID,
-    ) -> None:
-        """
-        Delete a flow run notification policy by id.
-
-        Args:
-            id: UUID of the flow run notification policy to delete.
-        Raises:
-            prefect.exceptions.ObjectNotFound: If request returns 404
-            httpx.RequestError: If requests fails
-        """
-        try:
-            await self._client.delete(f"/flow_run_notification_policies/{id}")
-        except httpx.HTTPStatusError as e:
-            if e.response.status_code == status.HTTP_404_NOT_FOUND:
-                raise prefect.exceptions.ObjectNotFound(http_exc=e) from e
-            else:
-                raise
-
-    async def update_flow_run_notification_policy(
-        self,
-        id: UUID,
-        block_document_id: Optional[UUID] = None,
-        is_active: Optional[bool] = None,
-        tags: Optional[list[str]] = None,
-        state_names: Optional[list[str]] = None,
-        message_template: Optional[str] = None,
-    ) -> None:
-        """
-        Update a notification policy for flow runs
-
-        Args:
-            id: UUID of the notification policy
-            block_document_id: The block document UUID
-            is_active: Whether the notification policy is active
-            tags: List of flow tags
-            state_names: List of state names
-            message_template: Notification message template
-        Raises:
-            prefect.exceptions.ObjectNotFound: If request returns 404
-            httpx.RequestError: If requests fails
-        """
-        params: dict[str, Any] = {}
-        if block_document_id is not None:
-            params["block_document_id"] = block_document_id
-        if is_active is not None:
-            params["is_active"] = is_active
-        if tags is not None:
-            params["tags"] = tags
-        if state_names is not None:
-            params["state_names"] = state_names
-        if message_template is not None:
-            params["message_template"] = message_template
-
-        policy = FlowRunNotificationPolicyUpdate(**params)
-
-        try:
-            await self._client.patch(
-                f"/flow_run_notification_policies/{id}",
-                json=policy.model_dump(mode="json", exclude_unset=True),
-            )
-        except httpx.HTTPStatusError as e:
-            if e.response.status_code == status.HTTP_404_NOT_FOUND:
-                raise prefect.exceptions.ObjectNotFound(http_exc=e) from e
-            else:
-                raise
-
-    async def read_flow_run_notification_policies(
-        self,
-        flow_run_notification_policy_filter: FlowRunNotificationPolicyFilter,
-        limit: Optional[int] = None,
-        offset: int = 0,
-    ) -> list[FlowRunNotificationPolicy]:
-        """
-        Query the Prefect API for flow run notification policies. Only policies matching all criteria will
-        be returned.
-
-        Args:
-            flow_run_notification_policy_filter: filter criteria for notification policies
-            limit: a limit for the notification policies query
-            offset: an offset for the notification policies query
-
-        Returns:
-            a list of FlowRunNotificationPolicy model representations
-                of the notification policies
-        """
-        body: dict[str, Any] = {
-            "flow_run_notification_policy_filter": (
-                flow_run_notification_policy_filter.model_dump(mode="json")
-                if flow_run_notification_policy_filter
-                else None
-            ),
-            "limit": limit,
-            "offset": offset,
-        }
-        response = await self._client.post(
-            "/flow_run_notification_policies/filter", json=body
-        )
-        return pydantic.TypeAdapter(list[FlowRunNotificationPolicy]).validate_python(
-            response.json()
-        )
-
     async def read_work_queues(
         self,
         work_pool_name: Optional[str] = None,

{prefect_client-3.3.8.dev2 → prefect_client-3.3.8.dev4}/src/prefect/client/orchestration/routes.py

@@ -63,9 +63,6 @@ ServerRoutes = Literal[
     "/events/count-by/{countable}",
     "/events/filter",
     "/events/filter/next",
-    "/flow_run_notification_policies/",
-    "/flow_run_notification_policies/{id}",
-    "/flow_run_notification_policies/filter",
     "/flow_run_states/",
     "/flow_run_states/{id}",
     "/flow_runs/",

{prefect_client-3.3.8.dev2 → prefect_client-3.3.8.dev4}/src/prefect/client/schemas/actions.py

@@ -15,7 +15,6 @@ from prefect._internal.schemas.validators import (
     validate_artifact_key,
     validate_block_document_name,
     validate_block_type_slug,
-    validate_message_template_variables,
     validate_name_present_on_nonanonymous_blocks,
     validate_schedule_max_scheduled_runs,
     validate_variable_name,
@@ -46,7 +45,7 @@ from prefect.types import (
     PositiveInteger,
     StrictVariableValue,
 )
-from prefect.utilities.collections import listrepr, visit_collection
+from prefect.utilities.collections import visit_collection
 from prefect.utilities.pydantic import get_class_fields_only
 
 if TYPE_CHECKING:
@@ -812,51 +811,6 @@ class WorkQueueUpdate(ActionBaseModel):
     )
 
 
-class FlowRunNotificationPolicyCreate(ActionBaseModel):
-    """Data used by the Prefect REST API to create a flow run notification policy."""
-
-    is_active: bool = Field(
-        default=True, description="Whether the policy is currently active"
-    )
-    state_names: list[str] = Field(
-        default=..., description="The flow run states that trigger notifications"
-    )
-    tags: list[str] = Field(
-        default=...,
-        description="The flow run tags that trigger notifications (set [] to disable)",
-    )
-    block_document_id: UUID = Field(
-        default=..., description="The block document ID used for sending notifications"
-    )
-    message_template: Optional[str] = Field(
-        default=None,
-        description=(
-            "A templatable notification message. Use {braces} to add variables."
-            " Valid variables include:"
-            f" {listrepr(sorted(objects.FLOW_RUN_NOTIFICATION_TEMPLATE_KWARGS), sep=', ')}"
-        ),
-        examples=[
-            "Flow run {flow_run_name} with id {flow_run_id} entered state"
-            " {flow_run_state_name}."
-        ],
-    )
-
-    @field_validator("message_template")
-    @classmethod
-    def validate_message_template_variables(cls, v: Optional[str]) -> Optional[str]:
-        return validate_message_template_variables(v)
-
-
-class FlowRunNotificationPolicyUpdate(ActionBaseModel):
-    """Data used by the Prefect REST API to update a flow run notification policy."""
-
-    is_active: Optional[bool] = Field(default=None)
-    state_names: Optional[list[str]] = Field(default=None)
-    tags: Optional[list[str]] = Field(default=None)
-    block_document_id: Optional[UUID] = Field(default=None)
-    message_template: Optional[str] = Field(default=None)
-
-
 class ArtifactCreate(ActionBaseModel):
     """Data used by the Prefect REST API to create an artifact."""
 

{prefect_client-3.3.8.dev2 → prefect_client-3.3.8.dev4}/src/prefect/client/schemas/filters.py

@@ -799,26 +799,6 @@ class BlockDocumentFilter(PrefectBaseModel, OperatorMixin):
     )
 
 
-class FlowRunNotificationPolicyFilterIsActive(PrefectBaseModel):
-    """Filter by `FlowRunNotificationPolicy.is_active`."""
-
-    eq_: Optional[bool] = Field(
-        default=None,
-        description=(
-            "Filter notification policies for only those that are or are not active."
-        ),
-    )
-
-
-class FlowRunNotificationPolicyFilter(PrefectBaseModel):
-    """Filter FlowRunNotificationPolicies."""
-
-    is_active: Optional[FlowRunNotificationPolicyFilterIsActive] = Field(
-        default=FlowRunNotificationPolicyFilterIsActive(eq_=False),
-        description="Filter criteria for `FlowRunNotificationPolicy.is_active`. ",
-    )
-
-
 class WorkQueueFilterId(PrefectBaseModel):
     """Filter by `WorkQueue.id`."""
 

{prefect_client-3.3.8.dev2 → prefect_client-3.3.8.dev4}/src/prefect/client/schemas/objects.py

@@ -45,7 +45,6 @@ from prefect._internal.schemas.validators import (
     validate_block_document_name,
     validate_default_queue_id_not_none,
     validate_max_metadata_length,
-    validate_message_template_variables,
     validate_name_present_on_nonanonymous_blocks,
     validate_not_negative,
     validate_parent_and_ref_diff,
@@ -63,7 +62,7 @@ from prefect.types import (
 )
 from prefect.types._datetime import DateTime, now
 from prefect.utilities.asyncutils import run_coro_as_sync
-from prefect.utilities.collections import AutoEnum, listrepr, visit_collection
+from prefect.utilities.collections import AutoEnum, visit_collection
 from prefect.utilities.names import generate_slug
 from prefect.utilities.pydantic import handle_secret_render
 
@@ -72,19 +71,6 @@ R = TypeVar("R", default=Any)
 
 DEFAULT_BLOCK_SCHEMA_VERSION: Literal["non-versioned"] = "non-versioned"
 DEFAULT_AGENT_WORK_POOL_NAME: Literal["default-agent-pool"] = "default-agent-pool"
-FLOW_RUN_NOTIFICATION_TEMPLATE_KWARGS: list[str] = [
-    "flow_run_notification_policy_id",
-    "flow_id",
-    "flow_name",
-    "flow_run_url",
-    "flow_run_id",
-    "flow_run_name",
-    "flow_run_parameters",
-    "flow_run_state_type",
-    "flow_run_state_name",
-    "flow_run_state_timestamp",
-    "flow_run_state_message",
-]
 
 
 class StateType(AutoEnum):
@@ -1446,41 +1432,6 @@ class WorkQueueStatusDetail(PrefectBaseModel):
     )
 
 
-class FlowRunNotificationPolicy(ObjectBaseModel):
-    """An ORM representation of a flow run notification."""
-
-    is_active: bool = Field(
-        default=True, description="Whether the policy is currently active"
-    )
-    state_names: list[str] = Field(
-        default=..., description="The flow run states that trigger notifications"
-    )
-    tags: list[str] = Field(
-        default=...,
-        description="The flow run tags that trigger notifications (set [] to disable)",
-    )
-    block_document_id: UUID = Field(
-        default=..., description="The block document ID used for sending notifications"
-    )
-    message_template: Optional[str] = Field(
-        default=None,
-        description=(
-            "A templatable notification message. Use {braces} to add variables."
-            " Valid variables include:"
-            f" {listrepr(sorted(FLOW_RUN_NOTIFICATION_TEMPLATE_KWARGS), sep=', ')}"
-        ),
-        examples=[
-            "Flow run {flow_run_name} with id {flow_run_id} entered state"
-            " {flow_run_state_name}."
-        ],
-    )
-
-    @field_validator("message_template")
-    @classmethod
-    def validate_message_template_variables(cls, v: Optional[str]) -> Optional[str]:
-        return validate_message_template_variables(v)
-
-
 class Agent(ObjectBaseModel):
     """An ORM representation of an agent"""
 
@@ -1502,8 +1453,6 @@ class Agent(ObjectBaseModel):
 class WorkPoolStorageConfiguration(PrefectBaseModel):
     """A work pool storage configuration"""
 
-    model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
-
     bundle_upload_step: Optional[dict[str, Any]] = Field(
         default=None, description="The bundle upload step for the work pool."
     )

{prefect_client-3.3.8.dev2 → prefect_client-3.3.8.dev4}/src/prefect/deployments/steps/pull.py

@@ -50,6 +50,7 @@ async def _pull_git_repository_with_retries(repo: GitRepository):
 async def agit_clone(
     repository: str,
     branch: Optional[str] = None,
+    commit_sha: Optional[str] = None,
     include_submodules: bool = False,
     access_token: Optional[str] = None,
     credentials: Optional["Block"] = None,
@@ -61,6 +62,7 @@ async def agit_clone(
     Args:
         repository: the URL of the repository to clone
         branch: the branch to clone; if not provided, the default branch will be used
+        commit_sha: the commit SHA to clone; if not provided, the default branch will be used
         include_submodules (bool): whether to include git submodules when cloning the repository
         access_token: an access token to use for cloning the repository; if not provided
             the repository will be cloned using the default git credentials
@@ -84,6 +86,7 @@ async def agit_clone(
         url=repository,
         credentials=_credentials,
         branch=branch,
+        commit_sha=commit_sha,
         include_submodules=include_submodules,
         directories=directories,
     )
@@ -97,6 +100,7 @@ async def agit_clone(
 def git_clone(
     repository: str,
     branch: Optional[str] = None,
+    commit_sha: Optional[str] = None,
     include_submodules: bool = False,
     access_token: Optional[str] = None,
     credentials: Optional["Block"] = None,
@@ -108,6 +112,7 @@ def git_clone(
     Args:
         repository: the URL of the repository to clone
         branch: the branch to clone; if not provided, the default branch will be used
+        commit_sha: the commit SHA to clone; if not provided, the default branch will be used
         include_submodules (bool): whether to include git submodules when cloning the repository
         access_token: an access token to use for cloning the repository; if not provided
             the repository will be cloned using the default git credentials
@@ -191,6 +196,7 @@ def git_clone(
         url=repository,
         credentials=_credentials,
         branch=branch,
+        commit_sha=commit_sha,
         include_submodules=include_submodules,
         directories=directories,
     )

{prefect_client-3.3.8.dev2 → prefect_client-3.3.8.dev4}/src/prefect/futures.py

@@ -14,7 +14,7 @@ from typing_extensions import NamedTuple, Self, TypeVar
 from prefect._waiters import FlowRunWaiter
 from prefect.client.orchestration import get_client
 from prefect.exceptions import ObjectNotFound
-from prefect.logging.loggers import get_logger, get_run_logger
+from prefect.logging.loggers import get_logger
 from prefect.states import Pending, State
 from prefect.task_runs import TaskRunWaiter
 from prefect.utilities.annotations import quote
@@ -224,19 +224,6 @@ class PrefectConcurrentFuture(PrefectWrappedFuture[R, concurrent.futures.Future[
         )
         return _result
 
-    def __del__(self) -> None:
-        if self._final_state or self._wrapped_future.done():
-            return
-        try:
-            local_logger = get_run_logger()
-        except Exception:
-            local_logger = logger
-        local_logger.warning(
-            "A future was garbage collected before it resolved."
-            " Please call `.wait()` or `.result()` on futures to ensure they resolve."
-            "\nSee https://docs.prefect.io/latest/develop/task-runners for more details.",
-        )
-
 
 class PrefectDistributedFuture(PrefectTaskRunFuture[R]):
     """