fix(py-sdk): handle case when function is async

ApeWorX · Oct 3, 2024 · d20a794 · d20a794
1 parent 4c9ad42
commit d20a794
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/sdk/py/apepay/manager.py b/sdk/py/apepay/manager.py
@@ -1,3 +1,4 @@
+import inspect
 from collections.abc import Iterator
 from datetime import timedelta
 from difflib import Differ
@@ -214,8 +215,13 @@ def decorator(f):
 
             @app.on_(container)
             @wraps(f)
-            def inner(log, **dependencies):
-                return f(Stream(manager=self, id=log.stream_id), **dependencies)
+            async def inner(log, **dependencies):
+                result = f(Stream(manager=self, id=log.stream_id), **dependencies)
+
+                if inspect.isawaitable(result):
+                    return await result
+
+                return result
 
             return inner