Have a look at Powertools for AWS Lambda (Python)
https://docs.powertools.aws.dev/lambda/python/3.1.0/core/logger/