File tree Expand file tree Collapse file tree
src/aws_durable_execution_sdk_python Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -862,11 +862,19 @@ def initialize_local_runner_client() -> LambdaClient:
862862 @staticmethod
863863 def initialize_from_env () -> LambdaClient :
864864 LambdaClient .load_preview_botocore_models ()
865- client = boto3 .client (
866- "lambdainternal" ,
867- )
868865
869- logger .debug ("Initialized lambda client" )
866+ """
867+ TODO - we can remove this when were using the actual lambda client,
868+ but we need this with the preview model because boto won't match against lambdainternal.
869+ """
870+ endpoint_url = os .getenv ("AWS_ENDPOINT_URL_LAMBDA" , None )
871+ if not endpoint_url :
872+ client = boto3 .client (
873+ "lambdainternal" ,
874+ )
875+ else :
876+ client = boto3 .client ("lambdainternal" , endpoint_url = endpoint_url )
877+
870878 return LambdaClient (client = client )
871879
872880 def checkpoint (
You can’t perform that action at this time.
0 commit comments