diff --git a/aws_lambda_powertools/middleware_factory/factory.py b/aws_lambda_powertools/middleware_factory/factory.py index 8ab16c5e8b7..056c7604f5b 100644 --- a/aws_lambda_powertools/middleware_factory/factory.py +++ b/aws_lambda_powertools/middleware_factory/factory.py @@ -122,9 +122,9 @@ def final_decorator(func: Optional[Callable] = None, **kwargs): ) @functools.wraps(func) - def wrapper(event, context): + def wrapper(**kwargs_wrapper): try: - middleware = functools.partial(decorator, func, event, context, **kwargs) + middleware = functools.partial(decorator, func, **(kwargs_wrapper | kwargs)) if trace_execution: tracer = Tracer(auto_patch=False) with tracer.provider.in_subsegment(name=f"## {decorator.__qualname__}"):