prefect-client 2.20.8__tar.gz → 2.20.10__tar.gz

{prefect-client-2.20.8/src/prefect_client.egg-info → prefect_client-2.20.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prefect-client
-Version: 2.20.8
+Version: 2.20.10
 Summary: Workflow orchestration and management.
 Home-page: https://www.prefect.io
 Author: Prefect Technologies, Inc.
@@ -21,8 +21,46 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Topic :: Software Development :: Libraries
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
-Provides-Extra: notifications
 License-File: LICENSE
+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: cloudpickle<4.0,>=2.0
+Requires-Dist: coolname<3.0.0,>=1.0.4
+Requires-Dist: croniter<3.0.0,>=1.0.12
+Requires-Dist: fsspec>=2022.5.0
+Requires-Dist: exceptiongroup>=1.2.1
+Requires-Dist: graphviz>=0.20.1
+Requires-Dist: griffe<2.0.0,>=0.49.0
+Requires-Dist: httpcore<2.0.0,>=1.0.5
+Requires-Dist: httpx[http2]!=0.23.2,>=0.23
+Requires-Dist: importlib_metadata>=4.4; python_version < "3.10"
+Requires-Dist: importlib-resources<6.5.0,>=6.1.3
+Requires-Dist: jsonpatch<2.0,>=1.32
+Requires-Dist: jsonschema<5.0.0,>=4.0.0
+Requires-Dist: orjson<4.0,>=3.7
+Requires-Dist: packaging<24.3,>=21.3
+Requires-Dist: pathspec>=0.8.0
+Requires-Dist: pendulum<3.0; python_version < "3.12"
+Requires-Dist: pendulum<4,>=3.0.0; python_version >= "3.12"
+Requires-Dist: pydantic[email]!=2.0.0,!=2.0.1,!=2.1.0,<3.0.0,>=1.10.0
+Requires-Dist: pydantic_core<3.0.0,>=2.12.0
+Requires-Dist: python_dateutil<3.0.0,>=2.8.2
+Requires-Dist: python-slugify<9.0,>=5.0
+Requires-Dist: pyyaml<7.0.0,>=5.4.1
+Requires-Dist: rfc3339-validator<0.2.0,>=0.1.4
+Requires-Dist: rich<14.0,>=11.0
+Requires-Dist: ruamel.yaml>=0.17.0
+Requires-Dist: sniffio<2.0.0,>=1.3.0
+Requires-Dist: toml>=0.10.0
+Requires-Dist: typing_extensions<5.0.0,>=4.5.0
+Requires-Dist: ujson<6.0.0,>=5.8.0
+Requires-Dist: uvicorn!=0.29.0,>=0.14.0
+Requires-Dist: websockets<14.0,>=10.4
+Requires-Dist: itsdangerous
+Requires-Dist: python-multipart>=0.0.7
+Provides-Extra: notifications
+Requires-Dist: apprise<2.0.0,>=1.1.0; extra == "notifications"
 
 <p align="center"><img src="https://github.com/PrefectHQ/prefect/assets/3407835/c654cbc6-63e8-4ada-a92a-efd2f8f24b85" width=1000></p>
 

{prefect-client-2.20.8 → prefect_client-2.20.10}/requirements-client.txt

@@ -21,7 +21,7 @@ pathspec >= 0.8.0
 pendulum < 3.0; python_version < '3.12'
 pendulum >= 3.0.0, <4; python_version >= '3.12'
 # the version constraints for pydantic are merged with those from fastapi 0.103.2
-pydantic[email]>=1.10.0,!=2.0.0,!=2.0.1,!=2.1.0,<2.9.0
+pydantic[email]>=1.10.0,!=2.0.0,!=2.0.1,!=2.1.0,<3.0.0
 pydantic_core >= 2.12.0, < 3.0.0
 python_dateutil >= 2.8.2, < 3.0.0
 python-slugify >= 5.0, < 9.0

{prefect-client-2.20.8 → prefect_client-2.20.10}/requirements.txt

@@ -12,7 +12,7 @@ graphviz >= 0.20.1
 jinja2 >= 3.0.0, < 4.0.0
 jinja2-humanize-extension >= 0.4.0
 humanize >= 4.9.0, < 5.0.0
-kubernetes >= 24.2.0, < 30.0.0
+kubernetes >= 24.2.0, < 32.0.0
 pytz >= 2021.1, < 2025
 readchar >= 4.0.0, < 5.0.0
 sqlalchemy[asyncio] >= 1.4.22, != 1.4.33, < 3.0.0

{prefect-client-2.20.8 → prefect_client-2.20.10}/src/prefect/_internal/pydantic/v2_schema.py

@@ -9,6 +9,8 @@ import pydantic
 from pydantic import BaseModel as V2BaseModel
 from pydantic import ConfigDict, PydanticUndefinedAnnotation, create_model
 from pydantic.type_adapter import TypeAdapter
+from pydantic.v1 import BaseModel as V1BaseModel
+from pydantic.v1 import SecretStr as V1SecretStr
 
 from prefect._internal.pydantic.annotations.pendulum import (
     PydanticPendulumDateTimeType,
@@ -30,6 +32,18 @@ def is_v2_model(v) -> bool:
     return False
 
 
+def is_v1_model(v) -> bool:
+    if isinstance(v, V1BaseModel):
+        return True
+    try:
+        if inspect.isclass(v) and issubclass(v, V1BaseModel):
+            return True
+    except TypeError:
+        pass
+
+    return False
+
+
 def is_v2_type(v) -> bool:
     if is_v2_model(v):
         return True
@@ -40,6 +54,18 @@ def is_v2_type(v) -> bool:
         return False
 
 
+def is_v1_type(v) -> bool:
+    if is_v1_model(v):
+        return True
+
+    try:
+        return v.__module__.startswith("pydantic.v1.types") or isinstance(
+            v, V1SecretStr
+        )
+    except AttributeError:
+        return False
+
+
 def has_v2_type_as_param(signature: inspect.Signature) -> bool:
     parameters = signature.parameters.values()
     for p in parameters:

{prefect-client-2.20.8 → prefect_client-2.20.10}/src/prefect/blocks/notifications.py

@@ -281,6 +281,27 @@ class PagerDutyWebHook(AbstractAppriseNotificationBlock):
         )
         self._start_apprise_client(url)
 
+    @sync_compatible
+    async def notify(
+        self,
+        body: str,
+        subject: Optional[str] = None,
+    ):
+        """
+        Apprise will combine subject and body by default, so we need to move
+        the body into the custom_details field. custom_details is part of the
+        webhook url, so we need to update the url and restart the client.
+        """
+        if subject:
+            self.custom_details = self.custom_details or {}
+            self.custom_details.update(
+                {"Prefect Notification Body": body.replace(" ", "%20")}
+            )
+            body = " "
+            self.block_initialization()
+
+        await super().notify(body, subject)
+
 
 class TwilioSMS(AbstractAppriseNotificationBlock):
     """Enables sending notifications via Twilio SMS.

{prefect-client-2.20.8 → prefect_client-2.20.10}/src/prefect/client/orchestration.py

@@ -8,12 +8,14 @@ from typing import (
     Dict,
     Iterable,
     List,
+    Literal,
     NoReturn,
     Optional,
     Set,
     Tuple,
     TypeVar,
     Union,
+    overload,
 )
 from uuid import UUID, uuid4
 
@@ -178,6 +180,20 @@ class ServerType(AutoEnum):
         return PREFECT_EXPERIMENTAL_EVENTS and PREFECT_API_SERVICES_TRIGGERS_ENABLED
 
 
+@overload
+def get_client(
+    httpx_settings: Optional[Dict[str, Any]] = None, sync_client: Literal[False] = False
+) -> "PrefectClient":
+    ...
+
+
+@overload
+def get_client(
+    httpx_settings: Optional[Dict[str, Any]] = None, sync_client: Literal[True] = True
+) -> "SyncPrefectClient":
+    ...
+
+
 def get_client(
     httpx_settings: Optional[Dict[str, Any]] = None, sync_client: bool = False
 ) -> Union["PrefectClient", "SyncPrefectClient"]:

{prefect-client-2.20.8 → prefect_client-2.20.10}/src/prefect/events/clients.py

@@ -473,11 +473,11 @@ class PrefectEventSubscriber:
                 f"Reason: {e.args[0]}"
             )
         except ConnectionClosedError as e:
-            raise Exception(
-                "Unable to authenticate to the event stream. Please ensure the "
-                "provided api_key you are using is valid for this environment. "
-                f"Reason: {e.reason}"
-            ) from e
+            reason = getattr(e.rcvd, "reason", None)
+            msg = "Unable to authenticate to the event stream. Please ensure the "
+            msg += "provided api_key you are using is valid for this environment. "
+            msg += f"Reason: {reason}" if reason else ""
+            raise Exception(msg) from e
 
         from prefect.events.filters import EventOccurredFilter
 

{prefect-client-2.20.8 → prefect_client-2.20.10}/src/prefect/flows.py

@@ -45,11 +45,15 @@ from prefect._internal.pydantic import HAS_PYDANTIC_V2
 
 if HAS_PYDANTIC_V2:
     import pydantic.v1 as pydantic
+    from pydantic import BaseModel as V2BaseModel
     from pydantic import ValidationError as V2ValidationError
     from pydantic.v1 import BaseModel as V1BaseModel
     from pydantic.v1.decorator import ValidatedFunction as V1ValidatedFunction
 
-    from ._internal.pydantic.v2_schema import is_v2_type
+    from ._internal.pydantic.v2_schema import (
+        is_v1_type,
+        is_v2_type,
+    )
     from ._internal.pydantic.v2_validated_func import V2ValidatedFunction
     from ._internal.pydantic.v2_validated_func import (
         V2ValidatedFunction as ValidatedFunction,
@@ -339,6 +343,16 @@ class Flow(Generic[P, R]):
             # is not picklable in some environments
             try:
                 ValidatedFunction(self.fn, config={"arbitrary_types_allowed": True})
+            except TypeError as exc:
+                if (
+                    HAS_PYDANTIC_V2
+                    and "`__modify_schema__` method is not supported" in str(exc)
+                ):
+                    V1ValidatedFunction(
+                        self.fn, config={"arbitrary_types_allowed": True}
+                    )
+                else:
+                    raise
             except pydantic.ConfigError as exc:
                 raise ValueError(
                     "Flow function is not compatible with `validate_parameters`. "
@@ -510,19 +524,33 @@ class Flow(Generic[P, R]):
         args, kwargs = parameters_to_args_kwargs(self.fn, parameters)
 
         if HAS_PYDANTIC_V2:
-            has_v1_models = any(isinstance(o, V1BaseModel) for o in args) or any(
-                isinstance(o, V1BaseModel) for o in kwargs.values()
+            sig = inspect.signature(self.fn)
+
+            has_v1_types = any(
+                is_v1_type(param.annotation) for param in sig.parameters.values()
+            )
+            has_v1_models = any(
+                issubclass(param.annotation, V1BaseModel)
+                if isinstance(param.annotation, type)
+                else False
+                for param in sig.parameters.values()
             )
-            has_v2_types = any(is_v2_type(o) for o in args) or any(
-                is_v2_type(o) for o in kwargs.values()
+            has_v2_types = any(
+                is_v2_type(param.annotation) for param in sig.parameters.values()
+            )
+            has_v2_models = any(
+                issubclass(param.annotation, V2BaseModel)
+                if isinstance(param.annotation, type)
+                else False
+                for param in sig.parameters.values()
             )
 
-            if has_v1_models and has_v2_types:
+            if (has_v1_types and has_v2_types) or (has_v1_models and has_v2_models):
                 raise ParameterTypeError(
                     "Cannot mix Pydantic v1 and v2 types as arguments to a flow."
                 )
 
-            if has_v1_models:
+            if has_v1_models or has_v1_types:
                 validated_fn = V1ValidatedFunction(
                     self.fn, config={"arbitrary_types_allowed": True}
                 )
@@ -530,7 +558,6 @@ class Flow(Generic[P, R]):
                 validated_fn = V2ValidatedFunction(
                     self.fn, config={"arbitrary_types_allowed": True}
                 )
-
         else:
             validated_fn = ValidatedFunction(
                 self.fn, config={"arbitrary_types_allowed": True}
@@ -1148,6 +1175,15 @@ class Flow(Generic[P, R]):
     ) -> T:
         ...
 
+    @overload
+    def __call__(
+        self: "Flow[P, Coroutine[Any, Any, T]]",
+        *args: P.args,
+        return_state: Literal[True],
+        **kwargs: P.kwargs,
+    ) -> Awaitable[State[T]]:
+        ...
+
     @overload
     def __call__(
         self: "Flow[P, T]",

{prefect-client-2.20.8 → prefect_client-2.20.10/src/prefect_client.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prefect-client
-Version: 2.20.8
+Version: 2.20.10
 Summary: Workflow orchestration and management.
 Home-page: https://www.prefect.io
 Author: Prefect Technologies, Inc.
@@ -21,8 +21,46 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Topic :: Software Development :: Libraries
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
-Provides-Extra: notifications
 License-File: LICENSE
+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: cloudpickle<4.0,>=2.0
+Requires-Dist: coolname<3.0.0,>=1.0.4
+Requires-Dist: croniter<3.0.0,>=1.0.12
+Requires-Dist: fsspec>=2022.5.0
+Requires-Dist: exceptiongroup>=1.2.1
+Requires-Dist: graphviz>=0.20.1
+Requires-Dist: griffe<2.0.0,>=0.49.0
+Requires-Dist: httpcore<2.0.0,>=1.0.5
+Requires-Dist: httpx[http2]!=0.23.2,>=0.23
+Requires-Dist: importlib_metadata>=4.4; python_version < "3.10"
+Requires-Dist: importlib-resources<6.5.0,>=6.1.3
+Requires-Dist: jsonpatch<2.0,>=1.32
+Requires-Dist: jsonschema<5.0.0,>=4.0.0
+Requires-Dist: orjson<4.0,>=3.7
+Requires-Dist: packaging<24.3,>=21.3
+Requires-Dist: pathspec>=0.8.0
+Requires-Dist: pendulum<3.0; python_version < "3.12"
+Requires-Dist: pendulum<4,>=3.0.0; python_version >= "3.12"
+Requires-Dist: pydantic[email]!=2.0.0,!=2.0.1,!=2.1.0,<3.0.0,>=1.10.0
+Requires-Dist: pydantic_core<3.0.0,>=2.12.0
+Requires-Dist: python_dateutil<3.0.0,>=2.8.2
+Requires-Dist: python-slugify<9.0,>=5.0
+Requires-Dist: pyyaml<7.0.0,>=5.4.1
+Requires-Dist: rfc3339-validator<0.2.0,>=0.1.4
+Requires-Dist: rich<14.0,>=11.0
+Requires-Dist: ruamel.yaml>=0.17.0
+Requires-Dist: sniffio<2.0.0,>=1.3.0
+Requires-Dist: toml>=0.10.0
+Requires-Dist: typing_extensions<5.0.0,>=4.5.0
+Requires-Dist: ujson<6.0.0,>=5.8.0
+Requires-Dist: uvicorn!=0.29.0,>=0.14.0
+Requires-Dist: websockets<14.0,>=10.4
+Requires-Dist: itsdangerous
+Requires-Dist: python-multipart>=0.0.7
+Provides-Extra: notifications
+Requires-Dist: apprise<2.0.0,>=1.1.0; extra == "notifications"
 
 <p align="center"><img src="https://github.com/PrefectHQ/prefect/assets/3407835/c654cbc6-63e8-4ada-a92a-efd2f8f24b85" width=1000></p>
 

{prefect-client-2.20.8 → prefect_client-2.20.10}/src/prefect_client.egg-info/SOURCES.txt

@@ -95,6 +95,7 @@ src/prefect/_vendor/fastapi/exceptions.py
 src/prefect/_vendor/fastapi/logger.py
 src/prefect/_vendor/fastapi/param_functions.py
 src/prefect/_vendor/fastapi/params.py
+src/prefect/_vendor/fastapi/py.typed
 src/prefect/_vendor/fastapi/requests.py
 src/prefect/_vendor/fastapi/responses.py
 src/prefect/_vendor/fastapi/routing.py
@@ -140,6 +141,7 @@ src/prefect/_vendor/starlette/datastructures.py
 src/prefect/_vendor/starlette/endpoints.py
 src/prefect/_vendor/starlette/exceptions.py
 src/prefect/_vendor/starlette/formparsers.py
+src/prefect/_vendor/starlette/py.typed
 src/prefect/_vendor/starlette/requests.py
 src/prefect/_vendor/starlette/responses.py
 src/prefect/_vendor/starlette/routing.py
@@ -299,4 +301,30 @@ src/prefect_client.egg-info/PKG-INFO
 src/prefect_client.egg-info/SOURCES.txt
 src/prefect_client.egg-info/dependency_links.txt
 src/prefect_client.egg-info/requires.txt
-src/prefect_client.egg-info/top_level.txt
+src/prefect_client.egg-info/top_level.txt
+tests/test_artifacts.py
+tests/test_automations.py
+tests/test_background_tasks.py
+tests/test_context.py
+tests/test_deployments.py
+tests/test_engine.py
+tests/test_exceptions.py
+tests/test_filesystems.py
+tests/test_flow_runs.py
+tests/test_flows.py
+tests/test_flows_compat.py
+tests/test_futures.py
+tests/test_highlighters.py
+tests/test_import_interceptor.py
+tests/test_log_prints.py
+tests/test_logging.py
+tests/test_new_flow_engine.py
+tests/test_new_task_engine.py
+tests/test_plugins.py
+tests/test_serializers.py
+tests/test_settings.py
+tests/test_states.py
+tests/test_task_runners.py
+tests/test_task_server.py
+tests/test_tasks.py
+tests/test_variables.py

{prefect-client-2.20.8 → prefect_client-2.20.10}/src/prefect_client.egg-info/requires.txt

@@ -16,7 +16,7 @@ jsonschema<5.0.0,>=4.0.0
 orjson<4.0,>=3.7
 packaging<24.3,>=21.3
 pathspec>=0.8.0
-pydantic[email]!=2.0.0,!=2.0.1,!=2.1.0,<2.9.0,>=1.10.0
+pydantic[email]!=2.0.0,!=2.0.1,!=2.1.0,<3.0.0,>=1.10.0
 pydantic_core<3.0.0,>=2.12.0
 python_dateutil<3.0.0,>=2.8.2
 python-slugify<9.0,>=5.0