This example shows the receiving code within a Python Lambda to retrieve the task token for a callback

brevity-stepfunctioncallbacklambda.py

import json, boto3, os, re

def lambda_handler(event, context):
    
    if event['program'] is None:
        return {"isBase64Encoded":False,"statusCode":400,"body":json.dumps({"error":"Missing program name."})}
    if event['operation'] is None:
        return {"isBase64Encoded":False,"statusCode":400,"body":json.dumps({"error":"Missing operation name."})}
    else:
        operationName = str(event['operation']) 
    
    programName = str(event['program'])
    taskToken = str(event['token'])
    
    stepFunctionsStatus = generateScriptStepFunctions(programName, taskToken, operationName)