Static typing: Argument missing for parameter "context" when using `@tracer.capture_lambda_handler(capture_response=False)`

[tiagohconte]

Static type checker used

pyright/pylance

AWS Lambda function runtime

3.11

Powertools for AWS Lambda (Python) version

3.6.0

Static type checker info

Argument missing for parameter "context"

The issue only happens when there is any parameter set in capture_lambda_handler.

Code snippet

@logger.inject_lambda_context(log_event=True)
@tracer.capture_lambda_handler(capture_response=False)
def handle(event, context: LambdaContext) -> dict:
    return app.resolve(event, context)

Possible Solution

May be related to #4088 and #4795

[boring-cyborg]

Thanks for opening your first issue here! We'll come back to you as soon as we can.
In the meantime, check out the #python channel on our Powertools for AWS Lambda Discord: Invite link

[leandrodamascena]

Oi @tiagohconte, tudo bem?!

We are missing the return type annotation in the capture_lambda_handler decorator, that's why pyright is complaining and mypy will complain too. Although we don't fully support strict typing in Powertools, I see we might fix this. Do you want to submit a PR to fix it? I think annotating the return type with a Callable with Event and Context should solve the problem.

Obrigado!

[tiagohconte]

Oi @leandrodamascena, tudo bem e contigo?

Thanks for the quick response! Just opened a PR to solve the issue 😃