可以使用Task Token来从上一步获取原始输入。以下是一个代码示例：

在状态机定义文件中，添加以下代码：

{
  "Comment": "Example state machine",
  "StartAt": "Step1",
  "States": {
    "Step1": {
      "Type": "Task",
      "Resource": "arn:aws:lambda:us-east-1:123456789012:function:example",
      "InputPath": "$.input",
      "ResultPath": "$.result",
      "Next": "Step2"
    },
    "Step2": {
      "Type": "Task",
      "Resource": "arn:aws:lambda:us-east-1:123456789012:function:example",
      "InputPath": "$.result",
      "ResultPath": "$.result",
      "End": true
    }
  }
}

在Lambda函数中，在调用PutOutput的时候，指定Task Token和OriginalInput参数。以下是一个示例代码：

import json
import boto3

def lambda_handler(event, context):
    input = event['input']
    task_token = event['token']

    # process input here

    original_input = event['originalInput']

    # do something with original_input

    client = boto3.client('stepfunctions')

    response = client.send_task_success(
        taskToken=task_token,
        output=json.dumps({"result": "success"}),
        originalInput=original_input
    )

    return {
        'statusCode': 200,
        'body': json.dumps('Success!')
    }

在Lambda函数中，注意需要从event参数中获取task_token和originalInput。使用send_task_success向状态机返回处理结果和原始输入。