prefect-client 3.6.3.dev2__tar.gz → 3.6.3.dev3__tar.gz

{prefect_client-3.6.3.dev2 → prefect_client-3.6.3.dev3}/PKG-INFO

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

prefect_client-3.6.3.dev3/src/prefect/_build_info.py

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.6.3.dev3"
+__build_date__ = "2025-11-19 08:09:12.579019+00:00"
+__git_commit__ = "a3db33d4f9ee7a665430ae6017c649d057139bd3"
+__dirty__ = False

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

@@ -25,12 +25,6 @@ router: PrefectRouter = PrefectRouter(
     prefix="/v2/concurrency_limits", tags=["Concurrency Limits V2"]
 )
 
-# Max retry delay (in seconds) for non-tag concurrency limits when slot occupancy is high.
-# Tag-based limits use the legacy tag_concurrency_slot_wait_seconds setting (default 30s)
-# to restore V1 behavior. Non-tag limits use this higher value to allow more uniform queues
-# to check more frequently while still preventing excessive delays from long-running tasks.
-DEFAULT_SLOT_WAIT_SECONDS = 30.0
-
 
 @router.post("/", status_code=status.HTTP_201_CREATED)
 async def create_concurrency_limit_v2(
@@ -234,8 +228,8 @@ async def _generate_concurrency_locked_response(
 
     - Tag-based limits (name starts with "tag:"): Capped at tag_concurrency_slot_wait_seconds
       (default 30s) to restore V1 behavior that users relied on
-    - Other limits: Capped at DEFAULT_SLOT_WAIT_SECONDS (30s) to allow more uniform queues
-      while still preventing astronomical delays
+    - Other limits: Capped at maximum_concurrency_slot_wait_seconds (default 30s) to allow
+      more uniform queues while still preventing astronomical delays
 
     Low average occupancies are always respected (e.g., 2s stays 2s, not forced higher).
     Limits with slot decay use the decay rate directly without capping.
@@ -265,7 +259,7 @@ async def _generate_concurrency_locked_response(
         max_wait = (
             settings.server.tasks.tag_concurrency_slot_wait_seconds
             if blocking_limit.name.startswith("tag:")
-            else DEFAULT_SLOT_WAIT_SECONDS
+            else settings.server.concurrency.maximum_concurrency_slot_wait_seconds
         )
         wait_time_per_slot = min(blocking_limit.avg_slot_occupancy_seconds, max_wait)
     else:

{prefect_client-3.6.3.dev2 → prefect_client-3.6.3.dev3}/src/prefect/settings/models/server/concurrency.py

@@ -22,3 +22,9 @@ class ServerConcurrencySettings(PrefectBaseSettings):
         le=3600.0,  # Maximum 1 hour
         description="Initial duration for deployment concurrency lease in seconds.",
     )
+
+    maximum_concurrency_slot_wait_seconds: float = Field(
+        default=30,
+        ge=0,
+        description="The maximum number of seconds to wait before retrying when a concurrency slot cannot be acquired.",
+    )

{prefect_client-3.6.3.dev2 → prefect_client-3.6.3.dev3}/src/prefect/tasks.py

@@ -1074,18 +1074,8 @@ class Task(Generic[P, R]):
 
             return task_run
 
-    @overload
-    def __call__(
-        self: "Task[P, NoReturn]",
-        *args: P.args,
-        return_state: Literal[False] = False,
-        wait_for: Optional[OneOrManyFutureOrResult[Any]] = None,
-        **kwargs: P.kwargs,
-    ) -> None:
-        # `NoReturn` matches if a type can't be inferred for the function which stops a
-        # sync function from matching the `Coroutine` overload
-        ...
-
+    # PRIORITY OVERLOADS: Clean ParamSpec signatures for normal usage (no return_state/wait_for)
+    # These preserve full parameter type checking when users call tasks normally
     @overload
     def __call__(
         self: "Task[P, Coroutine[Any, Any, R]]",
@@ -1100,56 +1090,80 @@ class Task(Generic[P, R]):
         **kwargs: P.kwargs,
     ) -> R: ...
 
-    # Keyword parameters `return_state` and `wait_for` aren't allowed after the
-    # ParamSpec `*args` parameter, so we lose return type typing when either of
-    # those are provided.
-    # TODO: Find a way to expose this functionality without losing type information
-
-    # NOTE: return_state=False overloads must come before return_state=True
-    # When pyright can't match argument types, it falls back to these overloads with *args
-    # and picks the first match. We want the default (False) behavior.
     @overload
     def __call__(
-        self: "Task[P, Coroutine[Any, Any, R]]",
+        self: "Task[P, NoReturn]",
         *args: P.args,
-        return_state: Literal[False] = False,
-        wait_for: Optional[OneOrManyFutureOrResult[Any]] = None,
         **kwargs: P.kwargs,
+    ) -> None:
+        # `NoReturn` matches if a type can't be inferred for the function which stops a
+        # sync function from matching the `Coroutine` overload
+        ...
+
+    # SECONDARY OVERLOADS: With return_state/wait_for using Any
+    # When return_state or wait_for are used, we can't preserve ParamSpec semantics,
+    # so we use Any for parameters. This is an acceptable tradeoff since these
+    # are advanced use cases.
+    @overload
+    def __call__(
+        self: "Task[..., Coroutine[Any, Any, R]]",
+        *args: Any,
+        return_state: Literal[False],
+        wait_for: Optional[OneOrManyFutureOrResult[Any]] = None,
+        **kwargs: Any,
     ) -> Coroutine[Any, Any, R]: ...
 
     @overload
     def __call__(
-        self: "Task[P, Coroutine[Any, Any, R]]",
-        *args: P.args,
-        return_state: Literal[True] = True,
+        self: "Task[..., Coroutine[Any, Any, R]]",
+        *args: Any,
+        return_state: Literal[True],
         wait_for: Optional[OneOrManyFutureOrResult[Any]] = None,
-        **kwargs: P.kwargs,
+        **kwargs: Any,
     ) -> State[R]: ...
 
     @overload
     def __call__(
-        self: "Task[P, R]",
-        *args: P.args,
-        return_state: Literal[False] = False,
+        self: "Task[..., R]",
+        *args: Any,
+        return_state: Literal[False],
         wait_for: Optional[OneOrManyFutureOrResult[Any]] = None,
-        **kwargs: P.kwargs,
+        **kwargs: Any,
     ) -> R: ...
 
     @overload
     def __call__(
-        self: "Task[P, R]",
-        *args: P.args,
-        return_state: Literal[True] = True,
+        self: "Task[..., R]",
+        *args: Any,
+        return_state: Literal[True],
         wait_for: Optional[OneOrManyFutureOrResult[Any]] = None,
-        **kwargs: P.kwargs,
+        **kwargs: Any,
     ) -> State[R]: ...
 
+    @overload
     def __call__(
-        self: "Union[Task[P, R], Task[P, NoReturn]]",
-        *args: P.args,
+        self: "Task[..., Coroutine[Any, Any, R]]",
+        *args: Any,
+        wait_for: OneOrManyFutureOrResult[Any],
+        return_state: Literal[False] = False,
+        **kwargs: Any,
+    ) -> Coroutine[Any, Any, R]: ...
+
+    @overload
+    def __call__(
+        self: "Task[..., R]",
+        *args: Any,
+        wait_for: OneOrManyFutureOrResult[Any],
+        return_state: Literal[False] = False,
+        **kwargs: Any,
+    ) -> R: ...
+
+    def __call__(
+        self: "Union[Task[..., R], Task[..., NoReturn]]",
+        *args: Any,
         return_state: bool = False,
         wait_for: Optional[OneOrManyFutureOrResult[Any]] = None,
-        **kwargs: P.kwargs,
+        **kwargs: Any,
     ) -> Union[R, State[R], None]:
         """
         Run the task and return the result. If `return_state` is True returns

prefect_client-3.6.3.dev2/src/prefect/_build_info.py

@@ -1,5 +0,0 @@
-# Generated by versioningit
-__version__ = "3.6.3.dev2"
-__build_date__ = "2025-11-18 08:09:13.019207+00:00"
-__git_commit__ = "6a68cfff794e700d85800476c38a733eb61d1aae"
-__dirty__ = False