prefect-client 3.4.3.dev1__tar.gz → 3.4.4__tar.gz

{prefect_client-3.4.3.dev1 → prefect_client-3.4.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prefect-client
-Version: 3.4.3.dev1
+Version: 3.4.4
 Summary: Workflow orchestration and management.
 Project-URL: Changelog, https://github.com/PrefectHQ/prefect/releases
 Project-URL: Documentation, https://docs.prefect.io
@@ -23,7 +23,7 @@ Classifier: Topic :: Software Development :: Libraries
 Requires-Python: <3.14,>=3.9
 Requires-Dist: anyio<5.0.0,>=4.4.0
 Requires-Dist: asgi-lifespan<3.0,>=1.0
-Requires-Dist: cachetools<6.0,>=5.3
+Requires-Dist: cachetools<7.0,>=5.3
 Requires-Dist: cloudpickle<4.0,>=2.0
 Requires-Dist: coolname<3.0.0,>=1.0.4
 Requires-Dist: dateparser<2.0.0,>=1.1.1

{prefect_client-3.4.3.dev1 → prefect_client-3.4.4}/pyproject.toml

@@ -27,7 +27,7 @@ dependencies = [
     # Client dependencies
     "anyio>=4.4.0,<5.0.0",
     "asgi-lifespan>=1.0,<3.0",
-    "cachetools>=5.3,<6.0",
+    "cachetools>=5.3,<7.0",
     "cloudpickle>=2.0,<4.0",
     "coolname>=1.0.4,<3.0.0",
     "dateparser>=1.1.1,<2.0.0",

prefect_client-3.4.4/src/prefect/_build_info.py

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.4.4"
+__build_date__ = "2025-05-29 21:36:59.371770+00:00"
+__git_commit__ = "0367d7aa7939aa1a26152a9db9997a75aadd6e3b"
+__dirty__ = False

{prefect_client-3.4.3.dev1 → prefect_client-3.4.4}/src/prefect/_internal/concurrency/calls.py

@@ -222,7 +222,7 @@ class Future(concurrent.futures.Future[T]):
             self._cancel_scope = None
 
 
-@dataclasses.dataclass
+@dataclasses.dataclass(eq=False)
 class Call(Generic[T]):
     """
     A deferred function call.
@@ -236,6 +236,42 @@ class Call(Generic[T]):
     timeout: Optional[float]
     runner: Optional["Portal"] = None
 
+    def __eq__(self, other: object) -> bool:
+        """this is to avoid attempts at invalid access of args/kwargs in <3.13 stemming from the
+        auto-generated __eq__ method on the dataclass.
+
+        this will no longer be required in 3.13+, see https://github.com/python/cpython/issues/128294
+        """
+        if self is other:
+            return True
+        if not isinstance(other, Call):
+            return NotImplemented
+
+        try:
+            # Attempt to access args/kwargs. If any are missing on self or other,
+            # an AttributeError will be raised by the access attempt on one of them.
+            s_args, s_kwargs = self.args, self.kwargs
+            o_args, o_kwargs = other.args, other.kwargs
+        except AttributeError:
+            # If args/kwargs are missing on self or other (and self is not other),
+            # they are considered not equal. This ensures that a Call with deleted
+            # args/kwargs compares as different from one that still has them
+            return False
+
+        # If all args/kwargs were accessible on both, proceed with full field comparison.
+        # Note: self.future == other.future will use Future's __eq__ (default is identity).
+        return (
+            (self.future == other.future)
+            and (self.fn == other.fn)
+            and (s_args == o_args)
+            and (s_kwargs == o_kwargs)
+            and (self.context == other.context)
+            and (self.timeout == other.timeout)
+            and (self.runner == other.runner)
+        )
+
+    __hash__ = None  # type: ignore
+
     @classmethod
     def new(
         cls,

{prefect_client-3.4.3.dev1 → prefect_client-3.4.4}/src/prefect/_internal/schemas/validators.py

@@ -402,6 +402,18 @@ def validate_compressionlib(value: str) -> str:
 
 
 # TODO: if we use this elsewhere we can change the error message to be more generic
+@overload
+def list_length_50_or_less(v: int) -> int: ...
+
+
+@overload
+def list_length_50_or_less(v: float) -> float: ...
+
+
+@overload
+def list_length_50_or_less(v: list[int]) -> list[int]: ...
+
+
 @overload
 def list_length_50_or_less(v: list[float]) -> list[float]: ...
 
@@ -410,7 +422,9 @@ def list_length_50_or_less(v: list[float]) -> list[float]: ...
 def list_length_50_or_less(v: None) -> None: ...
 
 
-def list_length_50_or_less(v: Optional[list[float]]) -> Optional[list[float]]:
+def list_length_50_or_less(
+    v: Optional[int | float | list[int] | list[float]],
+) -> Optional[int | float | list[int] | list[float]]:
     if isinstance(v, list) and (len(v) > 50):
         raise ValueError("Can not configure more than 50 retry delays per task.")
     return v

{prefect_client-3.4.3.dev1 → prefect_client-3.4.4}/src/prefect/client/schemas/objects.py

@@ -698,7 +698,7 @@ class TaskRunPolicy(PrefectBaseModel):
         deprecated=True,
     )
     retries: Optional[int] = Field(default=None, description="The number of retries.")
-    retry_delay: Union[None, int, list[int]] = Field(
+    retry_delay: Union[None, int, float, list[int], list[float]] = Field(
         default=None,
         description="A delay time or list of delay times between retries, in seconds.",
     )
@@ -728,8 +728,8 @@ class TaskRunPolicy(PrefectBaseModel):
     @field_validator("retry_delay")
     @classmethod
     def validate_configured_retry_delays(
-        cls, v: Optional[list[float]]
-    ) -> Optional[list[float]]:
+        cls, v: Optional[int | float | list[int] | list[float]]
+    ) -> Optional[int | float | list[int] | list[float]]:
         return list_length_50_or_less(v)
 
     @field_validator("retry_jitter_factor")

{prefect_client-3.4.3.dev1 → prefect_client-3.4.4}/src/prefect/server/api/clients.py

@@ -18,7 +18,7 @@ from prefect.server.schemas.actions import DeploymentFlowRunCreate, StateCreate
 from prefect.server.schemas.core import WorkPool
 from prefect.server.schemas.filters import VariableFilter, VariableFilterName
 from prefect.server.schemas.responses import DeploymentResponse, OrchestrationResult
-from prefect.settings import PREFECT_SERVER_API_AUTH_STRING
+from prefect.settings import get_current_settings
 from prefect.types import StrictVariableValue
 
 if TYPE_CHECKING:
@@ -39,18 +39,19 @@ class BaseClient:
         # will point it to the the currently running server instance
         api_app = create_app()
 
-        # we pull the auth string from _server_ settings because this client is run on the server
-        auth_string = PREFECT_SERVER_API_AUTH_STRING.value()
+        settings = get_current_settings()
 
-        if auth_string:
-            token = base64.b64encode(auth_string.encode("utf-8")).decode("utf-8")
-            additional_headers.setdefault("Authorization", f"Basic {token}")
+        # we pull the auth string from _server_ settings because this client is run on the server
+        if auth_string_secret := settings.server.api.auth_string:
+            if auth_string := auth_string_secret.get_secret_value():
+                token = base64.b64encode(auth_string.encode("utf-8")).decode("utf-8")
+                additional_headers.setdefault("Authorization", f"Basic {token}")
 
         self._http_client = PrefectHttpxAsyncClient(
             transport=httpx.ASGITransport(app=api_app, raise_app_exceptions=False),
             headers={**additional_headers},
             base_url="http://prefect-in-memory/api",
-            enable_csrf_support=False,
+            enable_csrf_support=settings.server.api.csrf_protection_enabled,
             raise_on_all_errors=False,
         )
 

{prefect_client-3.4.3.dev1 → prefect_client-3.4.4}/src/prefect/settings/models/tasks.py

@@ -63,6 +63,11 @@ class TasksSettings(PrefectBaseSettings):
         description="If `True`, sets the default cache policy on all tasks to `NO_CACHE`.",
     )
 
+    disable_caching: bool = Field(
+        default=False,
+        description="If `True`, disables caching on all tasks regardless of cache policy.",
+    )
+
     default_retries: int = Field(
         default=0,
         ge=0,

{prefect_client-3.4.3.dev1 → prefect_client-3.4.4}/src/prefect/task_engine.py

@@ -546,9 +546,13 @@ class SyncTaskRunEngine(BaseTaskRunEngine[P, R]):
             self.retries: int = self.retries + 1
             return True
         elif self.retries >= self.task.retries:
+            retry_message_suffix = (
+                "Retries are exhausted"
+                if self.task.retries > 0
+                else "No retries configured for this task."
+            )
             self.logger.error(
-                "Task run failed with exception: %r - Retries are exhausted",
-                exc,
+                f"Task run failed with exception: {exc!r} - {retry_message_suffix}",
                 exc_info=True,
             )
             return False
@@ -1096,9 +1100,13 @@ class AsyncTaskRunEngine(BaseTaskRunEngine[P, R]):
             self.retries: int = self.retries + 1
             return True
         elif self.retries >= self.task.retries:
+            retry_message_suffix = (
+                "Retries are exhausted"
+                if self.task.retries > 0
+                else "No retries configured for this task."
+            )
             self.logger.error(
-                "Task run failed with exception: %r - Retries are exhausted",
-                exc,
+                f"Task run failed with exception: {exc!r} - {retry_message_suffix}",
                 exc_info=True,
             )
             return False

{prefect_client-3.4.3.dev1 → prefect_client-3.4.4}/src/prefect/task_runners.py

@@ -11,11 +11,8 @@ from typing import (
     TYPE_CHECKING,
     Any,
     Coroutine,
-    Dict,
     Generic,
     Iterable,
-    List,
-    Optional,
     overload,
 )
 
@@ -110,7 +107,7 @@ class TaskRunner(abc.ABC, Generic[F]):
         self,
         task: "Task[P, R]",
         parameters: dict[str, Any | unmapped[Any] | allow_failure[Any]],
-        wait_for: Optional[Iterable[PrefectFuture[R]]] = None,
+        wait_for: Iterable[PrefectFuture[R]] | None = None,
     ) -> PrefectFutureList[F]:
         """
         Submit multiple tasks to the task run engine.
@@ -183,7 +180,7 @@ class TaskRunner(abc.ABC, Generic[F]):
 
         map_length = list(lengths)[0]
 
-        futures: List[PrefectFuture[Any]] = []
+        futures: list[PrefectFuture[Any]] = []
         for i in range(map_length):
             call_parameters: dict[str, Any] = {
                 key: value[i] for key, value in iterable_parameters.items()
@@ -229,15 +226,32 @@ class TaskRunner(abc.ABC, Generic[F]):
 
 
 class ThreadPoolTaskRunner(TaskRunner[PrefectConcurrentFuture[R]]):
-    def __init__(self, max_workers: Optional[int] = None):
+    """
+    A task runner that executes tasks in a separate thread pool.
+
+    Attributes:
+        max_workers: The maximum number of threads to use for executing tasks.
+            Defaults to `PREFECT_TASK_RUNNER_THREAD_POOL_MAX_WORKERS` or `sys.maxsize`.
+
+    Note:
+        This runner uses `contextvars.copy_context()` for thread-safe context propagation.
+        However, because contextvars are thread-local, frequent task submissions
+        that modify context (e.g., using `prefect.tags` in a loop) can lead to
+        new thread creation per task. This may cause an increase in threads and
+        file descriptors, potentially hitting OS limits (`OSError: Too many open files`).
+        If this occurs, consider minimizing context changes within looped tasks or
+        adjusting system limits for open file descriptors.
+    """
+
+    def __init__(self, max_workers: int | None = None):
         super().__init__()
-        self._executor: Optional[ThreadPoolExecutor] = None
+        self._executor: ThreadPoolExecutor | None = None
         self._max_workers = (
             (PREFECT_TASK_RUNNER_THREAD_POOL_MAX_WORKERS.value() or sys.maxsize)
             if max_workers is None
             else max_workers
         )
-        self._cancel_events: Dict[uuid.UUID, threading.Event] = {}
+        self._cancel_events: dict[uuid.UUID, threading.Event] = {}
 
     def duplicate(self) -> "ThreadPoolTaskRunner[R]":
         return type(self)(max_workers=self._max_workers)

{prefect_client-3.4.3.dev1 → prefect_client-3.4.4}/src/prefect/tasks.py

@@ -466,6 +466,10 @@ class Task(Generic[P, R]):
             ):
                 persist_result = True
 
+        # Check for global cache disable setting
+        if settings.tasks.disable_caching:
+            cache_policy = NO_CACHE
+
         if persist_result is False:
             self.cache_policy = None if cache_policy is None else NO_CACHE
             if cache_policy and cache_policy is not NotSet and cache_policy != NO_CACHE:

prefect_client-3.4.3.dev1/src/prefect/_build_info.py

@@ -1,5 +0,0 @@
-# Generated by versioningit
-__version__ = "3.4.3.dev1"
-__build_date__ = "2025-05-20 08:09:25.169022+00:00"
-__git_commit__ = "26c5fa05c23e2bffcdb97596e083a490352e332d"
-__dirty__ = False