Create client_injector for better type hints (#12793)

Author: bunchesofdonald

src/prefect/client/utilities.py

@@ -6,14 +6,25 @@
 # circular imports for decorators such as `inject_client` which are widely used.
 
 from functools import wraps
-from typing import TYPE_CHECKING, Any, Callable, Coroutine, Optional, Tuple, cast
-
-from typing_extensions import ParamSpec
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Awaitable,
+    Callable,
+    Coroutine,
+    Optional,
+    Tuple,
+    TypeVar,
+    cast,
+)
+
+from typing_extensions import Concatenate, ParamSpec
 
 if TYPE_CHECKING:
     from prefect.client.orchestration import PrefectClient
 
 P = ParamSpec("P")
+R = TypeVar("R")
 
 
 def get_or_create_client(
@@ -52,6 +63,17 @@ def get_or_create_client(
         return get_httpx_client(), False
 
 
+def client_injector(
+    func: Callable[Concatenate["PrefectClient", P], Awaitable[R]],
+) -> Callable[P, Awaitable[R]]:
+    @wraps(func)
+    async def wrapper(*args: P.args, **kwargs: P.kwargs) -> R:
+        client, _ = get_or_create_client()
+        return await func(client, *args, **kwargs)
+
+    return wrapper
+
+
 def inject_client(
     fn: Callable[P, Coroutine[Any, Any, Any]],
 ) -> Callable[P, Coroutine[Any, Any, Any]]: