Add wrap_async helper

src/haiway/__init__.py

@@ -15,6 +15,7 @@
     throttle,
     timeout,
     traced,
+    wrap_async,
 )
 from haiway.state import State
 from haiway.types import (
@@ -76,4 +77,5 @@
     "timeout",
     "traced",
     "when_missing",
+    "wrap_async",
 ]

src/haiway/helpers/__init__.py

@@ -1,4 +1,4 @@
-from haiway.helpers.asynchrony import asynchronous
+from haiway.helpers.asynchrony import asynchronous, wrap_async
 from haiway.helpers.caching import cache
 from haiway.helpers.retries import retry
 from haiway.helpers.throttling import throttle
@@ -14,4 +14,5 @@
     "throttle",
     "timeout",
     "traced",
+    "wrap_async",
 ]

src/haiway/helpers/asynchrony.py

@@ -9,9 +9,26 @@
 
 __all__ = [
     "asynchronous",
+    "wrap_async",
 ]
 
 
+def wrap_async[**Args, Result](
+    function: Callable[Args, Coroutine[None, None, Result]] | Callable[Args, Result],
+    /,
+) -> Callable[Args, Coroutine[None, None, Result]]:
+    if iscoroutinefunction(function):
+        return function
+
+    else:
+
+        async def async_function(*args: Args.args, **kwargs: Args.kwargs) -> Result:
+            return cast(Callable[Args, Result], function)(*args, **kwargs)
+
+        _mimic_async(function, within=async_function)
+        return async_function
+
+
 @overload
 def asynchronous[**Args, Result]() -> (
     Callable[