prefect 3.4.18.dev2__tar.gz → 3.4.18.dev3__tar.gz

{prefect-3.4.18.dev2 → prefect-3.4.18.dev3}/PKG-INFO

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

prefect-3.4.18.dev3/src/prefect/_build_info.py

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.4.18.dev3"
+__build_date__ = "2025-09-10 08:09:37.770853+00:00"
+__git_commit__ = "40d354fe1837259ba451a2e8cd345aa5f58eb864"
+__dirty__ = False

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

@@ -269,6 +269,7 @@ class ConcurrencyLimitClient(BaseClient):
         slots: int,
         mode: Literal["concurrency", "rate_limit"],
         lease_duration: float,
+        holder: dict[str, Any] | None = None,
     ) -> "Response":
         """
         Increment concurrency slots for the specified limits with a lease.
@@ -278,16 +279,21 @@ class ConcurrencyLimitClient(BaseClient):
             slots: The number of concurrency slots to occupy.
             mode: The mode of the concurrency limits.
             lease_duration: The duration of the lease in seconds.
+            holder: Optional holder information for tracking who holds the slots.
         """
+        body: dict[str, Any] = {
+            "names": names,
+            "slots": slots,
+            "mode": mode,
+            "lease_duration": lease_duration,
+        }
+        if holder is not None:
+            body["holder"] = holder
+
         return self.request(
             "POST",
             "/v2/concurrency_limits/increment-with-lease",
-            json={
-                "names": names,
-                "slots": slots,
-                "mode": mode,
-                "lease_duration": lease_duration,
-            },
+            json=body,
         )
 
     def renew_concurrency_lease(
@@ -715,6 +721,7 @@ class ConcurrencyLimitAsyncClient(BaseAsyncClient):
         slots: int,
         mode: Literal["concurrency", "rate_limit"],
         lease_duration: float,
+        holder: dict[str, Any] | None = None,
     ) -> "Response":
         """
         Increment concurrency slots for the specified limits with a lease.
@@ -724,16 +731,21 @@ class ConcurrencyLimitAsyncClient(BaseAsyncClient):
             slots: The number of concurrency slots to occupy.
             mode: The mode of the concurrency limits.
             lease_duration: The duration of the lease in seconds.
+            holder: Optional holder information for tracking who holds the slots.
         """
+        body: dict[str, Any] = {
+            "names": names,
+            "slots": slots,
+            "mode": mode,
+            "lease_duration": lease_duration,
+        }
+        if holder is not None:
+            body["holder"] = holder
+
         return await self.request(
             "POST",
             "/v2/concurrency_limits/increment-with-lease",
-            json={
-                "names": names,
-                "slots": slots,
-                "mode": mode,
-                "lease_duration": lease_duration,
-            },
+            json=body,
         )
 
     async def renew_concurrency_lease(

{prefect-3.4.18.dev2 → prefect-3.4.18.dev3}/src/prefect/server/api/concurrency_limits_v2.py

@@ -16,6 +16,7 @@ from prefect.server.database import PrefectDBInterface, provide_database_interfa
 from prefect.server.schemas import actions
 from prefect.server.utilities.schemas import PrefectBaseModel
 from prefect.server.utilities.server import PrefectRouter
+from prefect.types._concurrency import ConcurrencyLeaseHolder
 
 router: PrefectRouter = PrefectRouter(
     prefix="/v2/concurrency_limits", tags=["Concurrency Limits V2"]
@@ -294,6 +295,10 @@ async def bulk_increment_active_slots_with_lease(
         le=60 * 60 * 24,  # 1 day
         description="The duration of the lease in seconds.",
     ),
+    holder: Optional[ConcurrencyLeaseHolder] = Body(
+        None,
+        description="The holder of the lease with type (flow_run, task_run, or deployment) and id.",
+    ),
     db: PrefectDBInterface = Depends(provide_database_interface),
 ) -> ConcurrencyLimitWithLeaseResponse:
     async with db.session_context(begin_transaction=True) as session:
@@ -309,7 +314,10 @@ async def bulk_increment_active_slots_with_lease(
         lease = await lease_storage.create_lease(
             resource_ids=[limit.id for limit in acquired_limits],
             ttl=timedelta(seconds=lease_duration),
-            metadata=ConcurrencyLimitLeaseMetadata(slots=slots),
+            metadata=ConcurrencyLimitLeaseMetadata(
+                slots=slots,
+                holder=holder.model_dump() if holder else None,
+            ),
         )
         return ConcurrencyLimitWithLeaseResponse(
             lease_id=lease.id,

{prefect-3.4.18.dev2 → prefect-3.4.18.dev3}/src/prefect/server/concurrency/lease_storage/__init__.py

@@ -1,23 +1,30 @@
 from __future__ import annotations
 
-from dataclasses import dataclass
 from datetime import timedelta
 import importlib
-from typing import Protocol, runtime_checkable
+from typing import ClassVar, Optional, Protocol, runtime_checkable
 from uuid import UUID
 
+from pydantic import BaseModel, ConfigDict
+
 from prefect.server.utilities.leasing import LeaseStorage, ResourceLease
 from prefect.settings.context import get_current_settings
 
+from prefect.types._concurrency import ConcurrencyLeaseHolder
+
 
 @runtime_checkable
 class ConcurrencyLeaseStorageModule(Protocol):
     ConcurrencyLeaseStorage: type[ConcurrencyLeaseStorage]
 
 
-@dataclass
-class ConcurrencyLimitLeaseMetadata:
+class ConcurrencyLimitLeaseMetadata(BaseModel):
+    """Model for validating concurrency limit lease metadata."""
+
+    model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
+
     slots: int
+    holder: Optional[ConcurrencyLeaseHolder] = None
 
 
 class ConcurrencyLeaseStorage(LeaseStorage[ConcurrencyLimitLeaseMetadata]):

{prefect-3.4.18.dev2 → prefect-3.4.18.dev3}/src/prefect/server/concurrency/lease_storage/filesystem.py

@@ -83,10 +83,16 @@ class ConcurrencyLeaseStorage(_ConcurrencyLeaseStorage):
     def _serialize_lease(
         self, lease: ResourceLease[ConcurrencyLimitLeaseMetadata]
     ) -> _LeaseFile:
+        metadata_dict: dict[str, Any] | None = None
+        if lease.metadata:
+            metadata_dict = {"slots": lease.metadata.slots}
+            if lease.metadata.holder is not None:
+                metadata_dict["holder"] = lease.metadata.holder.model_dump(mode="json")
+
         return {
             "id": str(lease.id),
             "resource_ids": [str(rid) for rid in lease.resource_ids],
-            "metadata": {"slots": lease.metadata.slots} if lease.metadata else None,
+            "metadata": metadata_dict,
             "expiration": lease.expiration.isoformat(),
             "created_at": lease.created_at.isoformat(),
         }
@@ -96,11 +102,11 @@ class ConcurrencyLeaseStorage(_ConcurrencyLeaseStorage):
     ) -> ResourceLease[ConcurrencyLimitLeaseMetadata]:
         lease_id = UUID(data["id"])
         resource_ids = [UUID(rid) for rid in data["resource_ids"]]
-        metadata = (
-            ConcurrencyLimitLeaseMetadata(slots=data["metadata"]["slots"])
-            if data["metadata"]
-            else None
-        )
+        metadata = None
+        if data["metadata"]:
+            metadata = ConcurrencyLimitLeaseMetadata(
+                slots=data["metadata"]["slots"], holder=data["metadata"].get("holder")
+            )
         expiration = datetime.fromisoformat(data["expiration"])
         created_at = datetime.fromisoformat(data["created_at"])
         lease = ResourceLease(

prefect-3.4.18.dev3/src/prefect/types/_concurrency.py

@@ -0,0 +1,13 @@
+from typing import ClassVar, Literal
+from uuid import UUID
+
+from pydantic import BaseModel, ConfigDict
+
+
+class ConcurrencyLeaseHolder(BaseModel):
+    """Model for validating concurrency lease holder information."""
+
+    model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
+
+    type: Literal["flow_run", "task_run", "deployment"]
+    id: UUID

prefect-3.4.18.dev2/src/prefect/_build_info.py

@@ -1,5 +0,0 @@
-# Generated by versioningit
-__version__ = "3.4.18.dev2"
-__build_date__ = "2025-09-09 08:09:26.103378+00:00"
-__git_commit__ = "07c65121162f4c6640a13fa3b68f107760cc5760"
-__dirty__ = False