Hello, I'm currently using the Pulumi Automation API to create my AWS infrastructure. Now, I want to switch to a specific IAM role when performing the setup. Is there a way to achieve this similar to the following code snippet?

stack, err := auto.UpsertStackInlineSource(context.Background(), stackName, projectName, pulumiMain, opts...)
if err != nil {
    fmt.Printf("Error in UpsertStackInlineSource: %v\n", err)
    panic(err)
}

stack.SetConfig(context.Background(), "aws:assumeRole", auto.ConfigValue{
    Value:  "arn:aws:iam::123456789:role/testDeployRole",
    Secret: false,
})