Updated default LambdaClient to include SDK name/version in UserAgent headers

Author: nvasiu

src/aws_durable_execution_sdk_python/__about__.py

@@ -2,3 +2,4 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 __version__ = "1.4.0"
+__sdk_name__ = "aws-durable-execution-sdk-python"

src/aws_durable_execution_sdk_python/lambda_service.py

@@ -11,12 +11,14 @@
 import boto3
 from botocore.config import Config
 
+from aws_durable_execution_sdk_python.__about__ import __sdk_name__, __version__
 from aws_durable_execution_sdk_python.exceptions import (
     CallableRuntimeError,
     CheckpointError,
     GetExecutionStateError,
 )
 
+
 if TYPE_CHECKING:
     from mypy_boto3_lambda import LambdaClient as Boto3LambdaClient
     from mypy_boto3_lambda.type_defs import (
@@ -1059,6 +1061,7 @@ def initialize_client(cls) -> LambdaClient:
                 config=Config(
                     connect_timeout=5,
                     read_timeout=50,
+                    user_agent_extra=f"{__sdk_name__}/{__version__}",
                 ),
             )
         return cls(client=cls._cached_boto_client)

tests/lambda_service_test.py

@@ -6,6 +6,7 @@
 
 import pytest
 
+from aws_durable_execution_sdk_python.__about__ import __sdk_name__, __version__
 from aws_durable_execution_sdk_python.exceptions import (
     CallableRuntimeError,
     CheckpointError,
@@ -39,6 +40,7 @@
     WaitOptions,
 )
 
+
 # =============================================================================
 # Fixtures
 # =============================================================================
@@ -1940,6 +1942,7 @@ def test_lambda_client_initialize_client_default(
     config = call_args[1]["config"]
     assert config.connect_timeout == 5
     assert config.read_timeout == 50
+    assert config.user_agent_extra == f"{__sdk_name__}/{__version__}"
     assert isinstance(client, LambdaClient)
 
 
@@ -1963,6 +1966,7 @@ def test_lambda_client_initialize_client_with_endpoint(
     config = call_args[1]["config"]
     assert config.connect_timeout == 5
     assert config.read_timeout == 50
+    assert config.user_agent_extra == f"{__sdk_name__}/{__version__}"
     assert isinstance(client, LambdaClient)
 
 
@@ -2038,6 +2042,8 @@ def test_lambda_client_initialize_client_no_endpoint(
     call_args = mock_boto_client.call_args
     assert call_args[0] == ("lambda",)
     assert "config" in call_args[1]
+    config = call_args[1]["config"]
+    assert config.user_agent_extra == f"{__sdk_name__}/{__version__}"
     assert isinstance(client, LambdaClient)