prefect-client 3.6.28.dev5__tar.gz → 3.6.28.dev6__tar.gz

{prefect_client-3.6.28.dev5 → prefect_client-3.6.28.dev6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prefect-client
-Version: 3.6.28.dev5
+Version: 3.6.28.dev6
 Summary: Workflow orchestration and management.
 Project-URL: Changelog, https://github.com/PrefectHQ/prefect/releases
 Project-URL: Documentation, https://docs.prefect.io
@@ -40,7 +40,7 @@ Requires-Dist: jsonpatch<2.0,>=1.32
 Requires-Dist: jsonschema<5.0.0,>=4.18.0
 Requires-Dist: opentelemetry-api<2.0.0,>=1.27.0
 Requires-Dist: orjson<4.0,>=3.7
-Requires-Dist: packaging<26.1,>=21.3
+Requires-Dist: packaging<26.2,>=21.3
 Requires-Dist: pathspec>=0.8.0
 Requires-Dist: pendulum<4,>=3.1.0; python_version < '3.13'
 Requires-Dist: prometheus-client>=0.20.0
@@ -53,7 +53,7 @@ Requires-Dist: python-slugify<9.0,>=5.0
 Requires-Dist: pytz<2027,>=2021.1
 Requires-Dist: pyyaml<7.0.0,>=5.4.1
 Requires-Dist: rfc3339-validator<0.2.0,>=0.1.4
-Requires-Dist: rich<15.0,>=11.0
+Requires-Dist: rich<16.0,>=11.0
 Requires-Dist: ruamel-yaml-clib>=0.2.8; platform_python_implementation == 'CPython'
 Requires-Dist: ruamel-yaml>=0.17.0
 Requires-Dist: sniffio<2.0.0,>=1.3.0

{prefect_client-3.6.28.dev5 → prefect_client-3.6.28.dev6}/pyproject.toml

@@ -44,7 +44,7 @@ dependencies = [
     "jsonschema>=4.18.0,<5.0.0",
     "opentelemetry-api>=1.27.0,<2.0.0",
     "orjson>=3.7,<4.0",
-    "packaging>=21.3,<26.1",
+    "packaging>=21.3,<26.2",
     "pathspec>=0.8.0",
     "pendulum>=3.1.0,<4; python_version<'3.13'",
     "prometheus-client>=0.20.0",
@@ -57,7 +57,7 @@ dependencies = [
     "pytz>=2021.1,<2027",
     "pyyaml>=5.4.1,<7.0.0",
     "rfc3339-validator>=0.1.4,<0.2.0",
-    "rich>=11.0,<15.0",
+    "rich>=11.0,<16.0",
     "ruamel.yaml>=0.17.0",
     "ruamel.yaml.clib>=0.2.8; platform_python_implementation == 'CPython'",
     "sniffio>=1.3.0,<2.0.0",

prefect_client-3.6.28.dev6/src/prefect/_build_info.py

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.6.28.dev6"
+__build_date__ = "2026-04-24 08:47:20.560734+00:00"
+__git_commit__ = "7ef768043f0f928ed47fa41215a792303d6b962c"
+__dirty__ = False

{prefect_client-3.6.28.dev5 → prefect_client-3.6.28.dev6}/src/prefect/_internal/pydantic/v2_schema.py

@@ -5,6 +5,7 @@ import typing as t
 import pydantic
 from pydantic import BaseModel as V2BaseModel
 from pydantic import ConfigDict, PydanticUndefinedAnnotation, create_model
+from pydantic.errors import PydanticUserError
 from pydantic.fields import FieldInfo
 from pydantic.type_adapter import TypeAdapter
 
@@ -115,11 +116,19 @@ def create_v2_schema(
         raise TypeError(exc.message)
 
     # root model references under #definitions
-    schema = adapter.json_schema(
-        by_alias=True,
-        ref_template="#/definitions/{model}",
-        schema_generator=GenerateEmptySchemaForUserClasses,
-    )
+    try:
+        schema = adapter.json_schema(
+            by_alias=True,
+            ref_template="#/definitions/{model}",
+            schema_generator=GenerateEmptySchemaForUserClasses,
+        )
+    except PydanticUserError as exc:
+        # In Pydantic >=2.13, unresolved forward references defer the error
+        # until the schema is built. Surface these as TypeError, matching the
+        # behavior parameter_schema expects for invalid parameter types.
+        if exc.code == "class-not-fully-defined":
+            raise TypeError(str(exc))
+        raise
     # ensure backwards compatibility by copying $defs into definitions
     if "$defs" in schema:
         schema["definitions"] = schema["$defs"]

{prefect_client-3.6.28.dev5 → prefect_client-3.6.28.dev6}/src/prefect/futures.py

@@ -194,6 +194,44 @@ class PrefectConcurrentFuture(PrefectWrappedFuture[R, concurrent.futures.Future[
     when the task run is submitted to a ThreadPoolExecutor.
     """
 
+    def __init__(
+        self,
+        task_run_id: uuid.UUID,
+        wrapped_future: concurrent.futures.Future[R],
+    ):
+        super().__init__(task_run_id, wrapped_future)
+        self._prefect_done_callbacks: list[Callable[[PrefectFuture[R]], None]] = []
+        self._prefect_done_callbacks_lock = threading.Lock()
+        self._prefect_done_event = threading.Event()
+
+        # Register Prefect's internal completion notifier before any user callbacks
+        # so wait()/as_completed() do not depend on callback chain ordering.
+        self._wrapped_future.add_done_callback(self._notify_prefect_done_callbacks)
+
+    def _add_prefect_done_callback(
+        self, fn: Callable[[PrefectFuture[R]], None]
+    ) -> None:
+        with self._prefect_done_callbacks_lock:
+            if self._prefect_done_event.is_set():
+                call_immediately = True
+            else:
+                self._prefect_done_callbacks.append(fn)
+                call_immediately = False
+
+        if call_immediately:
+            fn(self)
+
+    def _notify_prefect_done_callbacks(
+        self, future: concurrent.futures.Future[R]
+    ) -> None:
+        with self._prefect_done_callbacks_lock:
+            self._prefect_done_event.set()
+            callbacks = self._prefect_done_callbacks[:]
+            self._prefect_done_callbacks.clear()
+
+        for callback in callbacks:
+            callback(self)
+
     def wait(self, timeout: float | None = None) -> None:
         try:
             result = self._wrapped_future.result(timeout=timeout)
@@ -538,6 +576,17 @@ class PrefectFutureList(list[PrefectFuture[R]], Iterator[PrefectFuture[R]]):
         return results
 
 
+def _register_prefect_done_callback(
+    future: PrefectFuture[R], callback: Callable[[PrefectFuture[R]], None]
+) -> None:
+    add_prefect_done_callback = getattr(future, "_add_prefect_done_callback", None)
+    if add_prefect_done_callback is not None:
+        add_prefect_done_callback(callback)
+        return
+
+    future.add_done_callback(callback)
+
+
 def as_completed(
     futures: list[PrefectFuture[R]], timeout: float | None = None
 ) -> Generator[PrefectFuture[R], None]:
@@ -560,7 +609,7 @@ def as_completed(
                     finished_event.set()
 
             for future in pending:
-                future.add_done_callback(add_to_done)
+                _register_prefect_done_callback(future, add_to_done)
 
             while pending:
                 finished_event.wait()
@@ -651,7 +700,7 @@ def wait(
 
             # Add callbacks to all pending futures
             for future in not_done:
-                future.add_done_callback(mark_done)
+                _register_prefect_done_callback(future, mark_done)
 
             # Wait for futures to complete within timeout
             while not_done:

prefect_client-3.6.28.dev5/src/prefect/_build_info.py

@@ -1,5 +0,0 @@
-# Generated by versioningit
-__version__ = "3.6.28.dev5"
-__build_date__ = "2026-04-23 08:42:43.538152+00:00"
-__git_commit__ = "529fa0a6b6c26cdcdcfbe679a1d0e3312307e79b"
-__dirty__ = False