index.js

'use strict'

const aws = require('aws-sdk')
const parser = require('lambda-multipart-parser')

const accessTokenVerifier = require('./accessTokenVerifier')

const response = (statusCode, data) => ({
  statusCode,
  headers: {
    'Content-Type': 'application/json',
    'Access-Control-Allow-Origin': '*'
  },
  body: JSON.stringify(data)
})

exports.handler = async event => {
  const decodedToken = accessTokenVerifier(event.headers['Authorization'])
  if (!decodedToken) {
    return response(401, {
      error: 'Please include access token in authorization header.'
    })
  }

  const sagemakerRuntime = new aws.SageMakerRuntime()

  const { ENDPOINT_NAME_PREFIX } = process.env

  try {
    const data = await parser.parse(event)
    const result = await sagemakerRuntime
      .invokeEndpoint({
        Body: data.files[0].content,
        EndpointName: ENDPOINT_NAME_PREFIX + decodedToken.username,
        ContentType: data.files[0].contentType
      })
      .promise()
    console.log('Result: ', result)

    return response(200, result)
  } catch (error) {
    console.log('Error: ', error)
    return response(500, {
      error: error.message
    })
  }
}