AWS Documentation » AWS Step Functions » Developer Guide » How Step Functions Works » AWS Service Integrations » Supported AWS Service Integrations for Step Functions » Invoke Lambda with Step Functions

Invoke Lambda with Step Functions

Step Functions can control certain AWS services directly from the Amazon States Language. For more information, see:

• Service Integrations

• Pass Parameters to a Service API

For more information about managing state input, output and results, see Input and Output Processing in Step Functions.

Supported APIs:

• Invoke

  • Request Syntax

  • Supported Parameters

    • ClientContext

    • FunctionName

    • InvocationType

    • Qualifier

    • Payload

  • Response syntax

The following includes a Task state that invokes a Lambda function.

{  
   "StartAt":"CallLambda",
   "States":{  
      "CallLambda":{  
         "Type":"Task",
         "Resource":"arn:aws:states:::lambda:invoke",
         "Parameters":{  
            "FunctionName":"MyFunction"
         },
         "End":true
      }
   }
}

The following includes a Task state that implements the callback service integration pattern.

{  
   "StartAt":"GetManualReview",
   "States":{  
      "GetManualReview":{  
         "Type":"Task",
         "Resource":"arn:aws:states:::lambda:invoke.waitForTaskToken",
         "Parameters":{  
            "FunctionName":"get-model-review-decision",
            "Payload":{  
               "model.$":"$.new_model",
               "token.$":"$$.Task.Token"
            },
            "Qualifier":"prod-v1"
         },
         "End":true
      }
   }
}

Note

The Lambda invoke API includes logs in the response by default. Multiple Lambda invocations in a workflow can trigger States.DataLimitExceeded errors. To avoid this, include "LogType" = "None" as a parameter when you invoke your Lambda functions.

To simply invoke a Lambda function, you can also call the resource ARN directly in the "Resource" string.

{  
   "StartAt":"CallFunction",
   "States":{  
      "CallFunction": {  
         "Type":"Task",
         "Resource":"arn:aws:lambda:us-east-1:123456789012:function:HelloFunction",
         "End": true
      }
   }
}     

When you invoke Lambda using an ARN directly from the "Resource" field, you cannot specify .waitForTaskToken, and the state input is passed to Lambda as the payload. The output of the Lambda function is the result. To have a Lambda task wait for a task token, see the previous callback pattern example.

You can invoke a specific Lambda function version or alias by specifying those options in the Amazon Resource Name in the Resource field. See the following in the Lambda documentation.

• AWS Lambda versioning

• AWS Lambda aliases

For information on how to configure IAM when using Step Functions with other AWS services, see IAM Policies for Integrated Services.