test(sagemaker): tests for ep additional_args

Add a test to check for insertion of endpoint config additional_args

Author: athewsey

src/strands/models/sagemaker.py

@@ -283,7 +283,7 @@ def format_request(
         if target_variant:
             request["TargetVariant"] = target_variant
 
-        # Add additional args if provided
+        # Add additional request args if provided
         additional_args = self.endpoint_config.get("additional_args")
         if additional_args:
             request.update(additional_args)

tests/strands/models/test_sagemaker.py

@@ -112,6 +112,7 @@ def test_init_with_all_params(self, boto_session):
             "endpoint_name": "test-endpoint",
             "inference_component_name": "test-component",
             "region_name": "us-west-2",
+            "additional_args": {"test_arg_name": "test_arg_value"},
         }
         payload_config = {
             "stream": False,
@@ -129,6 +130,7 @@ def test_init_with_all_params(self, boto_session):
 
         assert model.endpoint_config["endpoint_name"] == "test-endpoint"
         assert model.endpoint_config["inference_component_name"] == "test-component"
+        assert model.endpoint_config["additional_args"]["test_arg_name"] == "test_arg_value"
         assert model.payload_config["stream"] is False
         assert model.payload_config["max_tokens"] == 1024
         assert model.payload_config["temperature"] == 0.7
@@ -239,6 +241,22 @@ def test_get_config(self, model, endpoint_config):
     #     assert "tools" in payload
     #     assert payload["tools"] == []
 
+    def test_format_request_with_additional_args(self, boto_session, endpoint_config, messages, payload_config):
+        """Test formatting a request's `additional_args` where provided"""
+        endpoint_config_ext = {
+            **endpoint_config,
+            "additional_args": {
+                "extra_key": "extra_value",
+            },
+        }
+        model = SageMakerAIModel(
+            boto_session=boto_session,
+            endpoint_config=endpoint_config_ext,
+            payload_config=payload_config,
+        )
+        request = model.format_request(messages)
+        assert request.get("extra_key") == "extra_value"
+
     @pytest.mark.asyncio
     async def test_stream_with_streaming_enabled(self, sagemaker_client, model, messages):
         """Test streaming response with streaming enabled."""