awslabs.cloudwatch-appsignals-mcp-server 0.1.3__tar.gz → 0.1.5__tar.gz

{awslabs_cloudwatch_appsignals_mcp_server-0.1.3 → awslabs_cloudwatch_appsignals_mcp_server-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: awslabs.cloudwatch-appsignals-mcp-server
-Version: 0.1.3
+Version: 0.1.5
 Summary: An AWS Labs Model Context Protocol (MCP) server for AWS Application Signals
 Project-URL: Homepage, https://awslabs.github.io/mcp/
 Project-URL: Documentation, https://awslabs.github.io/mcp/servers/cloudwatch-appsignals-mcp-server/
@@ -82,7 +82,7 @@ This server enables AI assistants like Claude, GitHub Copilot, and Amazon Q to h
 
 | Cursor | VS Code |
 |:------:|:-------:|
-| [![Install MCP Server](https://cursor.com/deeplink/mcp-install-light.svg)](https://cursor.com/install-mcp?name=awslabs.cloudwatch-appsignals-mcp-server&config=eyJhdXRvQXBwcm92ZSI6W10sImRpc2FibGVkIjpmYWxzZSwidGltZW91dCI6NjAsImNvbW1hbmQiOiJ1dnggYXdzbGFicy5jbG91ZHdhdGNoLWFwcHNpZ25hbHMtbWNwLXNlcnZlckBsYXRlc3QiLCJlbnYiOnsiQVdTX1BST0ZJTEUiOiJbVGhlIEFXUyBQcm9maWxlIE5hbWUgdG8gdXNlIGZvciBBV1MgYWNjZXNzXSIsIkFXU19SRUdJT04iOiJbVGhlIEFXUyByZWdpb24gdG8gcnVuIGluXSIsIkZBU1RNQ1BfTE9HX0xFVkVMIjoiRVJST1IifSwidHJhbnNwb3J0VHlwZSI6InN0ZGlvIn0%3D) | [![Install on VS Code](https://img.shields.io/badge/Install_on-VS_Code-FF9900?style=flat-square&logo=visualstudiocode&logoColor=white)](https://insiders.vscode.dev/redirect/mcp/install?name=CloudWatch%20Application%20Signals%20MCP%20Server&config=%7B%22autoApprove%22%3A%5B%5D%2C%22disabled%22%3Afalse%2C%22timeout%22%3A60%2C%22command%22%3A%22uvx%22%2C%22args%22%3A%5B%22awslabs.cloudwatch-appsignals-mcp-server%40latest%22%5D%2C%22env%22%3A%7B%22AWS_PROFILE%22%3A%22%5BThe%20AWS%20Profile%20Name%20to%20use%20for%20AWS%20access%5D%22%2C%22AWS_REGION%22%3A%22%5BThe%20AWS%20region%20to%20run%20in%5D%22%2C%22FASTMCP_LOG_LEVEL%22%3A%22ERROR%22%7D%2C%22transportType%22%3A%22stdio%22%7D) |
+| [![Install MCP Server](https://cursor.com/deeplink/mcp-install-light.svg)](https://cursor.com/en/install-mcp?name=awslabs.cloudwatch-appsignals-mcp-server&config=eyJhdXRvQXBwcm92ZSI6W10sImRpc2FibGVkIjpmYWxzZSwidGltZW91dCI6NjAsImNvbW1hbmQiOiJ1dnggYXdzbGFicy5jbG91ZHdhdGNoLWFwcHNpZ25hbHMtbWNwLXNlcnZlckBsYXRlc3QiLCJlbnYiOnsiQVdTX1BST0ZJTEUiOiJbVGhlIEFXUyBQcm9maWxlIE5hbWUgdG8gdXNlIGZvciBBV1MgYWNjZXNzXSIsIkFXU19SRUdJT04iOiJbVGhlIEFXUyByZWdpb24gdG8gcnVuIGluXSIsIkZBU1RNQ1BfTE9HX0xFVkVMIjoiRVJST1IifSwidHJhbnNwb3J0VHlwZSI6InN0ZGlvIn0%3D) | [![Install on VS Code](https://img.shields.io/badge/Install_on-VS_Code-FF9900?style=flat-square&logo=visualstudiocode&logoColor=white)](https://insiders.vscode.dev/redirect/mcp/install?name=CloudWatch%20Application%20Signals%20MCP%20Server&config=%7B%22autoApprove%22%3A%5B%5D%2C%22disabled%22%3Afalse%2C%22timeout%22%3A60%2C%22command%22%3A%22uvx%22%2C%22args%22%3A%5B%22awslabs.cloudwatch-appsignals-mcp-server%40latest%22%5D%2C%22env%22%3A%7B%22AWS_PROFILE%22%3A%22%5BThe%20AWS%20Profile%20Name%20to%20use%20for%20AWS%20access%5D%22%2C%22AWS_REGION%22%3A%22%5BThe%20AWS%20region%20to%20run%20in%5D%22%2C%22FASTMCP_LOG_LEVEL%22%3A%22ERROR%22%7D%2C%22transportType%22%3A%22stdio%22%7D) |
 
 ### Installing via `uv`
 
@@ -96,17 +96,15 @@ use [`uvx`](https://docs.astral.sh/uv/guides/tools/) to directly run *awslabs.cl
 ```json
 {
   "mcpServers": {
-    "awslabs.cloudwatch-appsignals-mcp-server": {
+    "awslabs.cloudwatch-appsignals-mcp": {
       "autoApprove": [],
       "disabled": false,
-      "timeout": 60,
       "command": "uvx",
       "args": [
         "awslabs.cloudwatch-appsignals-mcp-server@latest"
       ],
       "env": {
-        "AWS_ACCESS_KEY_ID": "[AWS Access Key ID]",
-        "AWS_SECRET_ACCESS_KEY": "[AWS Access Key]",
+        "AWS_PROFILE": "[The AWS Profile Name to use for AWS access]",
         "AWS_REGION": "[AWS Region]",
         "FASTMCP_LOG_LEVEL": "ERROR"
       },
@@ -130,7 +128,11 @@ On Windows: `%APPDATA%/Claude/claude_desktop_config.json`
     "mcpServers": {
       "awslabs.cloudwatch-appsignals-mcp-server": {
         "command": "uvx",
-        "args": ["--from", "/absolute/path/to/cloudwatch-appsignals-mcp-server", "awslabs.cloudwatch-appsignals-mcp-server"]
+        "args": ["--from", "/absolute/path/to/cloudwatch-appsignals-mcp-server", "awslabs.cloudwatch-appsignals-mcp-server"],
+        "env": {
+          "AWS_PROFILE": "[The AWS Profile Name to use for AWS access]",
+          "AWS_REGION": "[AWS Region]"
+        }
       }
     }
   }
@@ -145,7 +147,11 @@ On Windows: `%APPDATA%/Claude/claude_desktop_config.json`
     "mcpServers": {
       "awslabs.cloudwatch-appsignals-mcp-server": {
         "command": "uvx",
-        "args": ["awslabs.cloudwatch-appsignals-mcp-server@latest"]
+        "args": ["awslabs.cloudwatch-appsignals-mcp-server@latest"],
+        "env": {
+          "AWS_PROFILE": "[The AWS Profile Name to use for AWS access]",
+          "AWS_REGION": "[AWS Region]"
+        }
       }
     }
   }
@@ -168,9 +174,9 @@ On Windows: `%APPDATA%/Claude/claude_desktop_config.json`
         "run",
         "-i",
         "--rm",
-        "-e", "AWS_ACCESS_KEY_ID=[your data]",
-        "-e", "AWS_SECRET_ACCESS_KEY=[your data]",
-        "-e", "AWS_REGION=[your data]",
+        "-v", "${HOME}/.aws:/root/.aws:ro",
+        "-e", "AWS_PROFILE=[The AWS Profile Name to use for AWS access]",
+        "-e", "AWS_REGION=[AWS Region]",
         "awslabs/cloudwatch-appsignals-mcp-server:latest"
       ]
     }
@@ -293,19 +299,18 @@ The server requires the following AWS IAM permissions:
 
 ### Environment Variables
 
+- `AWS_PROFILE` - AWS profile name to use for authentication (defaults to `default` profile)
 - `AWS_REGION` - AWS region (defaults to us-east-1)
 - `MCP_CLOUDWATCH_APPSIGNALS_LOG_LEVEL` - Logging level (defaults to INFO)
 
 ### AWS Credentials
 
-This server uses the standard AWS credential chain via boto3. It will automatically use credentials from:
-- Environment variables (`AWS_ACCESS_KEY_ID`, `AWS_SECRET_ACCESS_KEY`, etc.)
-- AWS credentials file (`~/.aws/credentials`)
-- AWS config file (`~/.aws/config`)
-- IAM roles (when running on EC2, ECS, Lambda, etc.)
-- And other standard AWS credential providers
+This server uses AWS profiles for authentication. Set the `AWS_PROFILE` environment variable to use a specific profile from your `~/.aws/credentials` file.
 
-No additional credential configuration is needed beyond your standard AWS setup.
+The server will use the standard AWS credential chain via boto3, which includes:
+- AWS Profile specified by `AWS_PROFILE` environment variable
+- Default profile from AWS credentials file
+- IAM roles when running on EC2, ECS, Lambda, etc.
 
 ## Development
 

{awslabs_cloudwatch_appsignals_mcp_server-0.1.3 → awslabs_cloudwatch_appsignals_mcp_server-0.1.5}/README.md

@@ -52,7 +52,7 @@ This server enables AI assistants like Claude, GitHub Copilot, and Amazon Q to h
 
 | Cursor | VS Code |
 |:------:|:-------:|
-| [![Install MCP Server](https://cursor.com/deeplink/mcp-install-light.svg)](https://cursor.com/install-mcp?name=awslabs.cloudwatch-appsignals-mcp-server&config=eyJhdXRvQXBwcm92ZSI6W10sImRpc2FibGVkIjpmYWxzZSwidGltZW91dCI6NjAsImNvbW1hbmQiOiJ1dnggYXdzbGFicy5jbG91ZHdhdGNoLWFwcHNpZ25hbHMtbWNwLXNlcnZlckBsYXRlc3QiLCJlbnYiOnsiQVdTX1BST0ZJTEUiOiJbVGhlIEFXUyBQcm9maWxlIE5hbWUgdG8gdXNlIGZvciBBV1MgYWNjZXNzXSIsIkFXU19SRUdJT04iOiJbVGhlIEFXUyByZWdpb24gdG8gcnVuIGluXSIsIkZBU1RNQ1BfTE9HX0xFVkVMIjoiRVJST1IifSwidHJhbnNwb3J0VHlwZSI6InN0ZGlvIn0%3D) | [![Install on VS Code](https://img.shields.io/badge/Install_on-VS_Code-FF9900?style=flat-square&logo=visualstudiocode&logoColor=white)](https://insiders.vscode.dev/redirect/mcp/install?name=CloudWatch%20Application%20Signals%20MCP%20Server&config=%7B%22autoApprove%22%3A%5B%5D%2C%22disabled%22%3Afalse%2C%22timeout%22%3A60%2C%22command%22%3A%22uvx%22%2C%22args%22%3A%5B%22awslabs.cloudwatch-appsignals-mcp-server%40latest%22%5D%2C%22env%22%3A%7B%22AWS_PROFILE%22%3A%22%5BThe%20AWS%20Profile%20Name%20to%20use%20for%20AWS%20access%5D%22%2C%22AWS_REGION%22%3A%22%5BThe%20AWS%20region%20to%20run%20in%5D%22%2C%22FASTMCP_LOG_LEVEL%22%3A%22ERROR%22%7D%2C%22transportType%22%3A%22stdio%22%7D) |
+| [![Install MCP Server](https://cursor.com/deeplink/mcp-install-light.svg)](https://cursor.com/en/install-mcp?name=awslabs.cloudwatch-appsignals-mcp-server&config=eyJhdXRvQXBwcm92ZSI6W10sImRpc2FibGVkIjpmYWxzZSwidGltZW91dCI6NjAsImNvbW1hbmQiOiJ1dnggYXdzbGFicy5jbG91ZHdhdGNoLWFwcHNpZ25hbHMtbWNwLXNlcnZlckBsYXRlc3QiLCJlbnYiOnsiQVdTX1BST0ZJTEUiOiJbVGhlIEFXUyBQcm9maWxlIE5hbWUgdG8gdXNlIGZvciBBV1MgYWNjZXNzXSIsIkFXU19SRUdJT04iOiJbVGhlIEFXUyByZWdpb24gdG8gcnVuIGluXSIsIkZBU1RNQ1BfTE9HX0xFVkVMIjoiRVJST1IifSwidHJhbnNwb3J0VHlwZSI6InN0ZGlvIn0%3D) | [![Install on VS Code](https://img.shields.io/badge/Install_on-VS_Code-FF9900?style=flat-square&logo=visualstudiocode&logoColor=white)](https://insiders.vscode.dev/redirect/mcp/install?name=CloudWatch%20Application%20Signals%20MCP%20Server&config=%7B%22autoApprove%22%3A%5B%5D%2C%22disabled%22%3Afalse%2C%22timeout%22%3A60%2C%22command%22%3A%22uvx%22%2C%22args%22%3A%5B%22awslabs.cloudwatch-appsignals-mcp-server%40latest%22%5D%2C%22env%22%3A%7B%22AWS_PROFILE%22%3A%22%5BThe%20AWS%20Profile%20Name%20to%20use%20for%20AWS%20access%5D%22%2C%22AWS_REGION%22%3A%22%5BThe%20AWS%20region%20to%20run%20in%5D%22%2C%22FASTMCP_LOG_LEVEL%22%3A%22ERROR%22%7D%2C%22transportType%22%3A%22stdio%22%7D) |
 
 ### Installing via `uv`
 
@@ -66,17 +66,15 @@ use [`uvx`](https://docs.astral.sh/uv/guides/tools/) to directly run *awslabs.cl
 ```json
 {
   "mcpServers": {
-    "awslabs.cloudwatch-appsignals-mcp-server": {
+    "awslabs.cloudwatch-appsignals-mcp": {
       "autoApprove": [],
       "disabled": false,
-      "timeout": 60,
       "command": "uvx",
       "args": [
         "awslabs.cloudwatch-appsignals-mcp-server@latest"
       ],
       "env": {
-        "AWS_ACCESS_KEY_ID": "[AWS Access Key ID]",
-        "AWS_SECRET_ACCESS_KEY": "[AWS Access Key]",
+        "AWS_PROFILE": "[The AWS Profile Name to use for AWS access]",
         "AWS_REGION": "[AWS Region]",
         "FASTMCP_LOG_LEVEL": "ERROR"
       },
@@ -100,7 +98,11 @@ On Windows: `%APPDATA%/Claude/claude_desktop_config.json`
     "mcpServers": {
       "awslabs.cloudwatch-appsignals-mcp-server": {
         "command": "uvx",
-        "args": ["--from", "/absolute/path/to/cloudwatch-appsignals-mcp-server", "awslabs.cloudwatch-appsignals-mcp-server"]
+        "args": ["--from", "/absolute/path/to/cloudwatch-appsignals-mcp-server", "awslabs.cloudwatch-appsignals-mcp-server"],
+        "env": {
+          "AWS_PROFILE": "[The AWS Profile Name to use for AWS access]",
+          "AWS_REGION": "[AWS Region]"
+        }
       }
     }
   }
@@ -115,7 +117,11 @@ On Windows: `%APPDATA%/Claude/claude_desktop_config.json`
     "mcpServers": {
       "awslabs.cloudwatch-appsignals-mcp-server": {
         "command": "uvx",
-        "args": ["awslabs.cloudwatch-appsignals-mcp-server@latest"]
+        "args": ["awslabs.cloudwatch-appsignals-mcp-server@latest"],
+        "env": {
+          "AWS_PROFILE": "[The AWS Profile Name to use for AWS access]",
+          "AWS_REGION": "[AWS Region]"
+        }
       }
     }
   }
@@ -138,9 +144,9 @@ On Windows: `%APPDATA%/Claude/claude_desktop_config.json`
         "run",
         "-i",
         "--rm",
-        "-e", "AWS_ACCESS_KEY_ID=[your data]",
-        "-e", "AWS_SECRET_ACCESS_KEY=[your data]",
-        "-e", "AWS_REGION=[your data]",
+        "-v", "${HOME}/.aws:/root/.aws:ro",
+        "-e", "AWS_PROFILE=[The AWS Profile Name to use for AWS access]",
+        "-e", "AWS_REGION=[AWS Region]",
         "awslabs/cloudwatch-appsignals-mcp-server:latest"
       ]
     }
@@ -263,19 +269,18 @@ The server requires the following AWS IAM permissions:
 
 ### Environment Variables
 
+- `AWS_PROFILE` - AWS profile name to use for authentication (defaults to `default` profile)
 - `AWS_REGION` - AWS region (defaults to us-east-1)
 - `MCP_CLOUDWATCH_APPSIGNALS_LOG_LEVEL` - Logging level (defaults to INFO)
 
 ### AWS Credentials
 
-This server uses the standard AWS credential chain via boto3. It will automatically use credentials from:
-- Environment variables (`AWS_ACCESS_KEY_ID`, `AWS_SECRET_ACCESS_KEY`, etc.)
-- AWS credentials file (`~/.aws/credentials`)
-- AWS config file (`~/.aws/config`)
-- IAM roles (when running on EC2, ECS, Lambda, etc.)
-- And other standard AWS credential providers
+This server uses AWS profiles for authentication. Set the `AWS_PROFILE` environment variable to use a specific profile from your `~/.aws/credentials` file.
 
-No additional credential configuration is needed beyond your standard AWS setup.
+The server will use the standard AWS credential chain via boto3, which includes:
+- AWS Profile specified by `AWS_PROFILE` environment variable
+- Default profile from AWS credentials file
+- IAM roles when running on EC2, ECS, Lambda, etc.
 
 ## Development
 

{awslabs_cloudwatch_appsignals_mcp_server-0.1.3 → awslabs_cloudwatch_appsignals_mcp_server-0.1.5}/awslabs/cloudwatch_appsignals_mcp_server/__init__.py

@@ -14,4 +14,4 @@
 
 """AWS Application Signals MCP Server."""
 
-__version__ = '0.1.0'
+__version__ = '0.1.5'

{awslabs_cloudwatch_appsignals_mcp_server-0.1.3 → awslabs_cloudwatch_appsignals_mcp_server-0.1.5}/awslabs/cloudwatch_appsignals_mcp_server/server.py

@@ -44,14 +44,33 @@ logger.debug(f'CloudWatch AppSignals MCP Server initialized with log level: {log
 AWS_REGION = os.environ.get('AWS_REGION', 'us-east-1')
 logger.debug(f'Using AWS region: {AWS_REGION}')
 
-# Initialize AWS clients with logging
-try:
+
+# Initialize AWS clients
+def _initialize_aws_clients():
+    """Initialize AWS clients with proper configuration."""
     config = Config(user_agent_extra=f'awslabs.cloudwatch-appsignals-mcp-server/{__version__}')
-    logs_client = boto3.client('logs', region_name=AWS_REGION, config=config)
-    appsignals_client = boto3.client('application-signals', region_name=AWS_REGION, config=config)
-    cloudwatch_client = boto3.client('cloudwatch', region_name=AWS_REGION, config=config)
-    xray_client = boto3.client('xray', region_name=AWS_REGION, config=config)
+
+    # Check for AWS_PROFILE environment variable
+    if aws_profile := os.environ.get('AWS_PROFILE'):
+        logger.debug(f'Using AWS profile: {aws_profile}')
+        session = boto3.Session(profile_name=aws_profile, region_name=AWS_REGION)
+        logs = session.client('logs', config=config)
+        appsignals = session.client('application-signals', config=config)
+        cloudwatch = session.client('cloudwatch', config=config)
+        xray = session.client('xray', config=config)
+    else:
+        logs = boto3.client('logs', region_name=AWS_REGION, config=config)
+        appsignals = boto3.client('application-signals', region_name=AWS_REGION, config=config)
+        cloudwatch = boto3.client('cloudwatch', region_name=AWS_REGION, config=config)
+        xray = boto3.client('xray', region_name=AWS_REGION, config=config)
+
     logger.debug('AWS clients initialized successfully')
+    return logs, appsignals, cloudwatch, xray
+
+
+# Initialize clients at module level
+try:
+    logs_client, appsignals_client, cloudwatch_client, xray_client = _initialize_aws_clients()
 except Exception as e:
     logger.error(f'Failed to initialize AWS clients: {str(e)}')
     raise

{awslabs_cloudwatch_appsignals_mcp_server-0.1.3 → awslabs_cloudwatch_appsignals_mcp_server-0.1.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "awslabs.cloudwatch-appsignals-mcp-server"
-version = "0.1.3"
+version = "0.1.5"
 description = "An AWS Labs Model Context Protocol (MCP) server for AWS Application Signals"
 readme = "README.md"
 requires-python = ">=3.10"

awslabs_cloudwatch_appsignals_mcp_server-0.1.5/tests/conftest.py

@@ -0,0 +1,11 @@
+"""Pytest configuration for CloudWatch Application Signals MCP Server tests."""
+
+import os
+
+
+# Set test environment variables before any imports
+os.environ['AWS_ACCESS_KEY_ID'] = 'testing'
+os.environ['AWS_SECRET_ACCESS_KEY'] = 'testing'  # pragma: allowlist secret
+os.environ['AWS_SECURITY_TOKEN'] = 'testing'
+os.environ['AWS_SESSION_TOKEN'] = 'testing'
+os.environ.pop('AWS_PROFILE', None)

awslabs_cloudwatch_appsignals_mcp_server-0.1.5/tests/test_aws_profile.py

@@ -0,0 +1,145 @@
+"""Test AWS profile handling in isolation."""
+
+import os
+import pytest
+from unittest.mock import MagicMock, patch
+
+
+def test_aws_profile_branch_coverage():
+    """Test the AWS_PROFILE environment variable branch coverage."""
+    # Test the condition when AWS_PROFILE is set
+    with patch.dict(os.environ, {'AWS_PROFILE': 'test-profile'}):
+        assert os.environ.get('AWS_PROFILE') == 'test-profile'
+
+        # Test walrus operator assignment
+        if aws_profile := os.environ.get('AWS_PROFILE'):
+            assert aws_profile == 'test-profile'
+
+    # Test the condition when AWS_PROFILE is not set
+    with patch.dict(os.environ, {}, clear=True):
+        assert os.environ.get('AWS_PROFILE') is None
+
+        # Test walrus operator assignment with None
+        if aws_profile := os.environ.get('AWS_PROFILE'):
+            pytest.fail('Should not enter this branch when AWS_PROFILE is not set')
+        else:
+            assert aws_profile is None
+
+
+def test_aws_client_initialization_flow():
+    """Test the client initialization flow with and without AWS_PROFILE."""
+    import boto3
+    from botocore.config import Config
+
+    # Mock config
+    config = MagicMock(spec=Config)
+
+    # Test with AWS_PROFILE
+    with patch('boto3.Session') as mock_session:
+        mock_session_instance = MagicMock()
+        mock_session.return_value = mock_session_instance
+
+        # Simulate the server initialization logic
+        aws_profile = 'test-profile'
+        AWS_REGION = 'us-west-2'
+
+        if aws_profile:
+            session = boto3.Session(profile_name=aws_profile, region_name=AWS_REGION)
+            session.client('logs', config=config)
+            session.client('application-signals', config=config)
+            session.client('cloudwatch', config=config)
+            session.client('xray', config=config)
+
+            # Verify calls
+            mock_session.assert_called_once_with(
+                profile_name='test-profile', region_name='us-west-2'
+            )
+            assert mock_session_instance.client.call_count == 4
+
+    # Test without AWS_PROFILE
+    with patch('boto3.client') as mock_client:
+        AWS_REGION = 'us-east-1'
+        aws_profile = None
+
+        if not aws_profile:
+            boto3.client('logs', region_name=AWS_REGION, config=config)
+            boto3.client('application-signals', region_name=AWS_REGION, config=config)
+            boto3.client('cloudwatch', region_name=AWS_REGION, config=config)
+            boto3.client('xray', region_name=AWS_REGION, config=config)
+
+            # Verify calls
+            assert mock_client.call_count == 4
+            for call in mock_client.call_args_list:
+                assert call.kwargs['region_name'] == 'us-east-1'
+                assert call.kwargs['config'] == config
+
+
+def test_server_initialization_with_aws_profile_coverage():
+    """Test to ensure AWS_PROFILE code path gets coverage."""
+    # This test simulates the exact logic from server.py to ensure coverage
+    import boto3
+    from botocore.config import Config
+
+    # Mock the actual initialization logic
+    mock_session = MagicMock()
+    mock_session_instance = MagicMock()
+    mock_session.return_value = mock_session_instance
+
+    with patch.dict(os.environ, {'AWS_PROFILE': 'test-profile'}):
+        with patch('boto3.Session', mock_session):
+            # This is the exact code from server.py that needs coverage
+            config = MagicMock(spec=Config)
+            AWS_REGION = 'us-east-1'
+
+            # Check for AWS_PROFILE environment variable (exact code from server.py)
+            if aws_profile := os.environ.get('AWS_PROFILE'):
+                # This block needs coverage
+                session = boto3.Session(profile_name=aws_profile, region_name=AWS_REGION)
+                session.client('logs', config=config)
+                session.client('application-signals', config=config)
+                session.client('cloudwatch', config=config)
+                session.client('xray', config=config)
+
+                # Verify the AWS profile was used
+                mock_session.assert_called_once_with(
+                    profile_name='test-profile', region_name='us-east-1'
+                )
+                assert mock_session_instance.client.call_count == 4
+
+
+def test_initialize_aws_clients_with_profile():
+    """Test _initialize_aws_clients function with AWS_PROFILE set."""
+    from awslabs.cloudwatch_appsignals_mcp_server.server import _initialize_aws_clients
+
+    # Mock the necessary components
+    mock_session = MagicMock()
+    mock_session_instance = MagicMock()
+    mock_session.return_value = mock_session_instance
+    mock_client = MagicMock()
+    mock_session_instance.client.return_value = mock_client
+
+    with patch.dict(os.environ, {'AWS_PROFILE': 'test-profile', 'AWS_REGION': 'us-east-1'}):
+        with patch('awslabs.cloudwatch_appsignals_mcp_server.server.boto3.Session', mock_session):
+            with patch('awslabs.cloudwatch_appsignals_mcp_server.server.Config'):
+                # Call the initialization function
+                logs, appsignals, cloudwatch, xray = _initialize_aws_clients()
+
+                # Verify Session was called with the profile
+                mock_session.assert_called_once()
+                call_kwargs = mock_session.call_args[1]
+                assert call_kwargs['profile_name'] == 'test-profile'
+                assert call_kwargs['region_name'] == 'us-east-1'
+
+                # Verify all clients were created
+                assert mock_session_instance.client.call_count == 4
+                client_calls = [call[0][0] for call in mock_session_instance.client.call_args_list]
+                assert 'logs' in client_calls
+                assert 'application-signals' in client_calls
+                assert 'cloudwatch' in client_calls
+                assert 'xray' in client_calls
+
+                # Verify the returned clients
+                assert logs == mock_client
+                assert appsignals == mock_client
+                assert cloudwatch == mock_client
+                assert xray == mock_client