diff --git a/lambda_requests/lambda_request.py b/lambda_requests/lambda_request.py index 59a1ace..e5c334e 100644 --- a/lambda_requests/lambda_request.py +++ b/lambda_requests/lambda_request.py @@ -32,6 +32,7 @@ def __exit__(self, *args, **kwargs): class LambdaAdapter(BaseAdapter): def __init__(self, region=None): self.region = region or os.environ.get("AWS_DEFAULT_REGION", "us-east-1") + self.endpoint_url = region or os.environ.get("AWS_LAMBDA_ENDPOINT_URL", None) self._lambda_client = None super(LambdaAdapter, self).__init__() @@ -44,7 +45,9 @@ def __exit__(self, *_, **__): @property def lambda_client(self): if not self._lambda_client: - self._lambda_client = boto3.client("lambda", region_name=self.region) + self._lambda_client = boto3.client("lambda", + region_name=self.region, + endpoint_url=self.endpoint_url) return self._lambda_client def send(self, request, **kwargs):