lambda_function

prefect_aws.lambda_function

Integrations with AWS Lambda.

Examples:

Run a lambda function with a payload

LambdaFunction(
    function_name="test-function",
    aws_credentials=aws_credentials,
).invoke(payload={"foo": "bar"})

Specify a version of a lambda function

LambdaFunction(
    function_name="test-function",
    qualifier="1",
    aws_credentials=aws_credentials,
).invoke()

Invoke a lambda function asynchronously

LambdaFunction(
    function_name="test-function",
    aws_credentials=aws_credentials,
).invoke(invocation_type="Event")

Invoke a lambda function and return the last 4 KB of logs

LambdaFunction(
    function_name="test-function",
    aws_credentials=aws_credentials,
).invoke(tail=True)

Invoke a lambda function with a client context

LambdaFunction(
    function_name="test-function",
    aws_credentials=aws_credentials,
).invoke(client_context={"bar": "foo"})

Classes

LambdaFunction

Invoke a Lambda function. This block is part of the prefect-aws collection. Install prefect-aws with pip install prefect-aws to use this block.

Attributes:

• function_name: The name, ARN, or partial ARN of the Lambda function to run. This must be the name of a function that is already deployed to AWS Lambda.
• qualifier: The version or alias of the Lambda function to use when invoked. If not specified, the latest (unqualified) version of the Lambda function will be used.
• aws_credentials: The AWS credentials to use to connect to AWS Lambda with a default factory of AwsCredentials.

Methods:

ainvoke

ainvoke(self, payload: Optional[dict] = None, invocation_type: Literal['RequestResponse', 'Event', 'DryRun'] = 'RequestResponse', tail: bool = False, client_context: Optional[dict] = None) -> dict

Asynchronously invoke the Lambda function with the given payload.

Args:

• payload: The payload to send to the Lambda function.
• invocation_type: The invocation type of the Lambda function. This can be one of “RequestResponse”, “Event”, or “DryRun”. Uses “RequestResponse” by default.
• tail: If True, the response will include the base64-encoded last 4 KB of log data produced by the Lambda function.
• client_context: The client context to send to the Lambda function. Limited to 3583 bytes.

Returns:

• The response from the Lambda function.

Examples:

from prefect import flow
from prefect_aws.lambda_function import LambdaFunction
from prefect_aws.credentials import AwsCredentials

@flow
async def example_flow():
    credentials = AwsCredentials()
    lambda_function = LambdaFunction(
        function_name="test_lambda_function",
        aws_credentials=credentials,
    )
    response = await lambda_function.ainvoke(
        payload={"foo": "bar"},
        invocation_type="RequestResponse",
    )
    return response["Payload"].read()

invoke

invoke(self, payload: Optional[dict] = None, invocation_type: Literal['RequestResponse', 'Event', 'DryRun'] = 'RequestResponse', tail: bool = False, client_context: Optional[dict] = None) -> dict

Invoke the Lambda function with the given payload.

Args:

• payload: The payload to send to the Lambda function.
• invocation_type: The invocation type of the Lambda function. This can be one of “RequestResponse”, “Event”, or “DryRun”. Uses “RequestResponse” by default.
• tail: If True, the response will include the base64-encoded last 4 KB of log data produced by the Lambda function.
• client_context: The client context to send to the Lambda function. Limited to 3583 bytes.

Returns:

• The response from the Lambda function.

Examples:

from prefect_aws.lambda_function import LambdaFunction
from prefect_aws.credentials import AwsCredentials

credentials = AwsCredentials()
lambda_function = LambdaFunction(
    function_name="test_lambda_function",
    aws_credentials=credentials,
)
response = lambda_function.invoke(
    payload={"foo": "bar"},
    invocation_type="RequestResponse",
)
response["Payload"].read()