fix: Await a coroutine

julep-ai · Nov 29, 2024 · 9ce2d6b · 9ce2d6b
1 parent d20f2e5
commit 9ce2d6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/agents-api/agents_api/metrics/counters.py b/agents-api/agents_api/metrics/counters.py
@@ -24,7 +24,7 @@ def wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
         @wraps(func)
         async def async_wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
             metric.labels(kwargs.get(id_field_name, "not_set")).inc()
-            return func(*args, **kwargs)
+            return await func(*args, **kwargs)
 
         return async_wrapper if inspect.iscoroutinefunction(func) else wrapper