template.yaml

AWSTemplateFormatVersion: '2010-09-09'
Transform: AWS::Serverless-2016-10-31

Parameters:
  StageName:
    Type: String
    Default: Prod

Resources:
  MyHttpApi:
    Type: AWS::Serverless::HttpApi
    Properties:
      StageName: !Ref StageName
      CorsConfiguration:
        AllowOrigins:
          - '*'
        AllowMethods:
          - POST
          - OPTIONS
        AllowHeaders:
          - '*'
        AllowCredentials: "'true'"
        ExposeHeaders: "'*'"
      # Add a definition body to handle CORS preflight requests
      DefinitionBody:
        openapi: 3.0.1
        servers:
          variables:
            stage:
              default: !Ref StageName
        info:
          title: !Sub "${AWS::StackName}"
          version: '1.0'
        paths:
          /answer:
            post:
              responses:
                default:
                  description: "Default response for POST /answer"
              x-amazon-apigateway-integration:
                payloadFormatVersion: "2.0"
                type: "aws_proxy"
                httpMethod: "POST"
                uri: !Sub "arn:aws:apigateway:${AWS::Region}:lambda:path/2015-03-31/functions/${InferenceFunction.Arn}/invocations"
                connectionType: "INTERNET"
        x-amazon-apigateway-cors:
          allowMethods:
          - "*"
          allowHeaders:
          - "*"
          exposeHeaders:
          - "*"
          maxAge: 0
          allowCredentials: false
          allowOrigins:
          - "*"
  InferenceFunction:
    Type: AWS::Serverless::Function # More info about Function Resource: https://github.com/awslabs/serverless-application-model/blob/master/versions/2016-10-31.md#awsserverlessfunction
    Properties:
      PackageType: Image
      Timeout: 30
      MemorySize: 256
      Architectures:
        - x86_64
      Policies:
        - AmazonSSMReadOnlyAccess
      Events:
        Inference:
          Type: HttpApi
          Properties:
            Path: /answer
            Method: post
            ApiId: !Ref MyHttpApi
    Metadata:
      Dockerfile: Dockerfile
      DockerContext: ./app
      DockerTag: python3.9-v1
Outputs:
  InferenceApi:
    Description: "API Gateway endpoint URL for Prod stage for Inference function"
    Value: !Sub "https://${MyHttpApi}.execute-api.${AWS::Region}.amazonaws.com/${StageName}/answer"