fix: add kwargs to http client factory

Author: wzxxing

mcp_proxy_for_aws/utils.py

@@ -57,6 +57,7 @@ def client_factory(
         headers: Optional[Dict[str, str]] = None,
         timeout: Optional[httpx.Timeout] = None,
         auth: Optional[httpx.Auth] = None,
+        **kw,
     ) -> httpx.AsyncClient:
         return create_sigv4_client(
             service=service,
@@ -66,6 +67,7 @@ def client_factory(
             timeout=custom_timeout,
             metadata=metadata,
             auth=auth,
+            **kw,
         )
 
     return StreamableHttpTransport(

tests/integ/mcp/simple_mcp_client.py

@@ -89,6 +89,8 @@ def _build_args(endpoint: str, region_name: str, metadata: Optional[Dict[str, st
         'DEBUG',
         '--region',
         region_name,
+        '--profile',
+        'github-integ',
     ]
 
     # Add metadata arguments if provided

tests/unit/test_utils.py

@@ -117,6 +117,42 @@ def test_create_transport_with_sigv4_no_profile(
             # If we can't access the factory directly, just verify the transport was created
             assert result is not None
 
+    @patch('mcp_proxy_for_aws.utils.create_aws_session')
+    @patch('mcp_proxy_for_aws.utils.create_sigv4_client')
+    def test_create_transport_with_sigv4_kwargs_passthrough(
+        self, mock_create_sigv4_client, mock_create_session
+    ):
+        """Test that kwargs are passed through to create_sigv4_client."""
+        from httpx import Timeout
+
+        mock_session = MagicMock()
+        mock_create_session.return_value = mock_session
+
+        url = 'https://test-service.us-west-2.api.aws/mcp'
+        service = 'test-service'
+        region = 'test-region'
+        metadata = {'AWS_REGION': 'test-region'}
+        custom_timeout = Timeout(60.0)
+
+        result = create_transport_with_sigv4(url, service, region, metadata, custom_timeout)
+
+        if hasattr(result, 'httpx_client_factory') and result.httpx_client_factory:
+            factory = result.httpx_client_factory
+            factory(headers=None, timeout=None, auth=None, follow_redirects=True)
+
+            mock_create_sigv4_client.assert_called_once_with(
+                service=service,
+                session=mock_session,
+                region=region,
+                headers=None,
+                timeout=custom_timeout,
+                auth=None,
+                metadata=metadata,
+                follow_redirects=True,
+            )
+        else:
+            assert result is not None
+
 
 class TestValidateRequiredArgs:
     """Test cases for validate_service_name function."""