prefect-client 3.1.2__py3-none-any.whl → 3.1.3__py3-none-any.whl

prefect/_internal/compatibility/async_dispatch.py

@@ -1,73 +1,63 @@
 import asyncio
 import inspect
 from functools import wraps
-from typing import Any, Callable, Coroutine, Protocol, TypeVar, Union
+from typing import Any, Callable, Coroutine, Optional, TypeVar, Union
 
 from typing_extensions import ParamSpec
 
+from prefect.tasks import Task
+
 R = TypeVar("R")
 P = ParamSpec("P")
 
 
-class AsyncDispatchable(Protocol[P, R]):
-    """Protocol for functions decorated with async_dispatch."""
-
-    def __call__(
-        self, *args: P.args, **kwargs: P.kwargs
-    ) -> Union[R, Coroutine[Any, Any, R]]:
-        ...
-
-    aio: Callable[P, Coroutine[Any, Any, R]]
-    sync: Callable[P, R]
-
-
 def is_in_async_context() -> bool:
-    """Check if we're in an async context."""
+    """
+    Returns True if called from within an async context (coroutine or running event loop)
+    """
     try:
-        # First check if we're in a coroutine
-        if asyncio.current_task() is not None:
-            return True
-
-        # Check if we have a loop and it's running
-        loop = asyncio.get_event_loop()
-        return loop.is_running()
+        asyncio.get_running_loop()
+        return True
     except RuntimeError:
         return False
 
 
+def _is_acceptable_callable(obj: Union[Callable, Task]) -> bool:
+    if inspect.iscoroutinefunction(obj):
+        return True
+    if isinstance(obj, Task) and inspect.iscoroutinefunction(obj.fn):
+        return True
+    return False
+
+
 def async_dispatch(
     async_impl: Callable[P, Coroutine[Any, Any, R]],
-) -> Callable[[Callable[P, R]], AsyncDispatchable[P, R]]:
+) -> Callable[[Callable[P, R]], Callable[P, Union[R, Coroutine[Any, Any, R]]]]:
     """
-    Decorator that adds async compatibility to a sync function.
-
-    The decorated function will:
-    - Return a coroutine when in an async context (detected via running event loop)
-    - Run synchronously when in a sync context
-    - Provide .aio for explicit async access
-    - Provide .sync for explicit sync access
+    Decorator that dispatches to either sync or async implementation based on context.
 
     Args:
-        async_impl: The async implementation to dispatch to when async execution
-                   is needed
+        async_impl: The async implementation to dispatch to when in async context
     """
-    if not inspect.iscoroutinefunction(async_impl):
-        raise TypeError(
-            "async_impl must be an async function to dispatch in async contexts"
-        )
 
-    def decorator(sync_fn: Callable[P, R]) -> AsyncDispatchable[P, R]:
+    def decorator(
+        sync_fn: Callable[P, R],
+    ) -> Callable[P, Union[R, Coroutine[Any, Any, R]]]:
+        if not _is_acceptable_callable(async_impl):
+            raise TypeError("async_impl must be an async function")
+
         @wraps(sync_fn)
         def wrapper(
-            *args: P.args, **kwargs: P.kwargs
+            *args: P.args,
+            _sync: Optional[bool] = None,  # type: ignore
+            **kwargs: P.kwargs,
         ) -> Union[R, Coroutine[Any, Any, R]]:
-            if is_in_async_context():
-                return async_impl(*args, **kwargs)
-            return sync_fn(*args, **kwargs)
+            should_run_sync = _sync if _sync is not None else not is_in_async_context()
+
+            if should_run_sync:
+                return sync_fn(*args, **kwargs)
+            return async_impl(*args, **kwargs)
 
-        # Attach both async and sync implementations directly
-        wrapper.aio = async_impl
-        wrapper.sync = sync_fn
         return wrapper  # type: ignore
 
     return decorator

prefect/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2024-11-12T13:38:45-0800",
+ "date": "2024-11-19T15:25:34-0600",
  "dirty": true,
  "error": null,
- "full-revisionid": "02b99f0a756f0395e4292952dcb2d963fc839750",
- "version": "3.1.2"
+ "full-revisionid": "39b6028cea9f2b0437546ba13cc08bb3bf6d94a4",
+ "version": "3.1.3"
 }
 '''  # END VERSION_JSON
 

prefect/blocks/notifications.py

@@ -494,6 +494,7 @@ class OpsgenieWebhook(AbstractAppriseNotificationBlock):
                 entity=self.entity,
                 batch=self.batch,
                 tags=self.tags,
+                action="new",
             ).url()
         )
         self._start_apprise_client(url)

prefect/cache_policies.py

@@ -7,6 +7,7 @@ from typing import TYPE_CHECKING, Any, Callable, Dict, List, Literal, Optional,
 from typing_extensions import Self
 
 from prefect.context import TaskRunContext
+from prefect.exceptions import HashError
 from prefect.utilities.hashing import hash_objects
 
 if TYPE_CHECKING:
@@ -223,7 +224,6 @@ class TaskSource(CachePolicy):
                 lines = task_ctx.task.fn.__code__.co_code
             else:
                 raise
-
         return hash_objects(lines, raise_on_failure=True)
 
 
@@ -293,7 +293,18 @@ class Inputs(CachePolicy):
             if key not in exclude:
                 hashed_inputs[key] = val
 
-        return hash_objects(hashed_inputs, raise_on_failure=True)
+        try:
+            return hash_objects(hashed_inputs, raise_on_failure=True)
+        except HashError as exc:
+            msg = (
+                f"{exc}\n\n"
+                "This often occurs when task inputs contain objects that cannot be cached "
+                "like locks, file handles, or other system resources.\n\n"
+                "To resolve this, you can:\n"
+                "  1. Exclude these arguments by defining a custom `cache_key_fn`\n"
+                "  2. Disable caching by passing `cache_policy=NONE`\n"
+            )
+            raise ValueError(msg) from exc
 
     def __sub__(self, other: str) -> "CachePolicy":
         if not isinstance(other, str):

prefect/client/schemas/objects.py

@@ -497,6 +497,9 @@ class FlowRunPolicy(PrefectBaseModel):
     resuming: Optional[bool] = Field(
         default=False, description="Indicates if this run is resuming from a pause."
     )
+    retry_type: Optional[Literal["in_process", "reschedule"]] = Field(
+        default=None, description="The type of retry this run is undergoing."
+    )
 
     @model_validator(mode="before")
     @classmethod
@@ -560,6 +563,11 @@ class FlowRun(ObjectBaseModel):
         description="A list of tags on the flow run",
         examples=[["tag-1", "tag-2"]],
     )
+    labels: KeyValueLabels = Field(
+        default_factory=dict,
+        description="Prefect Cloud: A dictionary of key-value labels. Values can be strings, numbers, or booleans.",
+        examples=[{"key": "value1", "key2": 42}],
+    )
     parent_task_run_id: Optional[UUID] = Field(
         default=None,
         description=(

prefect/events/filters.py

@@ -83,17 +83,18 @@ class EventOccurredFilter(EventDataFilter):
 
 class EventNameFilter(EventDataFilter):
     prefix: Optional[List[str]] = Field(
-        None, description="Only include events matching one of these prefixes"
+        default=None, description="Only include events matching one of these prefixes"
     )
     exclude_prefix: Optional[List[str]] = Field(
-        None, description="Exclude events matching one of these prefixes"
+        default=None, description="Exclude events matching one of these prefixes"
     )
 
     name: Optional[List[str]] = Field(
-        None, description="Only include events matching one of these names exactly"
+        default=None,
+        description="Only include events matching one of these names exactly",
     )
     exclude_name: Optional[List[str]] = Field(
-        None, description="Exclude events matching one of these names exactly"
+        default=None, description="Exclude events matching one of these names exactly"
     )
 
     def includes(self, event: Event) -> bool:
@@ -230,17 +231,20 @@ class EventFilter(EventDataFilter):
         description="Filter criteria for when the events occurred",
     )
     event: Optional[EventNameFilter] = Field(
-        None,
+        default=None,
         description="Filter criteria for the event name",
     )
     any_resource: Optional[EventAnyResourceFilter] = Field(
-        None, description="Filter criteria for any resource involved in the event"
+        default=None,
+        description="Filter criteria for any resource involved in the event",
     )
     resource: Optional[EventResourceFilter] = Field(
-        None, description="Filter criteria for the resource of the event"
+        default=None,
+        description="Filter criteria for the resource of the event",
     )
     related: Optional[EventRelatedFilter] = Field(
-        None, description="Filter criteria for the related resources of the event"
+        default=None,
+        description="Filter criteria for the related resources of the event",
     )
     id: EventIDFilter = Field(
         default_factory=lambda: EventIDFilter(id=[]),
@@ -248,6 +252,6 @@ class EventFilter(EventDataFilter):
     )
 
     order: EventOrder = Field(
-        EventOrder.DESC,
+        default=EventOrder.DESC,
         description="The order to return filtered events",
     )

prefect/exceptions.py

@@ -443,3 +443,7 @@ class ProfileSettingsValidationError(PrefectException):
 
     def __init__(self, errors: List[Tuple[Any, ValidationError]]) -> None:
         self.errors = errors
+
+
+class HashError(PrefectException):
+    """Raised when hashing objects fails"""

prefect/flow_engine.py

@@ -22,8 +22,11 @@ from typing import (
 )
 from uuid import UUID
 
+from opentelemetry import trace
+from opentelemetry.trace import Tracer, get_tracer
 from typing_extensions import ParamSpec
 
+import prefect
 from prefect import Task
 from prefect.client.orchestration import SyncPrefectClient, get_client
 from prefect.client.schemas import FlowRun, TaskRun
@@ -124,6 +127,10 @@ class FlowRunEngine(Generic[P, R]):
     _client: Optional[SyncPrefectClient] = None
     short_circuit: bool = False
     _flow_run_name_set: bool = False
+    _tracer: Tracer = field(
+        default_factory=lambda: get_tracer("prefect", prefect.__version__)
+    )
+    _span: Optional[trace.Span] = None
 
     def __post_init__(self):
         if self.flow is None and self.flow_run_id is None:
@@ -233,6 +240,17 @@ class FlowRunEngine(Generic[P, R]):
         self.flow_run.state = state  # type: ignore
         self.flow_run.state_name = state.name  # type: ignore
         self.flow_run.state_type = state.type  # type: ignore
+
+        if self._span:
+            self._span.add_event(
+                state.name,
+                {
+                    "prefect.state.message": state.message or "",
+                    "prefect.state.type": state.type,
+                    "prefect.state.name": state.name or state.type,
+                    "prefect.state.id": str(state.id),
+                },
+            )
         return state
 
     def result(self, raise_on_failure: bool = True) -> "Union[R, State, None]":
@@ -281,6 +299,9 @@ class FlowRunEngine(Generic[P, R]):
         )
         self.set_state(terminal_state)
         self._return_value = resolved_result
+
+        self._end_span_on_success()
+
         return result
 
     def handle_exception(
@@ -311,6 +332,9 @@ class FlowRunEngine(Generic[P, R]):
             )
             state = self.set_state(Running())
         self._raised = exc
+
+        self._end_span_on_error(exc, state.message)
+
         return state
 
     def handle_timeout(self, exc: TimeoutError) -> None:
@@ -329,6 +353,8 @@ class FlowRunEngine(Generic[P, R]):
         self.set_state(state)
         self._raised = exc
 
+        self._end_span_on_error(exc, message)
+
     def handle_crash(self, exc: BaseException) -> None:
         state = run_coro_as_sync(exception_to_crashed_state(exc))
         self.logger.error(f"Crash detected! {state.message}")
@@ -336,6 +362,23 @@ class FlowRunEngine(Generic[P, R]):
         self.set_state(state, force=True)
         self._raised = exc
 
+        self._end_span_on_error(exc, state.message)
+
+    def _end_span_on_success(self):
+        if not self._span:
+            return
+        self._span.set_status(trace.Status(trace.StatusCode.OK))
+        self._span.end(time.time_ns())
+        self._span = None
+
+    def _end_span_on_error(self, exc: BaseException, description: Optional[str]):
+        if not self._span:
+            return
+        self._span.record_exception(exc)
+        self._span.set_status(trace.Status(trace.StatusCode.ERROR, description))
+        self._span.end(time.time_ns())
+        self._span = None
+
     def load_subflow_run(
         self,
         parent_task_run: TaskRun,
@@ -578,6 +621,18 @@ class FlowRunEngine(Generic[P, R]):
                     flow_version=self.flow.version,
                     empirical_policy=self.flow_run.empirical_policy,
                 )
+
+            self._span = self._tracer.start_span(
+                name=self.flow_run.name,
+                attributes={
+                    **self.flow_run.labels,
+                    "prefect.run.type": "flow",
+                    "prefect.run.id": str(self.flow_run.id),
+                    "prefect.tags": self.flow_run.tags,
+                    "prefect.flow.name": self.flow.name,
+                },
+            )
+
             try:
                 yield self
 
@@ -632,7 +687,7 @@ class FlowRunEngine(Generic[P, R]):
 
     @contextmanager
     def start(self) -> Generator[None, None, None]:
-        with self.initialize_run():
+        with self.initialize_run(), trace.use_span(self._span):
             self.begin_run()
 
             if self.state.is_running():

prefect/logging/handlers.py

@@ -32,7 +32,6 @@ from prefect.settings import (
     PREFECT_LOGGING_TO_API_BATCH_SIZE,
     PREFECT_LOGGING_TO_API_MAX_LOG_SIZE,
     PREFECT_LOGGING_TO_API_WHEN_MISSING_FLOW,
-    get_current_settings,
 )
 
 
@@ -241,10 +240,12 @@ class APILogHandler(logging.Handler):
 
 class WorkerAPILogHandler(APILogHandler):
     def emit(self, record: logging.LogRecord):
-        if get_current_settings().experiments.worker_logging_to_api_enabled:
-            super().emit(record)
-        else:
+        # Open-source API servers do not currently support worker logs, and
+        # worker logs only have an associated worker ID when connected to Cloud,
+        # so we won't send worker logs to the API unless they have a worker ID.
+        if not getattr(record, "worker_id", None):
             return
+        super().emit(record)
 
     def prepare(self, record: logging.LogRecord) -> Dict[str, Any]:
         """

prefect/settings/models/experiments.py

@@ -18,11 +18,6 @@ class ExperimentsSettings(PrefectBaseSettings):
         ),
     )
 
-    worker_logging_to_api_enabled: bool = Field(
-        default=False,
-        description="Enables the logging of worker logs to Prefect Cloud.",
-    )
-
     telemetry_enabled: bool = Field(
         default=False,
         description="Enables sending telemetry to Prefect Cloud.",

prefect/telemetry/__init__.py

@@ -0,0 +1 @@
+

prefect/telemetry/processors.py

@@ -1,6 +1,6 @@
 import time
 from threading import Event, Lock, Thread
-from typing import Optional
+from typing import Dict, Optional
 
 from opentelemetry.context import Context
 from opentelemetry.sdk.trace import ReadableSpan, Span, SpanProcessor
@@ -10,7 +10,7 @@ from opentelemetry.sdk.trace.export import SpanExporter
 class InFlightSpanProcessor(SpanProcessor):
     def __init__(self, span_exporter: SpanExporter):
         self.span_exporter = span_exporter
-        self._in_flight = {}
+        self._in_flight: Dict[int, Span] = {}
         self._lock = Lock()
         self._stop_event = Event()
         self._export_thread = Thread(target=self._export_periodically, daemon=True)

prefect/utilities/filesystem.py

@@ -7,7 +7,7 @@ import pathlib
 import threading
 from contextlib import contextmanager
 from pathlib import Path, PureWindowsPath
-from typing import Optional, Union
+from typing import Optional, Union, cast
 
 import fsspec
 import pathspec
@@ -22,8 +22,8 @@ def create_default_ignore_file(path: str) -> bool:
     Creates default ignore file in the provided path if one does not already exist; returns boolean specifying
     whether a file was created.
     """
-    path = pathlib.Path(path)
-    ignore_file = path / ".prefectignore"
+    _path = pathlib.Path(path)
+    ignore_file = _path / ".prefectignore"
     if ignore_file.exists():
         return False
     default_file = pathlib.Path(prefect.__module_path__) / ".prefectignore"
@@ -54,12 +54,34 @@ def filter_files(
 chdir_lock = threading.Lock()
 
 
+def _normalize_path(path: Union[str, Path]) -> str:
+    """
+    Normalize a path, handling UNC paths on Windows specially.
+    """
+    path = Path(path)
+
+    # Handle UNC paths on Windows differently
+    if os.name == "nt" and str(path).startswith("\\\\"):
+        # For UNC paths, use absolute() instead of resolve()
+        # to avoid the Windows path resolution issues
+        return str(path.absolute())
+    else:
+        try:
+            # For non-UNC paths, try resolve() first
+            return str(path.resolve())
+        except OSError:
+            # Fallback to absolute() if resolve() fails
+            return str(path.absolute())
+
+
 @contextmanager
 def tmpchdir(path: str):
     """
-    Change current-working directories for the duration of the context
+    Change current-working directories for the duration of the context,
+    with special handling for UNC paths on Windows.
     """
-    path = os.path.abspath(path)
+    path = _normalize_path(path)
+
     if os.path.isfile(path) or (not os.path.exists(path) and not path.endswith("/")):
         path = os.path.dirname(path)
 
@@ -67,7 +89,12 @@ def tmpchdir(path: str):
 
     with chdir_lock:
         try:
-            os.chdir(path)
+            # On Windows with UNC paths, we need to handle the directory change carefully
+            if os.name == "nt" and path.startswith("\\\\"):
+                # Use os.path.abspath to handle UNC paths
+                os.chdir(os.path.abspath(path))
+            else:
+                os.chdir(path)
             yield path
         finally:
             os.chdir(owd)
@@ -76,7 +103,7 @@ def tmpchdir(path: str):
 def filename(path: str) -> str:
     """Extract the file name from a path with remote file system support"""
     try:
-        of: OpenFile = fsspec.open(path)
+        of: OpenFile = cast(OpenFile, fsspec.open(path))
         sep = of.fs.sep
     except (ImportError, AttributeError):
         sep = "\\" if "\\" in path else "/"
@@ -98,7 +125,7 @@ def is_local_path(path: Union[str, pathlib.Path, OpenFile]):
     else:
         raise TypeError(f"Invalid path of type {type(path).__name__!r}")
 
-    return type(of.fs) == LocalFileSystem
+    return isinstance(of.fs, LocalFileSystem)
 
 
 def to_display_path(

prefect/utilities/hashing.py

@@ -6,6 +6,7 @@ from typing import Optional, Union
 
 import cloudpickle
 
+from prefect.exceptions import HashError
 from prefect.serializers import JSONSerializer
 
 if sys.version_info[:2] >= (3, 9):
@@ -53,19 +54,39 @@ def hash_objects(
 ) -> Optional[str]:
     """
     Attempt to hash objects by dumping to JSON or serializing with cloudpickle.
-    On failure of both, `None` will be returned; to raise on failure, set
-    `raise_on_failure=True`.
+
+    Args:
+        *args: Positional arguments to hash
+        hash_algo: Hash algorithm to use
+        raise_on_failure: If True, raise exceptions instead of returning None
+        **kwargs: Keyword arguments to hash
+
+    Returns:
+        A hash string or None if hashing failed
+
+    Raises:
+        HashError: If objects cannot be hashed and raise_on_failure is True
     """
+    json_error = None
+    pickle_error = None
+
     try:
         serializer = JSONSerializer(dumps_kwargs={"sort_keys": True})
         return stable_hash(serializer.dumps((args, kwargs)), hash_algo=hash_algo)
-    except Exception:
-        pass
+    except Exception as e:
+        json_error = str(e)
 
     try:
         return stable_hash(cloudpickle.dumps((args, kwargs)), hash_algo=hash_algo)
-    except Exception:
-        if raise_on_failure:
-            raise
+    except Exception as e:
+        pickle_error = str(e)
+
+    if raise_on_failure:
+        msg = (
+            "Unable to create hash - objects could not be serialized.\n"
+            f"  JSON error: {json_error}\n"
+            f"  Pickle error: {pickle_error}"
+        )
+        raise HashError(msg)
 
     return None

prefect/utilities/urls.py

@@ -3,7 +3,7 @@ import ipaddress
 import socket
 import urllib.parse
 from string import Formatter
-from typing import TYPE_CHECKING, Any, Dict, Literal, Optional, Union
+from typing import TYPE_CHECKING, Any, Literal, Optional, Union
 from urllib.parse import urlparse
 from uuid import UUID
 
@@ -135,7 +135,7 @@ def url_for(
     obj_id: Optional[Union[str, UUID]] = None,
     url_type: URLType = "ui",
     default_base_url: Optional[str] = None,
-    **additional_format_kwargs: Optional[Dict[str, Any]],
+    **additional_format_kwargs: Any,
 ) -> Optional[str]:
     """
     Returns the URL for a Prefect object.

prefect/workers/base.py

@@ -146,26 +146,26 @@ class BaseJobConfiguration(BaseModel):
         Important: this method expects that the base_job_template was already
         validated server-side.
         """
-        job_config: Dict[str, Any] = base_job_template["job_configuration"]
+        base_config: Dict[str, Any] = base_job_template["job_configuration"]
         variables_schema = base_job_template["variables"]
         variables = cls._get_base_config_defaults(
             variables_schema.get("properties", {})
         )
 
-        # copy variable defaults for `env` to job config before they're replaced by
+        # copy variable defaults for `env` to base config before they're replaced by
         # deployment overrides
         if variables.get("env"):
-            job_config["env"] = variables.get("env")
+            base_config["env"] = variables.get("env")
 
         variables.update(values)
 
         # deep merge `env`
-        if isinstance(job_config.get("env"), dict) and (
-            hardcoded_env := variables.get("env")
+        if isinstance(base_config.get("env"), dict) and (
+            deployment_env := variables.get("env")
         ):
-            job_config["env"] = hardcoded_env | job_config.get("env")
+            base_config["env"] = base_config.get("env") | deployment_env
 
-        populated_configuration = apply_values(template=job_config, values=variables)
+        populated_configuration = apply_values(template=base_config, values=variables)
         populated_configuration = await resolve_block_document_references(
             template=populated_configuration, client=client
         )
@@ -649,9 +649,14 @@ class BaseWorker(abc.ABC):
         for scope in self._scheduled_task_scopes:
             scope.cancel()
 
-        await self._exit_stack.__aexit__(*exc_info)
+        # Emit stopped event before closing client
         if self._started_event:
-            await self._emit_worker_stopped_event(self._started_event)
+            try:
+                await self._emit_worker_stopped_event(self._started_event)
+            except Exception:
+                self._logger.exception("Failed to emit worker stopped event")
+
+        await self._exit_stack.__aexit__(*exc_info)
         self._runs_task_group = None
         self._client = None
 
@@ -816,14 +821,14 @@ class BaseWorker(abc.ABC):
             self._logger = get_worker_logger(self)
 
         self._logger.debug(
-            f"Worker synchronized with the Prefect API server. Remote ID: {self.backend_id}"
+            "Worker synchronized with the Prefect API server. "
+            + (f"Remote ID: {self.backend_id}" if self.backend_id else "")
         )
 
     def _should_get_worker_id(self):
         """Determines if the worker should request an ID from the API server."""
         return (
-            get_current_settings().experiments.worker_logging_to_api_enabled
-            and self._client
+            self._client
             and self._client.server_type == ServerType.CLOUD
             and self.backend_id is None
         )
@@ -881,10 +886,7 @@ class BaseWorker(abc.ABC):
                 run_logger.info(
                     f"Worker '{self.name}' submitting flow run '{flow_run.id}'"
                 )
-                if (
-                    get_current_settings().experiments.worker_logging_to_api_enabled
-                    and self.backend_id
-                ):
+                if self.backend_id:
                     try:
                         worker_url = url_for(
                             "worker",
@@ -971,10 +973,11 @@ class BaseWorker(abc.ABC):
 
             else:
                 # If the run is not ready to submit, release the concurrency slot
-                if self._limiter:
-                    self._limiter.release_on_behalf_of(flow_run.id)
+                self._release_limit_slot(flow_run.id)
 
             self._submitting_flow_run_ids.remove(flow_run.id)
+        else:
+            self._release_limit_slot(flow_run.id)
 
     async def _submit_run_and_capture_errors(
         self, flow_run: "FlowRun", task_status: Optional[anyio.abc.TaskStatus] = None
@@ -1009,8 +1012,7 @@ class BaseWorker(abc.ABC):
                 )
             return exc
         finally:
-            if self._limiter:
-                self._limiter.release_on_behalf_of(flow_run.id)
+            self._release_limit_slot(flow_run.id)
 
         if not task_status._future.done():
             run_logger.error(
@@ -1035,6 +1037,14 @@ class BaseWorker(abc.ABC):
 
         return result
 
+    def _release_limit_slot(self, flow_run_id: str) -> None:
+        """
+        Frees up a slot taken by the given flow run id.
+        """
+        if self._limiter:
+            self._limiter.release_on_behalf_of(flow_run_id)
+            self._logger.debug("Limit slot released for flow run '%s'", flow_run_id)
+
     def get_status(self):
         """
         Retrieves the status of the current worker including its name, current worker

{prefect_client-3.1.2.dist-info → prefect_client-3.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prefect-client
-Version: 3.1.2
+Version: 3.1.3
 Summary: Workflow orchestration and management.
 Home-page: https://www.prefect.io
 Author: Prefect Technologies, Inc.
@@ -29,7 +29,7 @@ Requires-Dist: asgi-lifespan<3.0,>=1.0
 Requires-Dist: cachetools<6.0,>=5.3
 Requires-Dist: cloudpickle<4.0,>=2.0
 Requires-Dist: coolname<3.0.0,>=1.0.4
-Requires-Dist: croniter<5.0.0,>=1.0.12
+Requires-Dist: croniter<6.0.0,>=1.0.12
 Requires-Dist: exceptiongroup>=1.0.0
 Requires-Dist: fastapi<1.0.0,>=0.111.0
 Requires-Dist: fsspec>=2022.5.0
@@ -39,6 +39,7 @@ Requires-Dist: httpcore<2.0.0,>=1.0.5
 Requires-Dist: httpx[http2]!=0.23.2,>=0.23
 Requires-Dist: jsonpatch<2.0,>=1.32
 Requires-Dist: jsonschema<5.0.0,>=4.0.0
+Requires-Dist: opentelemetry-api<2.0.0,>=1.27.0
 Requires-Dist: orjson<4.0,>=3.7
 Requires-Dist: packaging<24.3,>=21.3
 Requires-Dist: pathspec>=0.8.0

{prefect_client-3.1.2.dist-info → prefect_client-3.1.3.dist-info}/RECORD

@@ -1,15 +1,15 @@
 prefect/.prefectignore,sha256=awSprvKT0vI8a64mEOLrMxhxqcO-b0ERQeYpA2rNKVQ,390
 prefect/__init__.py,sha256=UZPTpdap8ECK1zLoggfeOtZGkKcf6um1-lMb-nn4o5I,3450
-prefect/_version.py,sha256=bglRMlWNqQBudSTba4t4FcS844kr38QYKku_fRWBhcI,496
+prefect/_version.py,sha256=V4oVtbz7s0T9HhJbsnPlJc7W2E-m2PBHC2K7gFbgZOQ,496
 prefect/agent.py,sha256=BOVVY5z-vUIQ2u8LwMTXDaNys2fjOZSS5YGDwJmTQjI,230
 prefect/artifacts.py,sha256=dsxFWmdg2r9zbHM3KgKOR5YbJ29_dXUYF9kipJpbxkE,13009
 prefect/automations.py,sha256=NlQ62GPJzy-gnWQqX7c6CQJKw7p60WLGDAFcy82vtg4,5613
-prefect/cache_policies.py,sha256=T4RTII0qkzGq-VRoE1wGezhkdiuDtf6k7qUA46J_wbA,9196
+prefect/cache_policies.py,sha256=pG6BtQw0w3f5dxlAZ-fjpILrxcTFkZH35Pa6Z4MfDtU,9762
 prefect/context.py,sha256=sNZwFecVO4-OmatU_OTNb9HoxG5rtwNIzP_NNGsmbYo,21496
 prefect/engine.py,sha256=BpmDbe6miZcTl1vRkxfCPYcWSXADLigGPCagFwucMz0,1976
-prefect/exceptions.py,sha256=V_nRpS2Z93PvJMoQdXbx8zepVaFb-pWanCqVi7T1ngI,11803
+prefect/exceptions.py,sha256=KR8Vg7GetiBmwU5rvV5imlpg410Mv6RqSyk_gEojWAI,11884
 prefect/filesystems.py,sha256=CxwMmKY8LBUed_9IqE2jUqxVCWhXa1r2fjKgLbIC2Vg,17893
-prefect/flow_engine.py,sha256=BuxfnKDqYcJvmVfPlAv8yMSwkDM25SBeJVk_MGQnm8k,30076
+prefect/flow_engine.py,sha256=FwmdAJd2Z2RvRhEwpTrR4xky_jzdS4dhOL7Z8MQZJ8c,31873
 prefect/flow_runs.py,sha256=EaXRIQTOnwnA0fO7_EjwafFRmS57K_CRy0Xsz3JDIhc,16070
 prefect/flows.py,sha256=PUEPpQx5pjXMVfnrPbIeAvWUdEdeaNPO4DXj7tgxGcA,90185
 prefect/futures.py,sha256=DlZvdccKtwQKuDUFrZ4zcINeO9C1chLiNOwjE5gTgCk,16356
@@ -32,7 +32,7 @@ prefect/_internal/integrations.py,sha256=U4cZMDbnilzZSKaMxvzZcSL27a1tzRMjDoTfr2u
 prefect/_internal/pytz.py,sha256=WWl9x16rKFWequGmcOGs_ljpCDPf2LDHMyZp_4D8e6c,13748
 prefect/_internal/retries.py,sha256=xtgj6oPSvYQLbyk451LR6swcRQvRVWEzCxY6GMK7qA4,2284
 prefect/_internal/compatibility/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-prefect/_internal/compatibility/async_dispatch.py,sha256=QpfHgjdmLW4PXp-sBXIqUlGL0bJnXhtLBGXi7PWR8Bo,2174
+prefect/_internal/compatibility/async_dispatch.py,sha256=eTFioVEHtfPYnUxgiwJW8vBcuVA0bfOb86Rcul6i9_I,1794
 prefect/_internal/compatibility/deprecated.py,sha256=PVME2C3Oe4_8tKIGufx1W4EpGkz5IQY8gFohPVOjNcM,7533
 prefect/_internal/compatibility/migration.py,sha256=oifIj6mKwVUfrUO13ODLKOCwLON4YqaTXzhBFh8AK6E,6797
 prefect/_internal/concurrency/__init__.py,sha256=YlTwU9ryjPNwbJa45adLJY00t_DGCh1QrdtY9WdVFfw,2140
@@ -61,7 +61,7 @@ prefect/blocks/__init__.py,sha256=BUfh6gIwA6HEjRyVCAiv0he3M1zfM-oY-JrlBfeWeY8,18
 prefect/blocks/abstract.py,sha256=YLzCaf3yXv6wFCF5ZqCIHJNwH7fME1rLxC-SijARHzk,16319
 prefect/blocks/core.py,sha256=l_56oggt9uJOABHus-NCXLQ4akeY4kzyDUO37ZyosX0,52783
 prefect/blocks/fields.py,sha256=1m507VVmkpOnMF_7N-qboRjtw4_ceIuDneX3jZ3Jm54,63
-prefect/blocks/notifications.py,sha256=Us5xkkicFes-Xwe0h4U7kdeDBUe6qY55Ig259oGkHlY,30484
+prefect/blocks/notifications.py,sha256=ufGULZCdGENKN1KYTVwVF9ZD21c_U7gE6PGKZjHOyYU,30514
 prefect/blocks/redis.py,sha256=GUKYyx2QLtyNvgf5FT_dJxbgQcOzWCja3I23J1-AXhM,5629
 prefect/blocks/system.py,sha256=OacB-LLXaNiLY49bPx7aAjmvdEdBxNoaOdzsCUcDr2c,4563
 prefect/blocks/webhook.py,sha256=F0u1WSO17Gda8qwr9gYaA84Nfc8Qkic6HhhJMYXRzug,2496
@@ -76,7 +76,7 @@ prefect/client/utilities.py,sha256=89fmza0cRMOayxgXRdO51TKb11TczJ0ByOZmcZVrt44,3
 prefect/client/schemas/__init__.py,sha256=KlyqFV-hMulMkNstBn_0ijoHoIwJZaBj6B1r07UmgvE,607
 prefect/client/schemas/actions.py,sha256=m2HJr9oAoK9vqigCHSwMGwvKPTjGNEtP0xdXgforIT0,28629
 prefect/client/schemas/filters.py,sha256=ynzD3mdvHHkI51pJ_NWgeDv8Awr7-2QtpUq7fTInEYM,36316
-prefect/client/schemas/objects.py,sha256=CJCMc9XrxM2EZliyvNRJdtYz8ZaI948wPFG3-BMXQdg,56753
+prefect/client/schemas/objects.py,sha256=8RTtKzWOdFq8hROB77DPLCVwuKP5Jyc9gB-fOF6Qwwc,57147
 prefect/client/schemas/responses.py,sha256=tV06W8npA8oCjV9d0ZNvjro4QcbHxayb8PC4LmanXjo,15467
 prefect/client/schemas/schedules.py,sha256=8rpqjOYtknu2-1n5_WD4cOplgu93P3mCyX86B22LfL4,13070
 prefect/client/schemas/sorting.py,sha256=L-2Mx-igZPtsUoRUguTcG3nIEstMEMPD97NwPM2Ox5s,2579
@@ -109,7 +109,7 @@ prefect/docker/docker_image.py,sha256=Y84_ooCYA9NGl6FElJul9-FaW3teT-eia2SiNtZ1LG
 prefect/events/__init__.py,sha256=GtKl2bE--pJduTxelH2xy7SadlLJmmis8WR1EYixhuA,2094
 prefect/events/actions.py,sha256=A7jS8bo4zWGnrt3QfSoQs0uYC1xfKXio3IfU0XtTb5s,9129
 prefect/events/clients.py,sha256=fCR64VROlbMfVY5WL7Dy_1UroBKYrKNltll2sIiD8Ek,23028
-prefect/events/filters.py,sha256=IJ1TF-TCC7Wk2nJsbYW-HyAANToDQ6z1MdD63qE-lfw,8186
+prefect/events/filters.py,sha256=IaJ39jGpkTocJOyrrBmnXEJEYeYZyB600Ri2apw6cxI,8294
 prefect/events/related.py,sha256=TQPYIPJI_vZlZgZgq3YpsGCmFleiZCMDtn_jMrYBJRg,6537
 prefect/events/utilities.py,sha256=ajIAiNFTN5Bz57IEq-o-i1BJdUi7P2oYH_6GyQjCKs8,2635
 prefect/events/worker.py,sha256=VEz5xgJ9KuPFNH_H08YJ2_v2lYM5ASE-Nx2Yd9DqfZ8,4126
@@ -138,7 +138,7 @@ prefect/logging/__init__.py,sha256=zx9f5_dWrR4DbcTOFBpNGOPoCZ1QcPFudr7zxb2XRpA,1
 prefect/logging/configuration.py,sha256=5kOYdd4Hi6t-wJhxEncYS9bUAMEMTRKy_i22pepjDrw,3343
 prefect/logging/filters.py,sha256=9keHLN4-cpnsWcii1qU0RITNi9-m7pOhkJ_t0MtCM4k,1117
 prefect/logging/formatters.py,sha256=3nBWgawvD48slT0zgkKeus1gIyf0OjmDKdLwMEe5mPU,3923
-prefect/logging/handlers.py,sha256=86uVPKimc6sBqpig6-nMSSbgXFkftvlKiRgZ7EtMqsw,11776
+prefect/logging/handlers.py,sha256=ywaVcQmC8-v_B9l7Cand6kvjeFgPhfDTiqbHpQKt1vE,11943
 prefect/logging/highlighters.py,sha256=pH6TZmMaMdAkMLJB-U9NPWZIj3GUfczYCSFzCbBCq2s,1776
 prefect/logging/loggers.py,sha256=BKYjyP8N2aJPdHwpqNU5Od4uu0pKztsDQ7RPRNb4oGo,12146
 prefect/logging/logging.yml,sha256=-HoWPYLsKMt5bKRgjIqk8S7tQlAaQ93pGkwIkYtMVMo,3172
@@ -173,7 +173,7 @@ prefect/settings/models/cli.py,sha256=mHB-BHBVO9qfQcr9uHbBmU6MMDLlLUUDxjyaRz7v1l
 prefect/settings/models/client.py,sha256=Ua18wKXWrNi0Ktb6cd42kdSOQYhFnObwTHTc3QDYg2k,2985
 prefect/settings/models/cloud.py,sha256=7-WtW0rGPMOuG1O9Lwv_H_keEldXPo9YGHmoGwCXpeg,1767
 prefect/settings/models/deployments.py,sha256=ui4xqT7N-6B_gMnGnkJ8Ys0hzdfNTOtlcwzAQo2LlQI,1235
-prefect/settings/models/experiments.py,sha256=CXezPxuyV4pTxvQ9xroUmpEudXbDgUv-Y56A16307Wc,862
+prefect/settings/models/experiments.py,sha256=toCpajFA9d2rosqshdlQBayOnyJff9zyrpRCEKAF7Bo,707
 prefect/settings/models/flows.py,sha256=IrW3cFQJnu-31cXvSihCl6naRhmZc2eFqSgqdnNLUzg,1056
 prefect/settings/models/internal.py,sha256=mJ7h3d_WHw-0FBVhIbDHPjYI041YtuReD1bZO98G1OU,612
 prefect/settings/models/logging.py,sha256=YxD8fOcWOROUrq6dH6EIadJwYL7LlLSUGWlHVLn-l3Y,4560
@@ -194,11 +194,11 @@ prefect/settings/models/server/root.py,sha256=uhryCvrk6oGAu8S3GGZNc3IaxXyd_dFnsK
 prefect/settings/models/server/services.py,sha256=xDQyRUh18mU_cTgp1Q_BwKpdeZHKLSh2o_YNNXvzsoo,16748
 prefect/settings/models/server/tasks.py,sha256=YzfRTcmiYnD_INezR_29rvO0M_YbAqT1by7kiK2KP60,2814
 prefect/settings/models/server/ui.py,sha256=6cTSC2RQsS4c2HpB1Se6qRms4RMEKSsaI40T2CTkobg,1780
-prefect/telemetry/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+prefect/telemetry/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 prefect/telemetry/bootstrap.py,sha256=oOdEjIHDQHANviyP2ZKpz9B4yBq832f9y1PKwAQ5-ro,987
 prefect/telemetry/instrumentation.py,sha256=S7IRYqKXaixAt-W5oxDO4CqkyXc3dy4ATgBmuindJIM,4241
 prefect/telemetry/logging.py,sha256=yn5D4D2GGRrAv0y8wlHPN7PZDmQucGjQT_YauK9M9Yo,727
-prefect/telemetry/processors.py,sha256=5vxjk9m-0FxqGDmVySY_u4zceZ9mwmTPiSFuCDOyGWA,2006
+prefect/telemetry/processors.py,sha256=IEWBD8c_O3lx54TbIsBOF8kneIMcT27bcwh3O0VY7HI,2029
 prefect/types/__init__.py,sha256=oabzKSvuT7tKBX6OiBev40wUtN1JstJT2L7ocpEUZ14,3640
 prefect/types/entrypoint.py,sha256=2FF03-wLPgtnqR_bKJDB2BsXXINPdu8ptY9ZYEZnXg8,328
 prefect/utilities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -211,8 +211,8 @@ prefect/utilities/context.py,sha256=BThuUW94-IYgFYTeMIM9KMo8ShT3oiI7w5ajZHzU1j0,
 prefect/utilities/dispatch.py,sha256=EthEmyRwv-4W8z2BJclrsOQHJ_pJoZYL0t2cyYPEa-E,6098
 prefect/utilities/dockerutils.py,sha256=zjqeyE4gK8r0n5l3b2XK2AKviQ2F-pOd1LE2O4qfJt0,20372
 prefect/utilities/engine.py,sha256=pi-TQEpoJ98u2LNYh3ZpNoiN9VSlGRJV558Hjg2eFxY,32433
-prefect/utilities/filesystem.py,sha256=frAyy6qOeYa7c-jVbEUGZQEe6J1yF8I_SvUepPd59gI,4415
-prefect/utilities/hashing.py,sha256=SK18PDwJQjn3_qwNwwnz6Wuz1o8rR6Bja-sEuSfszxE,1879
+prefect/utilities/filesystem.py,sha256=FH5SRDeLei5_nouYmYkr44GdCJy7p9rIV-VyhW71ziE,5420
+prefect/utilities/hashing.py,sha256=IrcILlSvLz1hFtOFtIelPtkcyZ7qnaARYlrGFv3eVxs,2485
 prefect/utilities/importtools.py,sha256=aO-xhf2h2KzsLGvSKwRAZLB4ITeW9rsV0Ys-gwq3i7o,19426
 prefect/utilities/math.py,sha256=wLwcKVidpNeWQi1TUIWWLHGjlz9UgboX9FUGhx_CQzo,2821
 prefect/utilities/names.py,sha256=x-stHcF7_tebJPvB1dz-5FvdXJXNBTg2kFZXSnIBBmk,1657
@@ -224,20 +224,20 @@ prefect/utilities/slugify.py,sha256=57Vb14t13F3zm1P65KAu8nVeAz0iJCd1Qc5eMG-R5y8,
 prefect/utilities/templating.py,sha256=nAiOGMMHGbIDFkGYy-g-dzcbY311WfycdgAhsM3cLpY,13298
 prefect/utilities/text.py,sha256=JRs6WNUn07i5oRP_jyTUYD6I0hdVwyMI0JmJc8f9gFw,731
 prefect/utilities/timeout.py,sha256=BRDIOWnqcw3B7X9tIk83Y3n9nQrJzZgduDQ63z-ns8w,1286
-prefect/utilities/urls.py,sha256=Wj7Q2vSSc11lX6FC0Wd295uTHDArDEa3rHJ967JJfPs,8919
+prefect/utilities/urls.py,sha256=NQqtifJRoewrXXjYmxC7YAkaMb1nE00evwjxO40Gmoc,8892
 prefect/utilities/visualization.py,sha256=Lum4IvLQ0nHsdLt6GGzS3Wwo-828u1rmOKc5mmWu994,6502
 prefect/utilities/schema_tools/__init__.py,sha256=KsFsTEHQqgp89TkDpjggkgBBywoHQPxbx-m6YQhiNEI,322
 prefect/utilities/schema_tools/hydration.py,sha256=k12qVCdLLrK-mNo1hPCdhxM5f_N14Nj0vJdtiWYWffk,8858
 prefect/utilities/schema_tools/validation.py,sha256=2GCjxwApTFwzey40ul9OkcAXrU3r-kWK__9ucMo0qbk,9744
 prefect/workers/__init__.py,sha256=8dP8SLZbWYyC_l9DRTQSE3dEbDgns5DZDhxkp_NfsbQ,35
-prefect/workers/base.py,sha256=bgFZWymeu4RMUDH97R7tMPmrJSoLKVPZCoGAPkjHfVk,48898
+prefect/workers/base.py,sha256=7Gb19vDG1cFsyrfsvZIR1h-OvX5T3RFrSS0Vj7uO5JU,49209
 prefect/workers/block.py,sha256=BOVVY5z-vUIQ2u8LwMTXDaNys2fjOZSS5YGDwJmTQjI,230
 prefect/workers/cloud.py,sha256=BOVVY5z-vUIQ2u8LwMTXDaNys2fjOZSS5YGDwJmTQjI,230
 prefect/workers/process.py,sha256=tcJ3fbiraLCfpVGpv8dOHwMSfVzeD_kyguUOvPuIz6I,19796
 prefect/workers/server.py,sha256=lgh2FfSuaNU7b6HPxSFm8JtKvAvHsZGkiOo4y4tW1Cw,2022
 prefect/workers/utilities.py,sha256=VfPfAlGtTuDj0-Kb8WlMgAuOfgXCdrGAnKMapPSBrwc,2483
-prefect_client-3.1.2.dist-info/LICENSE,sha256=MCxsn8osAkzfxKC4CC_dLcUkU8DZLkyihZ8mGs3Ah3Q,11357
-prefect_client-3.1.2.dist-info/METADATA,sha256=zs_vo1177_sGYQJnaNXeYU34A0hvA6hHMWzOLIbJ7CE,7338
-prefect_client-3.1.2.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
-prefect_client-3.1.2.dist-info/top_level.txt,sha256=MJZYJgFdbRc2woQCeB4vM6T33tr01TmkEhRcns6H_H4,8
-prefect_client-3.1.2.dist-info/RECORD,,
+prefect_client-3.1.3.dist-info/LICENSE,sha256=MCxsn8osAkzfxKC4CC_dLcUkU8DZLkyihZ8mGs3Ah3Q,11357
+prefect_client-3.1.3.dist-info/METADATA,sha256=PdG3F-fwco9gK2XcOTh-Os4oyp_5HzQSR5tx_BMPXiM,7386
+prefect_client-3.1.3.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
+prefect_client-3.1.3.dist-info/top_level.txt,sha256=MJZYJgFdbRc2woQCeB4vM6T33tr01TmkEhRcns6H_H4,8
+prefect_client-3.1.3.dist-info/RECORD,,