diff --git a/newrelic/hooks/mlmodel_langgraph.py b/newrelic/hooks/mlmodel_langgraph.py index f7a3380f0..a2ab62a81 100644 --- a/newrelic/hooks/mlmodel_langgraph.py +++ b/newrelic/hooks/mlmodel_langgraph.py @@ -82,7 +82,7 @@ def wrap_AsyncBackgroundExecutor_submit(wrapped, instance, args, kwargs): return wrapped(*args, **kwargs) context = ContextOf(trace=trace, strict=True) - func = coroutine_wrapper(wrapped, context) + func = coroutine_wrapper(func, context) return wrapped(func, *args, **kwargs)