feat: update user agent with proxy info (#225)

* Include mcp-proxy-for-aws in the user agent

* Update tests

* Update uv.lock

Author: arnewouters

mcp_proxy_for_aws/server.py

@@ -28,6 +28,7 @@
 from fastmcp.server.middleware.error_handling import RetryMiddleware
 from fastmcp.server.middleware.logging import LoggingMiddleware
 from fastmcp.server.server import FastMCP
+from mcp_proxy_for_aws import __version__
 from mcp_proxy_for_aws.cli import parse_args
 from mcp_proxy_for_aws.logging_config import configure_logging
 from mcp_proxy_for_aws.middleware.initialize_middleware import InitializeMiddleware
@@ -89,6 +90,7 @@ async def run_proxy(args) -> None:
         proxy = AWSMCPProxy(
             client_factory=client_factory,
             name='MCP Proxy for AWS',
+            version=__version__,
             instructions=(
                 'MCP Proxy for AWS provides access to SigV4 protected MCP servers through a single interface. '
                 'This proxy handles authentication and request routing to the appropriate backend services.'

mcp_proxy_for_aws/sigv4_helper.py

@@ -22,6 +22,8 @@
 from botocore.awsrequest import AWSRequest
 from botocore.credentials import Credentials
 from functools import partial
+from httpx import __version__ as httpx_version
+from mcp_proxy_for_aws import __version__
 from typing import Any, Dict, Generator, Optional
 
 
@@ -154,8 +156,13 @@ def create_sigv4_client(
         **kwargs,
     }
 
+    user_agent = f'python-httpx/{httpx_version} mcp-proxy-for-aws/{__version__}'
+
     # Add headers if provided
-    default_headers = {'Accept': 'application/json, text/event-stream'}
+    default_headers = {
+        'Accept': 'application/json, text/event-stream',
+        'User-Agent': user_agent,
+    }
     if headers is not None:
         default_headers.update(headers)
     client_kwargs['headers'] = default_headers

tests/unit/test_sigv4_helper.py

@@ -16,6 +16,8 @@
 
 import httpx
 import pytest
+from httpx import __version__ as httpx_version
+from mcp_proxy_for_aws import __version__
 from mcp_proxy_for_aws.sigv4_helper import (
     SENSITIVE_HEADERS,
     SigV4HTTPXAuth,
@@ -142,6 +144,8 @@ def test_create_sigv4_client_default(self, mock_client_class, mock_create_sessio
         assert len(call_args[1]['event_hooks']['response']) == 1
         assert len(call_args[1]['event_hooks']['request']) == 2  # metadata + sign hooks
         assert call_args[1]['headers']['Accept'] == 'application/json, text/event-stream'
+        expected_user_agent = f'python-httpx/{httpx_version} mcp-proxy-for-aws/{__version__}'
+        assert call_args[1]['headers']['User-Agent'] == expected_user_agent
         assert result == mock_client
 
     @patch('mcp_proxy_for_aws.sigv4_helper.create_aws_session')
@@ -163,6 +167,7 @@ def test_create_sigv4_client_with_custom_headers(self, mock_client_class, mock_c
         call_args = mock_client_class.call_args
         expected_headers = {
             'Accept': 'application/json, text/event-stream',
+            'User-Agent': f'python-httpx/{httpx_version} mcp-proxy-for-aws/{__version__}',
             'Custom-Header': 'custom-value',
         }
         assert call_args[1]['headers'] == expected_headers
@@ -245,6 +250,7 @@ def test_create_sigv4_client_with_prompt_context(self, mock_client_class, mock_c
         # Verify headers include both default and prompt-context headers
         expected_headers = {
             'Accept': 'application/json, text/event-stream',
+            'User-Agent': f'python-httpx/{httpx_version} mcp-proxy-for-aws/{__version__}',
             'X-MCP-Prompt-Context': 'enabled',
             'Content-Type': 'application/json',
         }