File size: 2,290 Bytes
00b4479 3dbe011 d071ba9 43d87b3 c6054f0 85dd36a f2a79fa 7d6e00c 238719b fa76f5f 43d87b3 d071ba9 00b4479 d071ba9 00b4479 d071ba9 00b4479 fd9de0f 7d6e00c fd9de0f 6f1250c 7d6e00c 6f1250c c6054f0 ee2ed59 85dd36a ee2ed59 85dd36a d071ba9 43d87b3 5b88544 9c09a5a 7d6e00c 43d87b3 d071ba9 85dd36a 43d87b3 d071ba9 f2a79fa 9c09a5a f2a79fa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
"""Lambda entry point"""
from http import HTTPStatus
from typing import Dict
from aws_lambda_powertools.utilities.typing import LambdaContext
from pydantic import ValidationError
from src import app_logger
from src.io.lambda_helpers import get_parsed_request_body, get_parsed_bbox_points, get_response
from src.prediction_api.predictors import samexporter_predict
def lambda_handler(event: Dict, context: LambdaContext) -> str:
"""
Handle the request for the serverless backend and return the response
(success or a type of error based on the exception raised).
Args:
event: request content
context: request context
Returns:
json response from get_response() function
"""
from time import time
app_logger.info(f"start with aws_request_id:{context.aws_request_id}.")
start_time = time()
if "version" in event:
app_logger.info(f"event version: {event['version']}.")
try:
app_logger.info("try get_parsed_event...")
request_input = get_parsed_request_body(event)
app_logger.info("event parsed: ok")
body_request = get_parsed_bbox_points(request_input)
app_logger.info(f"body_request => {type(body_request)}, {body_request}.")
try:
body_response = samexporter_predict(body_request["bbox"], body_request["prompt"], body_request["zoom"])
app_logger.info(f"output body_response length:{len(body_response)}.")
app_logger.debug(f"output body_response:{body_response}.")
response = get_response(HTTPStatus.OK.value, start_time, context.aws_request_id, body_response)
except Exception as ex2:
app_logger.error(f"exception2:{ex2}.")
response = get_response(HTTPStatus.INTERNAL_SERVER_ERROR.value, start_time, context.aws_request_id, {})
except ValidationError as va1:
app_logger.error(f"ValidationError:{va1}.")
response = get_response(HTTPStatus.UNPROCESSABLE_ENTITY.value, start_time, context.aws_request_id, {})
except Exception as ex1:
app_logger.error(f"exception1:{ex1}.")
response = get_response(HTTPStatus.BAD_REQUEST.value, start_time, context.aws_request_id, {})
app_logger.debug(f"response_dumped:{response}...")
return response
|