Make endpoint positional argument (#12)

* Make endpoint positional argument

* Update DEVELOPMENT.md

Co-authored-by: Leonardo Araneda Freccero <arangatang@users.noreply.github.com>
Signed-off-by: Anna Lau <66322063+acmlau@users.noreply.github.com>

---------

Signed-off-by: Anna Lau <66322063+acmlau@users.noreply.github.com>
Co-authored-by: Leonardo Araneda Freccero <arangatang@users.noreply.github.com>

Author: acmlau

DEVELOPMENT.md

@@ -1,13 +1,12 @@
 ## Quickstart
 ### Running local code
 ```
-uv run src/aws_mcp_proxy/server.py --endpoint <your endpoint>
+uv run aws_mcp_proxy/server.py <your endpoint>
 ```
 
 ### Running local code with MCP inspector
 ```
 npx @modelcontextprotocol/inspector uv run \
-src/aws_mcp_proxy/server.py \
---endpoint <your endpoint>
+aws_mcp_proxy/server.py <your endpoint>
 ```
 Then click connect in the opened browser window.

README.md

@@ -21,7 +21,7 @@ The proxy handles SigV4 authentication using local AWS credentials and provides
 ### Quick start
 
 ```bash
-uv run src/aws_mcp_proxy/server.py --endpoint <a sigv4 mcp>
+uv run aws_mcp_proxy/server.py <a sigv4 mcp>
 ```
 
 #### Details
@@ -49,10 +49,9 @@ Example with all options
       "command": "uv",
       "args": [
         "--directory",
-        "/path/to/aws_mcp_proxy/src/aws_mcp_proxy",
+        "/path/to/aws_mcp_proxy",
         "run",
-        "server.py",
-        "--endpoint",
+        "aws_mcp_proxy/server.py",
         "<remote-server-url>",
         "--service",
         "<service-code>",
@@ -69,7 +68,7 @@ Example with all options
 
 In MCP mode, the backend server is configured directly through command-line arguments:
 
-* `--endpoint`: The MCP endpoint URL (required)
+* `endpoint`: The MCP endpoint URL (required, first positional argument)
 * `--service`: AWS service name for SigV4 signing (optional, inferred from endpoint if not provided)
 * `--profile`: AWS profile to use (optional, uses AWS_PROFILE environment variable if not provided)
 

aws_mcp_proxy/server.py

@@ -75,19 +75,18 @@ def parse_args():
         epilog="""
 Examples:
   # Run with EKS MCP endpoint
-  aws-mcp-proxy --endpoint https://eks-mcp.us-west-2.api.aws
+  aws-mcp-proxy https://eks-mcp.us-west-2.api.aws
 
   # Run with custom service and profile
-  aws-mcp-proxy --endpoint https://eks-mcp.us-west-2.api.aws --service eks-mcp --profile default
+  aws-mcp-proxy https://eks-mcp.us-west-2.api.aws --service eks-mcp --profile default
 
   # Run with write permissions enabled
-  aws-mcp-proxy --endpoint https://eks-mcp.us-west-2.api.aws --allow-write
+  aws-mcp-proxy https://eks-mcp.us-west-2.api.aws --allow-write
         """,
     )
 
     parser.add_argument(
-        '--endpoint',
-        required=True,
+        'endpoint',
         help='MCP endpoint URL (e.g., https://eks-mcp.us-west-2.api.aws)',
     )
 

tests/test_main.py

@@ -24,7 +24,7 @@ class TestMain:
     @patch('aws_mcp_proxy.server.asyncio.run')
     @patch('aws_mcp_proxy.server.setup_mcp_mode')
     @patch('aws_mcp_proxy.server.FastMCP')
-    @patch('sys.argv', ['aws-mcp-proxy', '--endpoint', 'https://test.example.com'])
+    @patch('sys.argv', ['aws-mcp-proxy', 'https://test.example.com'])
     def test_main_default(self, mock_fastmcp, mock_setup_mcp, mock_asyncio_run):
         """Test main function with default arguments."""
         # Create mock FastMCP instance

tests/test_server.py

@@ -127,14 +127,14 @@ async def test_setup_mcp_mode_tool_registration_error(
             await setup_mcp_mode(mock_mcp, mock_args)
         assert 'Tool registration failed' in str(exc_info.value)
 
-    @patch('sys.argv', ['test', '--endpoint', 'https://test.example.com'])
+    @patch('sys.argv', ['test', 'https://test.example.com'])
     def test_parse_args_default(self):
         """Test parse_args with default arguments."""
         args = parse_args()
         assert args.endpoint == 'https://test.example.com'
 
     @patch('aws_mcp_proxy.server.asyncio.run')
-    @patch('sys.argv', ['test', '--endpoint', 'https://test.example.com'])
+    @patch('sys.argv', ['test', 'https://test.example.com'])
     def test_main_function(self, mock_asyncio_run):
         """Test that main function runs server correctly."""
         # Arrange
@@ -147,7 +147,7 @@ def test_main_function(self, mock_asyncio_run):
         mock_asyncio_run.assert_called_once()
 
     @patch('aws_mcp_proxy.server.asyncio.run')
-    @patch('sys.argv', ['test', '--endpoint', 'https://test.example.com'])
+    @patch('sys.argv', ['test', 'https://test.example.com'])
     def test_main_error_handling(self, mock_asyncio_run):
         """Test that main function handles errors gracefully."""
         # Arrange