diff --git a/aws_lambda_powertools/utilities/data_classes/api_gateway_authorizer_event.py b/aws_lambda_powertools/utilities/data_classes/api_gateway_authorizer_event.py index 32d933ecf27..0b1aec43e8a 100644 --- a/aws_lambda_powertools/utilities/data_classes/api_gateway_authorizer_event.py +++ b/aws_lambda_powertools/utilities/data_classes/api_gateway_authorizer_event.py @@ -1,6 +1,6 @@ import enum import re -from typing import Any, Dict, List, Optional +from typing import Any, Dict, List, Optional, overload from aws_lambda_powertools.utilities.data_classes.common import ( BaseRequestContext, @@ -162,6 +162,22 @@ def stage_variables(self) -> Dict[str, str]: def request_context(self) -> BaseRequestContext: return BaseRequestContext(self._data) + @overload + def get_header_value( + self, + name: str, + default_value: str, + case_sensitive: Optional[bool] = False, + ) -> str: ... + + @overload + def get_header_value( + self, + name: str, + default_value: Optional[str] = None, + case_sensitive: Optional[bool] = False, + ) -> Optional[str]: ... + def get_header_value( self, name: str,