prefect-client 3.0.4__py3-none-any.whl → 3.0.5__py3-none-any.whl

prefect/transactions.py

@@ -284,11 +284,14 @@ class Transaction(ContextModel):
         self.children.append(transaction)
 
     def get_parent(self) -> Optional["Transaction"]:
-        prev_var = getattr(self._token, "old_value")
-        if prev_var != Token.MISSING:
-            parent = prev_var
+        parent = None
+        if self._token:
+            prev_var = getattr(self._token, "old_value")
+            if prev_var != Token.MISSING:
+                parent = prev_var
         else:
-            parent = None
+            # `_token` has been reset so we need to get the active transaction from the context var
+            parent = self.get_active()
         return parent
 
     def commit(self) -> bool:

prefect/types/__init__.py

@@ -1,17 +1,23 @@
-from typing import Annotated, Any, Dict, List, Union
+from functools import partial
+from typing import Annotated, Any, Dict, List, Set, TypeVar, Union
+from typing_extensions import Literal
 import orjson
 import pydantic
 
 from pydantic import (
     BeforeValidator,
     Field,
+    SecretStr,
     StrictBool,
     StrictFloat,
     StrictInt,
     StrictStr,
+    TypeAdapter,
 )
 from zoneinfo import available_timezones
 
+T = TypeVar("T")
+
 MAX_VARIABLE_NAME_LENGTH = 255
 MAX_VARIABLE_VALUE_LENGTH = 5000
 
@@ -82,11 +88,54 @@ StrictVariableValue = Annotated[VariableValue, BeforeValidator(check_variable_va
 LaxUrl = Annotated[str, BeforeValidator(lambda x: str(x).strip())]
 
 
+StatusCode = Annotated[int, Field(ge=100, le=599)]
+
+
 class SecretDict(pydantic.Secret[Dict[str, Any]]):
     pass
 
 
+def validate_set_T_from_delim_string(
+    value: Union[str, T, Set[T], None], type_, delim=None
+) -> Set[T]:
+    """
+    "no-info" before validator useful in scooping env vars
+
+    e.g. `PREFECT_CLIENT_RETRY_EXTRA_CODES=429,502,503` -> `{429, 502, 503}`
+    e.g. `PREFECT_CLIENT_RETRY_EXTRA_CODES=429` -> `{429}`
+    """
+    if not value:
+        return set()
+
+    T_adapter = TypeAdapter(type_)
+    delim = delim or ","
+    if isinstance(value, str):
+        return {T_adapter.validate_strings(s) for s in value.split(delim)}
+    errors = []
+    try:
+        return {T_adapter.validate_python(value)}
+    except pydantic.ValidationError as e:
+        errors.append(e)
+    try:
+        return TypeAdapter(Set[type_]).validate_python(value)
+    except pydantic.ValidationError as e:
+        errors.append(e)
+    raise ValueError(f"Invalid set[{type_}]: {errors}")
+
+
+ClientRetryExtraCodes = Annotated[
+    Union[str, StatusCode, Set[StatusCode], None],
+    BeforeValidator(partial(validate_set_T_from_delim_string, type_=StatusCode)),
+]
+
+LogLevel = Annotated[
+    Literal["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"],
+    BeforeValidator(lambda x: x.upper()),
+]
+
 __all__ = [
+    "ClientRetryExtraCodes",
+    "LogLevel",
     "NonNegativeInteger",
     "PositiveInteger",
     "NonNegativeFloat",
@@ -94,5 +143,6 @@ __all__ = [
     "NameOrEmpty",
     "NonEmptyishName",
     "SecretDict",
+    "StatusCode",
     "StrictVariableValue",
 ]

prefect/utilities/pydantic.py

@@ -26,6 +26,7 @@ from typing_extensions import Literal
 
 from prefect.utilities.dispatch import get_dispatch_key, lookup_type, register_base_type
 from prefect.utilities.importtools import from_qualified_name, to_qualified_name
+from prefect.utilities.names import obfuscate
 
 D = TypeVar("D", bound=Any)
 M = TypeVar("M", bound=BaseModel)
@@ -371,7 +372,7 @@ def handle_secret_render(value: object, context: dict[str, Any]) -> object:
         return (
             cast(Secret[object], value).get_secret_value()
             if context.get("include_secrets", False)
-            else "**********"
+            else obfuscate(value)
         )
     elif isinstance(value, BaseModel):
         return value.model_dump(context=context)

prefect/utilities/text.py

@@ -1,4 +1,5 @@
-from typing import Optional
+import difflib
+from typing import Iterable, Optional
 
 
 def truncated_to(length: int, value: Optional[str]) -> str:
@@ -17,3 +18,14 @@ def truncated_to(length: int, value: Optional[str]) -> str:
     proposed = f"{beginning}...{omitted} additional characters...{end}"
 
     return proposed if len(proposed) < len(value) else value
+
+
+def fuzzy_match_string(
+    word: str,
+    possibilities: Iterable[str],
+    *,
+    n: int = 1,
+    cutoff: float = 0.6,
+) -> Optional[str]:
+    match = difflib.get_close_matches(word, possibilities, n=n, cutoff=cutoff)
+    return match[0] if match else None

{prefect_client-3.0.4.dist-info → prefect_client-3.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prefect-client
-Version: 3.0.4
+Version: 3.0.5
 Summary: Workflow orchestration and management.
 Home-page: https://www.prefect.io
 Author: Prefect Technologies, Inc.

{prefect_client-3.0.4.dist-info → prefect_client-3.0.5.dist-info}/RECORD

@@ -4,29 +4,29 @@ prefect/_version.py,sha256=I9JsXwt7BjAAbMEZgtmE3a6dJ2jqV-wqWto9D6msb3k,24597
 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=thYNj0CcJjM4TJQvXsLKTIQl7t0qjEnSWzxPWPONcRw,9118
-prefect/context.py,sha256=J4GS70ZG_dkJ2v_dQWkdbuiN88iumFpoJhTu3hg7d60,21821
+prefect/cache_policies.py,sha256=PWUzyJue4h5XHVeIVolfPKhRGrx1hyWJt58AJyHbcqU,9104
+prefect/context.py,sha256=4CsGsfn8Kt8SHcTor5vhfcSIRm3qcQkO9vb9rfsKD3E,21571
 prefect/engine.py,sha256=BpmDbe6miZcTl1vRkxfCPYcWSXADLigGPCagFwucMz0,1976
-prefect/exceptions.py,sha256=ondjUe0fIXXjhoFnqg8twqgLoPMR02HuQv5Az-kSG50,11348
+prefect/exceptions.py,sha256=V_nRpS2Z93PvJMoQdXbx8zepVaFb-pWanCqVi7T1ngI,11803
 prefect/filesystems.py,sha256=CxwMmKY8LBUed_9IqE2jUqxVCWhXa1r2fjKgLbIC2Vg,17893
-prefect/flow_engine.py,sha256=Z6xOO1ONAGwVNcvyvEIkJv_LB0VE5iBptV4ZWgTFqbc,30000
+prefect/flow_engine.py,sha256=p1IoMa5okV0l-0KGjDxNDsR1N74K5oP_Lb3V0z7v49U,30076
 prefect/flow_runs.py,sha256=EaXRIQTOnwnA0fO7_EjwafFRmS57K_CRy0Xsz3JDIhc,16070
-prefect/flows.py,sha256=zgTnzasA8a1PiOGRRSUB1P2kIz5kNgaKMkPEI81TXcQ,89588
+prefect/flows.py,sha256=UxcyRGZElO4_ZwWUa5oa5POL6CL2N5M8FvrPUZrsOX4,89591
 prefect/futures.py,sha256=1Uq-Q3ommCHSku_lsASuP1s3yFuYoL980fGcHdCFg30,16298
 prefect/main.py,sha256=IdtnJR5-IwP8EZsfhMFKj92ylMhNyau9X_eMcTP2ZjM,2336
 prefect/plugins.py,sha256=HY7Z7OJlltqzsUiPMEL1Y_hQbHw0CeZKayWiK-k8DP4,2435
 prefect/profiles.toml,sha256=kTvqDNMzjH3fsm5OEI-NKY4dMmipor5EvQXRB6rPEjY,522
 prefect/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-prefect/results.py,sha256=dHaS_sHHi9CcHk5DmwONey2OQnVJkj7cLqoYmHaUB58,47377
+prefect/results.py,sha256=-V_JRaWeY2WXWhY2d_zL7KVIro660mIU6F3heNaih0o,47391
 prefect/serializers.py,sha256=Lo41EM0_qGzcfB_63390Izeo3DdK6cY6VZfxa9hpSGQ,8712
-prefect/settings.py,sha256=9T_JY0VIpnFly_BtDDihPT9ppukUQrOdUkqwEtJQa60,73501
+prefect/settings.py,sha256=032alJnUkrWxXnHFyZD-p6cPN09x9bIoK7q7qi1VSxs,73558
 prefect/states.py,sha256=2lysq6X5AvqPfE3eD3D0HYt-KpFA2OUgA0c4ZQ22A_U,24906
 prefect/task_engine.py,sha256=rcCPPrX01CxiOPhnf_7WcN0wGHbmB5VV7_OG7PKYOrY,57943
 prefect/task_runners.py,sha256=W1n0yMwbDIqnvffFVJADo9MGEbLaYkzWk52rqgnkMY4,15019
 prefect/task_runs.py,sha256=jkaQOkRKOHS8fgHUijteriFpjMSKv4zldn1D8tZHkUI,8777
 prefect/task_worker.py,sha256=a8Uw78Ms4p3ikt_la50lENmPLIa-jjbuvunvjVXvRKQ,16785
 prefect/tasks.py,sha256=35eOv7VfhziiC3hL9FxB3spYtG6tpxZBLzk5KP_8Ux8,68371
-prefect/transactions.py,sha256=oJKP4w5KjV1PSmN-ByyHN3bagSiKiTKvMVtLvAAfpAg,16387
+prefect/transactions.py,sha256=NTzflkehGQ5jKmuChpvsUv1-ZPBqLI7OmUeq-nZJGHQ,16558
 prefect/variables.py,sha256=023cfSj_ydwvz6lyChRKnjHFfkdoYZKK_zdTtuSxrYo,4665
 prefect/_internal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prefect/_internal/_logging.py,sha256=HvNHY-8P469o5u4LYEDBTem69XZEt1QUeUaLToijpak,810
@@ -35,7 +35,6 @@ prefect/_internal/pytz.py,sha256=WWl9x16rKFWequGmcOGs_ljpCDPf2LDHMyZp_4D8e6c,137
 prefect/_internal/retries.py,sha256=xtgj6oPSvYQLbyk451LR6swcRQvRVWEzCxY6GMK7qA4,2284
 prefect/_internal/compatibility/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prefect/_internal/compatibility/deprecated.py,sha256=PVME2C3Oe4_8tKIGufx1W4EpGkz5IQY8gFohPVOjNcM,7533
-prefect/_internal/compatibility/experimental.py,sha256=nrIeeAe1vZ0yMb1cPw5AroVR6_msx-bzTeBLzY4au6o,5634
 prefect/_internal/compatibility/migration.py,sha256=oifIj6mKwVUfrUO13ODLKOCwLON4YqaTXzhBFh8AK6E,6797
 prefect/_internal/concurrency/__init__.py,sha256=YlTwU9ryjPNwbJa45adLJY00t_DGCh1QrdtY9WdVFfw,2140
 prefect/_internal/concurrency/api.py,sha256=mOajv_f9ms2u3O4sgegJDV2kLeCg9lBavlEZzPw5kr4,7126
@@ -72,13 +71,13 @@ prefect/client/base.py,sha256=2K8UiWzorZNNM4c8c-OiGeZ5i5ViUfZ_Q31oPobbOO0,24956
 prefect/client/cloud.py,sha256=Wjm27jUG1K8UHb8sIamOqyAGlM26Oe9_OFpCO6x5s2E,6191
 prefect/client/collections.py,sha256=u-96saqu0RALAazRI0YaZCJahnuafMppY21KN6ggx80,1059
 prefect/client/constants.py,sha256=Z_GG8KF70vbbXxpJuqW5pLnwzujTVeHbcYYRikNmGH0,29
-prefect/client/orchestration.py,sha256=XImn-8TKOYJ8LBAZ83FEC4DOf0RP6WE9BeLpDXfYX4A,149371
+prefect/client/orchestration.py,sha256=qJp29dPR9OE6uSH3Tkl5uD4IKzQRsYv6dj3Ku-I-sSQ,150555
 prefect/client/subscriptions.py,sha256=oqF2MJsgN3psJg-MePfvwMtEWjromfP9StWF00xc1eg,3403
 prefect/client/utilities.py,sha256=89fmza0cRMOayxgXRdO51TKb11TczJ0ByOZmcZVrt44,3286
 prefect/client/schemas/__init__.py,sha256=KlyqFV-hMulMkNstBn_0ijoHoIwJZaBj6B1r07UmgvE,607
 prefect/client/schemas/actions.py,sha256=GT1VlvwV5koV690H7ViGFH3tpW7_PvDf0QJoYTcOLDg,28862
 prefect/client/schemas/filters.py,sha256=oYUBj59SC6keYHaQ8-qFaVynEAcHV8BABrQaob2mI6c,35864
-prefect/client/schemas/objects.py,sha256=zJGTbmFYiAORxJ3PB4EcRkOIVcQlxH4O91aOtjKBkyU,56149
+prefect/client/schemas/objects.py,sha256=hlFnM04FzH3HaKm_gcvu7AC1CZsTzVALCajIzosgQCg,56198
 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
@@ -99,7 +98,7 @@ prefect/concurrency/v1/sync.py,sha256=qKE0YzNbrmYooTwP7pz4m1BUz61THCUIF45_PE5IyY
 prefect/deployments/__init__.py,sha256=_wb7NxDKhq11z9MjYsPckmT3o6MRhGLRgCV9TmvYtew,1002
 prefect/deployments/base.py,sha256=OyaKZ1Uk16XtvABh5byO6I3jp_1FYG301ryjDq00qJE,16688
 prefect/deployments/deployments.py,sha256=EvC9qBdvJRc8CHJqRjFTqtzx75SE8bpZOl5C-2eULyA,109
-prefect/deployments/flow_runs.py,sha256=tH6lpEkgHhQ5Ipr0bhVAjN6AeOoDwY7UKrkbJihJ6D0,6567
+prefect/deployments/flow_runs.py,sha256=Pz6KYDKNPkgOnh4M2VhkiPhNtDQfuKBmqSjmYGaucbs,6812
 prefect/deployments/runner.py,sha256=b7jD1DHL7y2jeBXgdBfSsnBMJPHShs4Tt1c5jAeG5Dk,41823
 prefect/deployments/schedules.py,sha256=KCYA6dOmLAoElHZuoWqdJn4Yno4TtOZtXfPOpTLb1cE,2046
 prefect/deployments/steps/__init__.py,sha256=Dlz9VqMRyG1Gal8dj8vfGpPr0LyQhZdvcciozkK8WoY,206
@@ -137,10 +136,10 @@ prefect/locking/filesystem.py,sha256=GiZlLLj51cLH6QQgq7IeU6jUK6vGi0wMnOG0zaO95-c
 prefect/locking/memory.py,sha256=Y1fsMSUAk3jUILzRivbxlrE9Xv8OcVbaylVf-aiEGNc,7495
 prefect/locking/protocol.py,sha256=o5-48SxvEDAdVwW8RIn7rCN32CmvIsaVHTztESUXuHU,4232
 prefect/logging/__init__.py,sha256=zx9f5_dWrR4DbcTOFBpNGOPoCZ1QcPFudr7zxb2XRpA,148
-prefect/logging/configuration.py,sha256=bYqFJm0QgLz92Dub1Lbl3JONjjm0lTK149pNAGbxPdM,3467
+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=w532IhEH6890X3jlfwX3iYIxwJIM0WXDIHTT0K5e-U8,10453
+prefect/logging/handlers.py,sha256=8Ey7jJuhwYq1TG3lvsdMqWL_k7F-aL9DBFCkTUKLRSc,10417
 prefect/logging/highlighters.py,sha256=BpSXOy0n3lFVvlKWa7jC-HetAiClFi9jnQtEq5-rgok,1681
 prefect/logging/loggers.py,sha256=9fN5iTXQwBAHwKfE9iBo-90f2SyGs9Z3OKmkLPG91VY,10995
 prefect/logging/logging.yml,sha256=ALNY_E1i3E7tkagCB3Qg35IvuRBHt-t9QqVJvsSq5xA,3185
@@ -161,7 +160,7 @@ prefect/runtime/flow_run.py,sha256=33XsudriajxIRJtebPpqF8cWCRsUpmvb9V0CSipG4J0,9
 prefect/runtime/task_run.py,sha256=B6v_nZiHy9nKZfnKFQF7izZjAjaiZOT0j80m-VcLxmY,3403
 prefect/server/api/collections_data/views/aggregate-worker-metadata.json,sha256=gqrwGyylzBEzlFSPOJcMuUwdoK_zojpU0SZaBDgK5FE,79748
 prefect/server/api/static/prefect-logo-mark-gradient.png,sha256=ylRjJkI_JHCw8VbQasNnXQHwZW-sH-IQiUGSD3aWP1E,73430
-prefect/types/__init__.py,sha256=SAHJDtWEGidTKXQACJ38nj6fq8r57Gj0Pwo4Gy7pVWs,2234
+prefect/types/__init__.py,sha256=A714iHFE9makA5LiAKd0Y5wfdb7m13gBwvrpQzdLVgs,3647
 prefect/types/entrypoint.py,sha256=2FF03-wLPgtnqR_bKJDB2BsXXINPdu8ptY9ZYEZnXg8,328
 prefect/utilities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prefect/utilities/annotations.py,sha256=Ocj2s5zhnGr8uXUBnOli-OrybXVJdu4-uZvCRpKpV_Q,2820
@@ -179,12 +178,12 @@ prefect/utilities/importtools.py,sha256=aO-xhf2h2KzsLGvSKwRAZLB4ITeW9rsV0Ys-gwq3
 prefect/utilities/math.py,sha256=wLwcKVidpNeWQi1TUIWWLHGjlz9UgboX9FUGhx_CQzo,2821
 prefect/utilities/names.py,sha256=x-stHcF7_tebJPvB1dz-5FvdXJXNBTg2kFZXSnIBBmk,1657
 prefect/utilities/processutils.py,sha256=yo_GO48pZzgn4A0IK5irTAoqyUCYvWKDSqHXCrtP8c4,14547
-prefect/utilities/pydantic.py,sha256=YEY7hp5ptaYqOzsZJC4dXf9d2g37aOdepoH8FBPg7uw,12394
+prefect/utilities/pydantic.py,sha256=GElM_h4Mk6nBZsN7l60c-7JeipCnG99akl-e2hbHhVE,12444
 prefect/utilities/render_swagger.py,sha256=h2UrORVN3f7gM4zurtMnySjQXZIOWbji3uMinpbkl8U,3717
 prefect/utilities/services.py,sha256=pan_cq-REq2TfTCevdj0OC58qqQFOcetkPN_VV7nDRw,7622
 prefect/utilities/slugify.py,sha256=57Vb14t13F3zm1P65KAu8nVeAz0iJCd1Qc5eMG-R5y8,169
 prefect/utilities/templating.py,sha256=nAiOGMMHGbIDFkGYy-g-dzcbY311WfycdgAhsM3cLpY,13298
-prefect/utilities/text.py,sha256=eXGIsCcZ7h_6hy8T5GDQjL8GiKyktoOqavYub0QjgO4,445
+prefect/utilities/text.py,sha256=JRs6WNUn07i5oRP_jyTUYD6I0hdVwyMI0JmJc8f9gFw,731
 prefect/utilities/timeout.py,sha256=BRDIOWnqcw3B7X9tIk83Y3n9nQrJzZgduDQ63z-ns8w,1286
 prefect/utilities/urls.py,sha256=tKB7Bmbhbe3lBpiAIp7Idb_TWDmmDZI6nNTnDkwDKKI,8207
 prefect/utilities/visualization.py,sha256=Lum4IvLQ0nHsdLt6GGzS3Wwo-828u1rmOKc5mmWu994,6502
@@ -198,8 +197,8 @@ 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.0.4.dist-info/LICENSE,sha256=MCxsn8osAkzfxKC4CC_dLcUkU8DZLkyihZ8mGs3Ah3Q,11357
-prefect_client-3.0.4.dist-info/METADATA,sha256=ylM5VSYd7hcLhlSEVyvOk4Pmp5LuAfj7Fae0wdGUB0o,7332
-prefect_client-3.0.4.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
-prefect_client-3.0.4.dist-info/top_level.txt,sha256=MJZYJgFdbRc2woQCeB4vM6T33tr01TmkEhRcns6H_H4,8
-prefect_client-3.0.4.dist-info/RECORD,,
+prefect_client-3.0.5.dist-info/LICENSE,sha256=MCxsn8osAkzfxKC4CC_dLcUkU8DZLkyihZ8mGs3Ah3Q,11357
+prefect_client-3.0.5.dist-info/METADATA,sha256=z2Y18sxgwZTvdzyz5EzjKqyunbZ1riu3hS17perRosE,7332
+prefect_client-3.0.5.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
+prefect_client-3.0.5.dist-info/top_level.txt,sha256=MJZYJgFdbRc2woQCeB4vM6T33tr01TmkEhRcns6H_H4,8
+prefect_client-3.0.5.dist-info/RECORD,,

prefect/_internal/compatibility/experimental.py

@@ -1,195 +0,0 @@
-"""
-Utilities for experimental features.
-
-Experimental features have a group, feature name, and optional help string.
-
-When an experimental feature is used, a warning will be displayed. Warnings may be
-disabled per feature group with the setting `PREFECT_EXPERIMENTAL_WARN_<GROUP>`.
-Warnings may also be disabled globally with the setting `PREFECT_EXPERIMENTAL_WARN`.
-
-Some experimental features require opt-in to enable any usage. These require the setting
-`PREFECT_EXPERIMENTAL_ENABLE_<GROUP>` to be set or an error will be thrown on use.
-"""
-
-import functools
-import warnings
-from typing import Any, Callable, Optional, Set, TypeVar
-
-import pydantic
-
-from prefect.settings import PREFECT_EXPERIMENTAL_WARN, SETTING_VARIABLES, Setting
-from prefect.utilities.callables import get_call_parameters
-
-T = TypeVar("T", bound=Callable[..., Any])
-M = TypeVar("M", bound=pydantic.BaseModel)
-
-
-EXPERIMENTAL_WARNING = (
-    "{feature} is experimental. {help}The interface or behavior may change without"
-    " warning, we recommend pinning versions to prevent unexpected changes. To disable"
-    " warnings for this group of experiments, disable"
-    " PREFECT_EXPERIMENTAL_WARN_{group}."
-)
-
-EXPERIMENTAL_ERROR = (
-    "{feature} is experimental and requires opt-in for usage. {help}"
-    "To use this feature, enable PREFECT_EXPERIMENTAL_ENABLE_{group}."
-)
-
-
-class ExperimentalWarning(Warning):
-    """
-    A warning related to experimental code.
-    """
-
-
-class ExperimentalFeature(ExperimentalWarning):
-    """
-    A warning displayed on use of an experimental feature.
-
-    These can be globally disabled by the PREFECT_EXPIRIMENTAL_WARN setting.
-    """
-
-
-class ExperimentalError(Exception):
-    """
-    An exception related to experimental code.
-    """
-
-
-class ExperimentalFeatureDisabled(ExperimentalError):
-    """
-    An error displayed on use of a disabled experimental feature that requires opt-in.
-    """
-
-
-def _opt_in_setting_for_group(group: str) -> Setting[bool]:
-    group_opt_in_setting_name = f"PREFECT_EXPERIMENTAL_ENABLE_{group.upper()}"
-    group_opt_in = SETTING_VARIABLES.get(group_opt_in_setting_name)
-    if group_opt_in is None:
-        raise ValueError(
-            f"A opt-in setting for experimental feature {group!r} does not exist yet. "
-            f"{group_opt_in_setting_name!r} must be created before the group can be "
-            "used."
-        )
-    return group_opt_in
-
-
-def _warn_setting_for_group(group: str) -> Setting[bool]:
-    group_warn_setting_name = f"PREFECT_EXPERIMENTAL_WARN_{group.upper()}"
-    group_warn = SETTING_VARIABLES.get(group_warn_setting_name)
-    if group_warn is None:
-        raise ValueError(
-            f"A warn setting for experimental feature {group!r} does not exist yet. "
-            f"{group_warn_setting_name!r} must be created before the group can be used."
-        )
-    return group_warn
-
-
-def experimental(
-    feature: str,
-    *,
-    group: str,
-    help: str = "",
-    stacklevel: int = 2,
-    opt_in: bool = False,
-) -> Callable[[T], T]:
-    group = group.upper()
-
-    if help:
-        # Ensure help ends in a trailing space
-        help = help.rstrip() + " "
-
-    warn_message = EXPERIMENTAL_WARNING.format(feature=feature, group=group, help=help)
-    error_message = EXPERIMENTAL_ERROR.format(feature=feature, group=group, help=help)
-
-    if opt_in:
-        group_opt_in = _opt_in_setting_for_group(group)
-
-    group_warn = _warn_setting_for_group(group)
-
-    def decorator(fn: T) -> T:
-        @functools.wraps(fn)
-        def wrapper(*args, **kwargs):
-            if opt_in and not group_opt_in:
-                raise ExperimentalFeatureDisabled(error_message)
-
-            if PREFECT_EXPERIMENTAL_WARN and group_warn:
-                warnings.warn(
-                    warn_message,
-                    ExperimentalFeature,
-                    stacklevel=stacklevel,
-                )
-            return fn(*args, **kwargs)
-
-        return wrapper
-
-    return decorator
-
-
-def experiment_enabled(group: str) -> bool:
-    group_opt_in = _opt_in_setting_for_group(group)
-    return group_opt_in.value()
-
-
-def experimental_parameter(
-    name: str,
-    *,
-    group: str,
-    help: str = "",
-    stacklevel: int = 2,
-    opt_in: bool = False,
-    when: Optional[Callable[[Any], bool]] = None,
-) -> Callable[[T], T]:
-    """
-    Mark a parameter in a callable as experimental.
-
-    Example:
-
-        ```python
-
-        @experimental_parameter("y", group="example", when=lambda y: y is not None)
-        def foo(x, y = None):
-            return x + 1 + (y or 0)
-        ```
-    """
-
-    when = when or (lambda _: True)
-
-    @experimental(
-        group=group,
-        feature=f"The parameter {name!r}",
-        help=help,
-        opt_in=opt_in,
-        stacklevel=stacklevel + 2,
-    )
-    def experimental_check():
-        pass
-
-    def decorator(fn: T):
-        @functools.wraps(fn)
-        def wrapper(*args, **kwargs):
-            try:
-                parameters = get_call_parameters(fn, args, kwargs, apply_defaults=False)
-            except Exception:
-                # Avoid raising any parsing exceptions here
-                parameters = kwargs
-
-            if name in parameters and when(parameters[name]):
-                experimental_check()
-            return fn(*args, **kwargs)
-
-        return wrapper
-
-    return decorator
-
-
-def enabled_experiments() -> Set[str]:
-    """
-    Return the set of all enabled experiments.
-    """
-    return {
-        name[len("PREFECT_EXPERIMENTAL_ENABLE_") :].lower()
-        for name, setting in SETTING_VARIABLES.items()
-        if name.startswith("PREFECT_EXPERIMENTAL_ENABLE_") and setting.value()
-    }