from process.process import process
from process.utils.models import AWSEvent, Event


def handler(event, context):
    event = Event.model_validate(event).root

    if isinstance(event, AWSEvent):
        record = event.records[0]
        process(bucket=record.s3.bucket.name, key=record.s3.object.key)
    else:
        process(**event.model_dump())