How can I tell a AWS Lambda Function to redirect based off the environment?

Question

So currently I\'m using AWS Lambdas as triggers for my Cognito Passwordless authentication. For the create_auth_challenge trigger I have an AWS Lambda function that sends a link