diff --git a/aws_lambda_powertools/event_handler/__init__.py b/aws_lambda_powertools/event_handler/__init__.py index 0b0c08b20a1..d63cd584bd4 100644 --- a/aws_lambda_powertools/event_handler/__init__.py +++ b/aws_lambda_powertools/event_handler/__init__.py @@ -2,7 +2,14 @@ Event handler decorators for common Lambda events """ -from .api_gateway import ALBResolver, APIGatewayHttpResolver, ApiGatewayResolver, APIGatewayRestResolver +from .api_gateway import ALBResolver, APIGatewayHttpResolver, ApiGatewayResolver, APIGatewayRestResolver, CORSConfig from .appsync import AppSyncResolver -__all__ = ["AppSyncResolver", "APIGatewayRestResolver", "APIGatewayHttpResolver", "ALBResolver", "ApiGatewayResolver"] +__all__ = [ + "AppSyncResolver", + "APIGatewayRestResolver", + "APIGatewayHttpResolver", + "ALBResolver", + "ApiGatewayResolver", + "CORSConfig", +] diff --git a/docs/core/event_handler/api_gateway.md b/docs/core/event_handler/api_gateway.md index cc01b66600d..4f86dc8fdf3 100644 --- a/docs/core/event_handler/api_gateway.md +++ b/docs/core/event_handler/api_gateway.md @@ -537,7 +537,7 @@ You can use **`exception_handler`** decorator with any Python exception. This al from aws_lambda_powertools import Logger, Tracer from aws_lambda_powertools.logging import correlation_paths from aws_lambda_powertools.event_handler import content_types -from aws_lambda_powertools.event_handler import APIGatewayRestResolver, Response +from aws_lambda_powertools.event_handler.api_gateway import APIGatewayRestResolver, Response tracer = Tracer() logger = Logger() @@ -685,7 +685,7 @@ This will ensure that CORS headers are always returned as part of the response w ```python hl_lines="9 11" from aws_lambda_powertools import Logger, Tracer from aws_lambda_powertools.logging import correlation_paths - from aws_lambda_powertools.event_handler import APIGatewayRestResolver, CORSConfig + from aws_lambda_powertools.event_handler.api_gateway import APIGatewayRestResolver, CORSConfig tracer = Tracer() logger = Logger()