awslabs.cloudwatch-appsignals-mcp-server 0.1.9__tar.gz → 0.1.10__tar.gz

{awslabs_cloudwatch_appsignals_mcp_server-0.1.9 → awslabs_cloudwatch_appsignals_mcp_server-0.1.10}/Dockerfile

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 # dependabot should continue to update this to the latest hash.
-FROM public.ecr.aws/docker/library/python:3.13.5-alpine3.21@sha256:c9a09c45a4bcc618c7f7128585b8dd0d41d0c31a8a107db4c8255ffe0b69375d AS uv
+FROM public.ecr.aws/docker/library/python:3.13-alpine@sha256:070342a0cc1011532c0e69972cce2bbc6cc633eba294bae1d12abea8bd05303b AS uv
 
 # Install the project into `/app`
 WORKDIR /app
@@ -61,7 +61,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
 # Make the directory just in case it doesn't exist
 RUN mkdir -p /root/.local
 
-FROM public.ecr.aws/docker/library/python:3.13.5-alpine3.21@sha256:c9a09c45a4bcc618c7f7128585b8dd0d41d0c31a8a107db4c8255ffe0b69375d
+FROM public.ecr.aws/docker/library/python:3.13-alpine@sha256:070342a0cc1011532c0e69972cce2bbc6cc633eba294bae1d12abea8bd05303b
 
 # Place executables in the environment at the front of the path and include other binaries
 ENV PATH="/app/.venv/bin:$PATH" \

{awslabs_cloudwatch_appsignals_mcp_server-0.1.9 → awslabs_cloudwatch_appsignals_mcp_server-0.1.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: awslabs.cloudwatch-appsignals-mcp-server
-Version: 0.1.9
+Version: 0.1.10
 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/

{awslabs_cloudwatch_appsignals_mcp_server-0.1.9 → awslabs_cloudwatch_appsignals_mcp_server-0.1.10}/awslabs/cloudwatch_appsignals_mcp_server/__init__.py

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

{awslabs_cloudwatch_appsignals_mcp_server-0.1.9 → awslabs_cloudwatch_appsignals_mcp_server-0.1.10}/awslabs/cloudwatch_appsignals_mcp_server/trace_tools.py

@@ -402,6 +402,26 @@ async def query_sampled_traces(
                 return obj.isoformat()
             return obj
 
+        # Helper function to extract fault message from root causes for deduplication
+        def get_fault_message(trace_data):
+            """Extract fault message from a trace for deduplication.
+
+            Only checks FaultRootCauses (5xx server errors) since this is the primary
+            use case for root cause investigation. Traces without fault messages are
+            not deduplicated.
+            """
+            # Only check FaultRootCauses for deduplication
+            root_causes = trace_data.get('FaultRootCauses', [])
+            if root_causes:
+                for cause in root_causes:
+                    services = cause.get('Services', [])
+                    for service in services:
+                        exceptions = service.get('Exceptions', [])
+                        if exceptions and exceptions[0].get('Message'):
+                            return exceptions[0].get('Message')
+            return None
+
+        # Build trace summaries (original format)
         trace_summaries = []
         for trace in traces:
             # Create a simplified trace data structure to reduce size
@@ -417,17 +437,11 @@ async def query_sampled_traces(
 
             # Only include root causes if they exist (to save space)
             if trace.get('ErrorRootCauses'):
-                trace_data['ErrorRootCauses'] = trace.get('ErrorRootCauses', [])[
-                    :3
-                ]  # Limit to first 3
+                trace_data['ErrorRootCauses'] = trace.get('ErrorRootCauses', [])[:3]
             if trace.get('FaultRootCauses'):
-                trace_data['FaultRootCauses'] = trace.get('FaultRootCauses', [])[
-                    :3
-                ]  # Limit to first 3
+                trace_data['FaultRootCauses'] = trace.get('FaultRootCauses', [])[:3]
             if trace.get('ResponseTimeRootCauses'):
-                trace_data['ResponseTimeRootCauses'] = trace.get('ResponseTimeRootCauses', [])[
-                    :3
-                ]  # Limit to first 3
+                trace_data['ResponseTimeRootCauses'] = trace.get('ResponseTimeRootCauses', [])[:3]
 
             # Include limited annotations for key operations
             annotations = trace.get('Annotations', {})
@@ -447,15 +461,50 @@ async def query_sampled_traces(
             # Convert any datetime objects to ISO format strings
             for key, value in trace_data.items():
                 trace_data[key] = convert_datetime(value)
+
             trace_summaries.append(trace_data)
 
+        # Deduplicate trace summaries by fault message
+        seen_faults = {}
+        deduped_trace_summaries = []
+
+        for trace_summary in trace_summaries:
+            # Check if this trace has an error
+            has_issues = (
+                trace_summary.get('HasError')
+                or trace_summary.get('HasFault')
+                or trace_summary.get('HasThrottle')
+            )
+
+            if not has_issues:
+                # Always include healthy traces
+                deduped_trace_summaries.append(trace_summary)
+                continue
+
+            # Extract fault message for deduplication (only checks FaultRootCauses)
+            fault_msg = get_fault_message(trace_summary)
+
+            if fault_msg and fault_msg in seen_faults:
+                # Skip this trace - we already have one with the same fault message
+                seen_faults[fault_msg]['count'] += 1
+                logger.debug(
+                    f'Skipping duplicate trace {trace_summary.get("Id")} - fault message already seen: {fault_msg[:100]}...'
+                )
+                continue
+            else:
+                # First time seeing this fault (or no fault message) - include it
+                deduped_trace_summaries.append(trace_summary)
+                if fault_msg:
+                    seen_faults[fault_msg] = {'count': 1}
+
         # Check transaction search status
         is_tx_search_enabled, tx_destination, tx_status = check_transaction_search_enabled(region)
 
+        # Build response with original format but deduplicated traces
         result_data = {
-            'TraceSummaries': trace_summaries,
-            'TraceCount': len(trace_summaries),
-            'Message': f'Retrieved {len(trace_summaries)} traces (limited to prevent size issues)',
+            'TraceSummaries': deduped_trace_summaries,
+            'TraceCount': len(deduped_trace_summaries),
+            'Message': f'Retrieved {len(deduped_trace_summaries)} unique traces from {len(trace_summaries)} total (deduplicated by fault message)',
             'SamplingNote': "⚠️ This data is from X-Ray's 5% sampling. Results may not show all errors or issues.",
             'TransactionSearchStatus': {
                 'enabled': is_tx_search_enabled,
@@ -467,9 +516,18 @@ async def query_sampled_traces(
             },
         }
 
+        # Add dedup stats if we actually deduped anything
+        if len(deduped_trace_summaries) < len(trace_summaries):
+            duplicates_removed = len(trace_summaries) - len(deduped_trace_summaries)
+            result_data['DeduplicationStats'] = {
+                'OriginalTraceCount': len(trace_summaries),
+                'DuplicatesRemoved': duplicates_removed,
+                'UniqueFaultMessages': len(seen_faults),
+            }
+
         elapsed_time = timer() - start_time_perf
         logger.info(
-            f'query_sampled_traces completed in {elapsed_time:.3f}s - retrieved {len(trace_summaries)} traces'
+            f'query_sampled_traces completed in {elapsed_time:.3f}s - retrieved {len(deduped_trace_summaries)} unique traces from {len(trace_summaries)} total'
         )
         return json.dumps(result_data, indent=2)
 

{awslabs_cloudwatch_appsignals_mcp_server-0.1.9 → awslabs_cloudwatch_appsignals_mcp_server-0.1.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "awslabs.cloudwatch-appsignals-mcp-server"
-version = "0.1.9"
+version = "0.1.10"
 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.9 → awslabs_cloudwatch_appsignals_mcp_server-0.1.10}/tests/test_server.py

@@ -1634,7 +1634,7 @@ async def test_query_sampled_traces_with_fault_causes(mock_aws_clients):
         'Duration': 100,
         'HasFault': True,
         'FaultRootCauses': [
-            {'Services': [{'Name': 'service1'}]},
+            {'Services': [{'Name': 'service1', 'Exceptions': [{'Message': 'Test fault error'}]}]},
             {'Services': [{'Name': 'service2'}]},
             {'Services': [{'Name': 'service3'}]},
             {'Services': [{'Name': 'service4'}]},  # Should be limited to 3
@@ -1709,6 +1709,157 @@ async def test_query_sampled_traces_datetime_conversion(mock_aws_clients):
         assert 'EndTime' not in trace_summary
 
 
+@pytest.mark.asyncio
+async def test_query_sampled_traces_deduplication(mock_aws_clients):
+    """Test query_sampled_traces deduplicates traces with same fault message.
+
+    Note: Only FaultRootCauses are deduplicated, not ErrorRootCauses.
+    This is because the primary use case is investigating server faults (5xx errors),
+    not client errors (4xx).
+    """
+    # Create 5 traces with the same fault message
+    mock_traces = [
+        {
+            'Id': f'trace{i}',
+            'Duration': 100 + i * 10,
+            'ResponseTime': 95 + i * 10,
+            'HasFault': True,
+            'FaultRootCauses': [
+                {
+                    'Services': [
+                        {
+                            'Name': 'test-service',
+                            'Exceptions': [{'Message': 'Database connection timeout'}],
+                        }
+                    ]
+                }
+            ],
+        }
+        for i in range(1, 6)
+    ]
+
+    # Add 2 traces with ErrorRootCauses (these should NOT be deduplicated)
+    mock_traces.extend(
+        [
+            {
+                'Id': 'trace6',
+                'Duration': 200,
+                'HasError': True,
+                'ErrorRootCauses': [
+                    {
+                        'Services': [
+                            {
+                                'Name': 'api-service',
+                                'Exceptions': [{'Message': 'Invalid API key'}],
+                            }
+                        ]
+                    }
+                ],
+            },
+            {
+                'Id': 'trace7',
+                'Duration': 210,
+                'HasError': True,
+                'ErrorRootCauses': [
+                    {
+                        'Services': [
+                            {
+                                'Name': 'api-service',
+                                'Exceptions': [{'Message': 'Invalid API key'}],
+                            }
+                        ]
+                    }
+                ],
+            },
+        ]
+    )
+
+    # Add 2 healthy traces
+    mock_traces.extend(
+        [
+            {
+                'Id': 'trace8',
+                'Duration': 50,
+                'ResponseTime': 45,
+                'HasError': False,
+                'HasFault': False,
+            },
+            {
+                'Id': 'trace9',
+                'Duration': 55,
+                'ResponseTime': 50,
+                'HasError': False,
+                'HasFault': False,
+            },
+        ]
+    )
+
+    with patch(
+        'awslabs.cloudwatch_appsignals_mcp_server.trace_tools.get_trace_summaries_paginated'
+    ) as mock_paginated:
+        mock_paginated.return_value = mock_traces
+
+        result_json = await query_sampled_traces(
+            start_time='2024-01-01T00:00:00Z', end_time='2024-01-01T01:00:00Z'
+        )
+
+        result = json.loads(result_json)
+
+        # Verify deduplication worked - should only have 5 traces
+        # 1 for database timeout fault (deduplicated from 5)
+        # 2 for API key errors (NOT deduplicated - only faults are deduped)
+        # 2 healthy traces (not deduplicated)
+        assert result['TraceCount'] == 5
+        assert len(result['TraceSummaries']) == 5
+
+        # Verify deduplication stats
+        assert 'DeduplicationStats' in result
+        assert result['DeduplicationStats']['OriginalTraceCount'] == 9
+        assert result['DeduplicationStats']['DuplicatesRemoved'] == 4  # 9 - 5 = 4
+        assert (
+            result['DeduplicationStats']['UniqueFaultMessages'] == 1
+        )  # Only counting FaultRootCauses
+
+        # Find the trace with fault
+        db_trace = next(
+            (
+                t
+                for t in result['TraceSummaries']
+                if t.get('FaultRootCauses')
+                and any(
+                    'Database connection timeout' in str(s.get('Exceptions', []))
+                    for cause in t['FaultRootCauses']
+                    for s in cause.get('Services', [])
+                )
+            ),
+            None,
+        )
+        assert db_trace is not None
+        assert db_trace['HasFault'] is True
+
+        # Verify both error traces are present (not deduplicated)
+        error_traces = [
+            t
+            for t in result['TraceSummaries']
+            if t.get('ErrorRootCauses')
+            and any(
+                'Invalid API key' in str(s.get('Exceptions', []))
+                for cause in t['ErrorRootCauses']
+                for s in cause.get('Services', [])
+            )
+        ]
+        assert len(error_traces) == 2  # Both error traces should be kept
+        assert all(t['HasError'] is True for t in error_traces)
+
+        # Verify healthy traces are included
+        healthy_count = sum(
+            1
+            for t in result['TraceSummaries']
+            if not t.get('HasError') and not t.get('HasFault') and not t.get('HasThrottle')
+        )
+        assert healthy_count == 2
+
+
 def test_main_success(mock_aws_clients):
     """Test main function normal execution."""
     with patch('awslabs.cloudwatch_appsignals_mcp_server.server.mcp') as mock_mcp:

{awslabs_cloudwatch_appsignals_mcp_server-0.1.9 → awslabs_cloudwatch_appsignals_mcp_server-0.1.10}/uv.lock

@@ -46,7 +46,7 @@ wheels = [
 
 [[package]]
 name = "awslabs-cloudwatch-appsignals-mcp-server"
-version = "0.1.9"
+version = "0.1.10"
 source = { editable = "." }
 dependencies = [
     { name = "boto3" },