chore: remove local runner lambda client (#201)

Author: bchampp

src/aws_durable_execution_sdk_python/botocore/data/lambdainternal-local/2015-03-31/service-2.json

@@ -97,7 +97,6 @@ class DurableExecutionInvocationInput:
     durable_execution_arn: str
     checkpoint_token: str
     initial_execution_state: InitialExecutionState
-    is_local_runner: bool
 
     @staticmethod
     def from_dict(
@@ -109,15 +108,13 @@ def from_dict(
             initial_execution_state=InitialExecutionState.from_dict(
                 input_dict.get("InitialExecutionState", {})
             ),
-            is_local_runner=input_dict.get("LocalRunner", False),
         )
 
     def to_dict(self) -> MutableMapping[str, Any]:
         return {
             "DurableExecutionArn": self.durable_execution_arn,
             "CheckpointToken": self.checkpoint_token,
             "InitialExecutionState": self.initial_execution_state.to_dict(),
-            "LocalRunner": self.is_local_runner,
         }
 
 
@@ -139,7 +136,6 @@ def from_durable_execution_invocation_input(
             durable_execution_arn=invocation_input.durable_execution_arn,
             checkpoint_token=invocation_input.checkpoint_token,
             initial_execution_state=invocation_input.initial_execution_state,
-            is_local_runner=invocation_input.is_local_runner,
             service_client=service_client,
         )
 
@@ -237,15 +233,12 @@ def wrapper(event: Any, context: LambdaContext) -> MutableMapping[str, Any]:
                 )
                 raise ExecutionError(msg) from e
 
-            # Local runner always uses its own client, otherwise use custom or default
-            if invocation_input.is_local_runner:
-                service_client = LambdaClient.initialize_local_runner_client()
-            else:
-                service_client = (
-                    LambdaClient(client=boto3_client)
-                    if boto3_client is not None
-                    else LambdaClient.initialize_from_env()
-                )
+            # Use custom client if provided, otherwise initialize from environment
+            service_client = (
+                LambdaClient(client=boto3_client)
+                if boto3_client is not None
+                else LambdaClient.initialize_from_env()
+            )
 
         raw_input_payload: str | None = (
             invocation_input.initial_execution_state.get_input_payload()

src/aws_durable_execution_sdk_python/execution.py

@@ -951,33 +951,6 @@ def load_preview_botocore_models() -> None:
             Path(__file__).parent.joinpath("botocore", "data")
         )
 
-    @staticmethod
-    def initialize_local_runner_client() -> LambdaClient:
-        endpoint = os.getenv(
-            "DURABLE_LOCAL_RUNNER_ENDPOINT", "http://host.docker.internal:5000"
-        )
-        region = os.getenv("DURABLE_LOCAL_RUNNER_REGION", "us-west-2")
-
-        # The local runner client needs execute-api as the signing service name,
-        # so we have a second `lambdainternal-local` boto model with this.
-        LambdaClient.load_preview_botocore_models()
-        client = boto3.client(
-            "lambdainternal-local",
-            endpoint_url=endpoint,
-            region_name=region,
-            config=Config(
-                connect_timeout=5,
-                read_timeout=50,
-            ),
-        )
-
-        logger.debug(
-            "Initialized lambda client with endpoint: '%s', region: '%s'",
-            endpoint,
-            region,
-        )
-        return LambdaClient(client=client)
-
     @staticmethod
     def initialize_from_env() -> LambdaClient:
         LambdaClient.load_preview_botocore_models()

src/aws_durable_execution_sdk_python/lambda_service.py

@@ -68,7 +68,7 @@ def my_handler(event, context: DurableContext) -> list[str]:
         "aws_durable_execution_sdk_python.execution.LambdaClient"
     ) as mock_client_class:
         mock_client = Mock()
-        mock_client_class.initialize_local_runner_client.return_value = mock_client
+        mock_client_class.initialize_from_env.return_value = mock_client
 
         # Mock the checkpoint method to track calls
         checkpoint_calls = []
@@ -154,7 +154,7 @@ def my_handler(event, context: DurableContext):
         "aws_durable_execution_sdk_python.execution.LambdaClient"
     ) as mock_client_class:
         mock_client = Mock()
-        mock_client_class.initialize_local_runner_client.return_value = mock_client
+        mock_client_class.initialize_from_env.return_value = mock_client
 
         # Mock the checkpoint method to track calls
         checkpoint_calls = []
@@ -255,7 +255,7 @@ def my_handler(event, context):
         "aws_durable_execution_sdk_python.execution.LambdaClient"
     ) as mock_client_class:
         mock_client = Mock()
-        mock_client_class.initialize_local_runner_client.return_value = mock_client
+        mock_client_class.initialize_from_env.return_value = mock_client
 
         # Mock the checkpoint method to track calls
         checkpoint_calls = []
@@ -361,7 +361,7 @@ def my_handler(event, context: DurableContext):
         "aws_durable_execution_sdk_python.execution.LambdaClient"
     ) as mock_client_class:
         mock_client = Mock()
-        mock_client_class.initialize_local_runner_client.return_value = mock_client
+        mock_client_class.initialize_from_env.return_value = mock_client
 
         # Mock the checkpoint method to raise an error (using RuntimeError as a generic exception)
         def mock_checkpoint_failure(
@@ -424,7 +424,7 @@ def my_handler(event: Any, context: DurableContext):
         "aws_durable_execution_sdk_python.execution.LambdaClient"
     ) as mock_client_class:
         mock_client = Mock()
-        mock_client_class.initialize_local_runner_client.return_value = mock_client
+        mock_client_class.initialize_from_env.return_value = mock_client
 
         # Mock the checkpoint method to track calls
         checkpoint_calls = []