Exemplo de resposta personalizada para AWS SAM

Você pode personalizar algumas respostas de erro do API Gateway definindo cabeçalhos de resposta em seu AWS SAM modelo. Para fazer isso, use o tipo de dados Gateway Response Object.

Veja a seguir um exemplo AWS SAM de modelo que cria uma resposta personalizada para o DEFAULT_5XX erro.

AWSTemplateFormatVersion: '2010-09-09'
Transform: AWS::Serverless-2016-10-31
Resources:
  MyApi:
    Type: AWS::Serverless::Api
    Properties:
      StageName: Prod
      GatewayResponses:
        DEFAULT_5XX:
          ResponseParameters:
            Headers:
              Access-Control-Expose-Headers: "'WWW-Authenticate'"
              Access-Control-Allow-Origin: "'*'"
              ErrorHeader: "'MyCustomErrorHeader'"
          ResponseTemplates:
            application/json: "{\"message\": \"Error on the $context.resourcePath resource\" }"
              
  GetFunction:
    Type: AWS::Serverless::Function
    Properties:
      Runtime: python3.10
      Handler: index.handler
      InlineCode: |
        def handler(event, context):
          raise Exception('Check out the new response!')
      Events:
        GetResource:
          Type: Api
          Properties:
            Path: /error
            Method: get
            RestApiId: !Ref MyApi

Para obter mais informações sobre as respostas do API Gateway, consulte Respostas do API Gateway no Gateway no Guia do desenvolvedor do API Gateway.