awslabs.cost-explorer-mcp-server 0.0.1__py3-none-any.whl

awslabs/__init__.py

@@ -0,0 +1,3 @@
+"""
+AWS Labs Cost Explorer MCP Server package.
+"""

awslabs/cost_explorer_mcp_server/__init__.py

@@ -0,0 +1,6 @@
+"""
+Cost Explorer MCP Server module.
+
+This module provides MCP tools for analyzing AWS costs and usage data through the AWS Cost Explorer API.
+"""
+__version__ = '0.0.0'

awslabs/cost_explorer_mcp_server/helpers.py

@@ -0,0 +1,221 @@
+"""Helper functions for the Cost Explorer MCP server."""
+
+import logging
+import re
+from typing import Dict, Any, List, Tuple
+import boto3
+from datetime import datetime
+
+# Set up logging
+logger = logging.getLogger(__name__)
+
+# Initialize AWS Cost Explorer client
+ce = boto3.client('ce')
+
+def validate_date_format(date_str: str) -> Tuple[bool, str]:
+    """
+    Validate that a date string is in YYYY-MM-DD format and is a valid date.
+    
+    Args:
+        date_str: The date string to validate
+        
+    Returns:
+        Tuple of (is_valid, error_message)
+    """
+    # Check format with regex
+    if not re.match(r'^\d{4}-\d{2}-\d{2}$', date_str):
+        return False, f"Date '{date_str}' is not in YYYY-MM-DD format"
+    
+    # Check if it's a valid date
+    try:
+        datetime.strptime(date_str, '%Y-%m-%d')
+        return True, ""
+    except ValueError as e:
+        return False, f"Invalid date '{date_str}': {str(e)}"
+
+def get_dimension_values(key: str, billing_period_start: str, billing_period_end: str) -> Dict[str, Any]:
+    """Get available values for a specific dimension."""
+    # Validate date formats
+    is_valid_start, error_start = validate_date_format(billing_period_start)
+    if not is_valid_start:
+        return {'error': error_start}
+    
+    is_valid_end, error_end = validate_date_format(billing_period_end)
+    if not is_valid_end:
+        return {'error': error_end}
+    
+    # Validate date range
+    if billing_period_start > billing_period_end:
+        return {'error': f"Start date '{billing_period_start}' cannot be after end date '{billing_period_end}'"}
+    
+    try:
+        response = ce.get_dimension_values(
+            TimePeriod={
+                'Start': billing_period_start,
+                'End': billing_period_end
+            },
+            Dimension=key.upper()
+        )
+        dimension_values = response['DimensionValues']
+        values = [value['Value'] for value in dimension_values]
+        return {'dimension': key.upper(), 'values': values}
+    except Exception as e:
+        logger.error(f"Error getting dimension values: {e}")
+        return {'error': str(e)}
+
+
+def get_tag_values(tag_key: str, billing_period_start: str, billing_period_end: str) -> Dict[str, Any]:
+    """Get available values for a specific tag key."""
+    # Validate date formats
+    is_valid_start, error_start = validate_date_format(billing_period_start)
+    if not is_valid_start:
+        return {'error': error_start}
+    
+    is_valid_end, error_end = validate_date_format(billing_period_end)
+    if not is_valid_end:
+        return {'error': error_end}
+    
+    # Validate date range
+    if billing_period_start > billing_period_end:
+        return {'error': f"Start date '{billing_period_start}' cannot be after end date '{billing_period_end}'"}
+    
+    try:
+        response = ce.get_tags(
+            TimePeriod={'Start': billing_period_start,
+                        'End': billing_period_end},
+            TagKey=tag_key
+        )
+        tag_values = response['Tags']
+        return {'tag_key': tag_key, 'values': tag_values}
+    except Exception as e:
+        logger.error(f"Error getting tag values: {e}")
+        return {'error': str(e)}
+
+
+
+def validate_expression(expression: Dict[str, Any], billing_period_start: str, billing_period_end: str) -> Dict[str, Any]:
+    """
+    Recursively validate the filter expression.
+    
+    Args:
+        expression: The filter expression to validate
+        billing_period_start: Start date of the billing period
+        billing_period_end: End date of the billing period
+        
+    Returns:
+        Empty dictionary if valid, or an error dictionary
+    """
+    # Validate date formats
+    is_valid_start, error_start = validate_date_format(billing_period_start)
+    if not is_valid_start:
+        return {'error': error_start}
+    
+    is_valid_end, error_end = validate_date_format(billing_period_end)
+    if not is_valid_end:
+        return {'error': error_end}
+    
+    # Validate date range
+    if billing_period_start > billing_period_end:
+        return {'error': f"Start date '{billing_period_start}' cannot be after end date '{billing_period_end}'"}
+    
+    try:
+        if 'Dimensions' in expression:
+            dimension = expression['Dimensions']
+            if 'Key' not in dimension or 'Values' not in dimension or 'MatchOptions' not in dimension:
+                return {'error': 'Dimensions filter must include "Key", "Values", and "MatchOptions".'}
+
+            dimension_key = dimension['Key']
+            dimension_values = dimension['Values']
+            valid_values_response = get_dimension_values(
+                dimension_key, billing_period_start, billing_period_end)
+            if 'error' in valid_values_response:
+                return {'error': valid_values_response['error']}
+            valid_values = valid_values_response['values']
+            for value in dimension_values:
+                if value not in valid_values:
+                    return {'error': f"Invalid value '{value}' for dimension '{dimension_key}'. Valid values are: {valid_values}"}
+
+        if 'Tags' in expression:
+            tag = expression['Tags']
+            if 'Key' not in tag or 'Values' not in tag or 'MatchOptions' not in tag:
+                return {'error': 'Tags filter must include "Key", "Values", and "MatchOptions".'}
+
+            tag_key = tag['Key']
+            tag_values = tag['Values']
+            valid_tag_values_response = get_tag_values(
+                tag_key, billing_period_start, billing_period_end)
+            if 'error' in valid_tag_values_response:
+                return {'error': valid_tag_values_response['error']}
+            valid_tag_values = valid_tag_values_response['values']
+            for value in tag_values:
+                if value not in valid_tag_values:
+                    return {'error': f"Invalid value '{value}' for tag '{tag_key}'. Valid values are: {valid_tag_values}"}
+
+        if 'CostCategories' in expression:
+            cost_category = expression['CostCategories']
+            if 'Key' not in cost_category or 'Values' not in cost_category or 'MatchOptions' not in cost_category:
+                return {'error': 'CostCategories filter must include "Key", "Values", and "MatchOptions".'}
+
+        logical_operators = ['And', 'Or', 'Not']
+        logical_count = sum(1 for op in logical_operators if op in expression)
+
+        if logical_count > 1:
+            return {'error': 'Only one logical operator (And, Or, Not) is allowed per expression in filter parameter.'}
+
+        if logical_count == 0 and len(expression) > 1:
+            return {'error': 'Filter parameter with multiple expressions require a logical operator (And, Or, Not).'}
+
+        if 'And' in expression:
+            if not isinstance(expression['And'], list):
+                return {'error': 'And expression must be a list of expressions.'}
+            for sub_expression in expression['And']:
+                result = validate_expression(
+                    sub_expression, billing_period_start, billing_period_end)
+                if 'error' in result:
+                    return result
+
+        if 'Or' in expression:
+            if not isinstance(expression['Or'], list):
+                return {'error': 'Or expression must be a list of expressions.'}
+            for sub_expression in expression['Or']:
+                result = validate_expression(
+                    sub_expression, billing_period_start, billing_period_end)
+                if 'error' in result:
+                    return result
+
+        if 'Not' in expression:
+            if not isinstance(expression['Not'], dict):
+                return {'error': 'Not expression must be a single expression.'}
+            result = validate_expression(
+                expression['Not'], billing_period_start, billing_period_end)
+            if 'error' in result:
+                return result
+
+        if not any(k in expression for k in ['Dimensions', 'Tags', 'CostCategories', 'And', 'Or', 'Not']):
+            return {'error': 'Filter Expression must include at least one of the following keys: "Dimensions", "Tags", "CostCategories", "And", "Or", "Not".'}
+
+        return {}
+    except Exception as e:
+        return {'error': f'Error validating expression: {str(e)}'}
+
+
+def validate_group_by(group_by: Dict[str, Any]) -> Dict[str, Any]:
+    """
+    Validate the group_by parameter.
+    
+    Args:
+        group_by: The group_by dictionary to validate
+        
+    Returns:
+        Empty dictionary if valid, or an error dictionary
+    """
+    try:
+        if not isinstance(group_by, dict) or 'Type' not in group_by or 'Key' not in group_by:
+            return {'error': 'group_by must be a dictionary with "Type" and "Key" keys.'}
+        
+        if group_by['Type'].upper() not in ['DIMENSION', 'TAG', 'COST_CATEGORY']:
+            return {'error': 'Invalid group Type. Valid types are DIMENSION, TAG, and COST_CATEGORY.'}
+        
+        return {}
+    except Exception as e:
+        return {'error': f'Error validating group_by: {str(e)}'}

awslabs/cost_explorer_mcp_server/server.py

@@ -0,0 +1,408 @@
+"""Cost Explorer MCP server implementation.
+
+This server provides tools for analyzing AWS costs and usage data through the AWS Cost Explorer API.
+"""
+
+import boto3
+import logging
+from datetime import datetime, timedelta
+import pandas as pd
+from mcp.server.fastmcp import Context, FastMCP
+from pydantic import BaseModel, Field, field_validator
+from typing import Any, Dict, Optional, Union
+
+from awslabs.cost_explorer_mcp_server.helpers import (
+    get_dimension_values,
+    get_tag_values,
+    validate_expression,
+    validate_group_by,
+    validate_date_format
+)
+
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Initialize AWS Cost Explorer client
+ce = boto3.client('ce')
+
+
+class DateRange(BaseModel):
+    """Date range model for cost queries."""
+    start_date: str = Field(
+        ...,
+        description="The start date of the billing period in YYYY-MM-DD format. Defaults to last month, if not provided."
+    )
+    end_date: str = Field(
+        ...,
+        description="The end date of the billing period in YYYY-MM-DD format."
+    )
+    
+    @field_validator('start_date')
+    @classmethod
+    def validate_start_date(cls, v):
+        is_valid, error = validate_date_format(v)
+        if not is_valid:
+            raise ValueError(error)
+        return v
+    
+    @field_validator('end_date')
+    @classmethod
+    def validate_end_date(cls, v, info):
+        is_valid, error = validate_date_format(v)
+        if not is_valid:
+            raise ValueError(error)
+        
+        # Access the start_date from the data dictionary
+        start_date = info.data.get('start_date')
+        if start_date and v < start_date:
+            raise ValueError(f"End date '{v}' cannot be before start date '{start_date}'")
+        
+        return v
+
+
+class GroupBy(BaseModel):
+    """Group by model for cost queries."""
+    type: str = Field(
+        ...,
+        description="Type of grouping. Valid values are DIMENSION, TAG, and COST_CATEGORY."
+    )
+    key: str = Field(
+        ...,
+        description="Key to group by. For DIMENSION type, valid values include AZ, INSTANCE_TYPE, LEGAL_ENTITY_NAME, INVOICING_ENTITY, LINKED_ACCOUNT, OPERATION, PLATFORM, PURCHASE_TYPE, SERVICE, TENANCY, RECORD_TYPE, and USAGE_TYPE."
+    )
+
+
+class FilterExpression(BaseModel):
+    """Filter expression model for cost queries."""
+    filter_json: str = Field(
+        ...,
+        description="Filter criteria as a Python dictionary to narrow down AWS costs. Supports filtering by Dimensions (SERVICE, REGION, etc.), Tags, or CostCategories. You can use logical operators (And, Or, Not) for complex filters. Examples: 1) Simple service filter: {'Dimensions': {'Key': 'SERVICE', 'Values': ['Amazon Elastic Compute Cloud - Compute', 'Amazon Simple Storage Service'], 'MatchOptions': ['EQUALS']}}. 2) Region filter: {'Dimensions': {'Key': 'REGION', 'Values': ['us-east-1'], 'MatchOptions': ['EQUALS']}}. 3) Combined filter: {'And': [{'Dimensions': {'Key': 'SERVICE', 'Values': ['Amazon Elastic Compute Cloud - Compute'], 'MatchOptions': ['EQUALS']}}, {'Dimensions': {'Key': 'REGION', 'Values': ['us-east-1'], 'MatchOptions': ['EQUALS']}}]}."
+    )
+
+
+class CostMetric(BaseModel):
+    """Cost metric model."""
+    metric: str = Field(
+        "UnblendedCost",
+        description="The metric to return in the query. Valid values are AmortizedCost, BlendedCost, NetAmortizedCost, NetUnblendedCost, NormalizedUsageAmount, UnblendedCost, and UsageQuantity. Note: For UsageQuantity, the service aggregates usage numbers without considering units. To get meaningful UsageQuantity metrics, filter by UsageType or UsageTypeGroups."
+    )
+
+
+class DimensionKey(BaseModel):
+    """Dimension key model."""
+    dimension_key: str = Field(
+        ...,
+        description="The name of the dimension to retrieve values for. Valid values are AZ, INSTANCE_TYPE, LINKED_ACCOUNT, OPERATION, PURCHASE_TYPE, SERVICE, USAGE_TYPE, PLATFORM, TENANCY, RECORD_TYPE, LEGAL_ENTITY_NAME, INVOICING_ENTITY, DEPLOYMENT_OPTION, DATABASE_ENGINE, CACHE_ENGINE, INSTANCE_TYPE_FAMILY, REGION, BILLING_ENTITY, RESERVATION_ID, SAVINGS_PLANS_TYPE, SAVINGS_PLAN_ARN, OPERATING_SYSTEM."
+    )
+
+
+# Create FastMCP server
+app = FastMCP(title="Cost Explorer MCP Server")
+
+
+@app.tool("get_today_date")
+async def get_today_date(ctx: Context) -> Dict[str, str]:
+    """Retrieve current date information.
+
+    This tool retrieves the current date in YYYY-MM-DD format and the current month in YYYY-MM format.
+    It's useful for comparing if the billing period requested by the user is not in the future.
+
+    Returns:
+        Dictionary containing today's date and current month
+    """
+    return {
+        'today_date': datetime.now().strftime('%Y-%m-%d'),
+        'current_month': datetime.now().strftime('%Y-%m')
+    }
+
+
+@app.tool("get_dimension_values")
+async def get_dimension_values_tool(
+    ctx: Context,
+    date_range: DateRange,
+    dimension: DimensionKey
+) -> Dict[str, Any]:
+    """Retrieve available dimension values for AWS Cost Explorer.
+
+    This tool retrieves all available and valid values for a specified dimension (e.g., SERVICE, REGION)
+    over a period of time. This is useful for validating filter values or exploring available options
+    for cost analysis.
+
+    Args:
+        date_range: The billing period start and end dates in YYYY-MM-DD format
+        dimension: The dimension key to retrieve values for (e.g., SERVICE, REGION, LINKED_ACCOUNT)
+
+    Returns:
+        Dictionary containing the dimension name and list of available values
+    """
+    try:
+        response = get_dimension_values(
+            dimension.dimension_key,
+            date_range.start_date,
+            date_range.end_date
+        )
+        return response
+    except Exception as e:
+        logger.error(f"Error getting dimension values: {e}")
+        return {'error': f'Error getting dimension values: {str(e)}'}
+
+
+@app.tool("get_tag_values")
+async def get_tag_values_tool(
+    ctx: Context,
+    date_range: DateRange,
+    tag_key: str = Field(..., description="The tag key to retrieve values for")
+) -> Dict[str, Any]:
+    """Retrieve available tag values for AWS Cost Explorer.
+
+    This tool retrieves all available values for a specified tag key over a period of time.
+    This is useful for validating tag filter values or exploring available tag options for cost analysis.
+
+    Args:
+        date_range: The billing period start and end dates in YYYY-MM-DD format
+        tag_key: The tag key to retrieve values for
+
+    Returns:
+        Dictionary containing the tag key and list of available values
+    """
+    try:
+        response = get_tag_values(
+            tag_key,
+            date_range.start_date,
+            date_range.end_date
+        )
+        return response
+    except Exception as e:
+        logger.error(f"Error getting tag values: {e}")
+        return {'error': f'Error getting tag values: {str(e)}'}
+
+
+@app.tool("get_cost_and_usage")
+async def get_cost_and_usage(
+    ctx: Context,
+    date_range: DateRange,
+    granularity: str = Field(
+        "MONTHLY",
+        description="The granularity at which cost data is aggregated. Valid values are DAILY, MONTHLY, and HOURLY. If not provided, defaults to MONTHLY."
+    ),
+    group_by: Optional[Union[Dict[str, str], str]] = Field(
+        None,
+        description="Either a dictionary with Type and Key for grouping costs, or simply a string key to group by (which will default to DIMENSION type). Example dictionary: {'Type': 'DIMENSION', 'Key': 'SERVICE'}. Example string: 'SERVICE'."
+    ),
+    filter_expression: Optional[Dict[str, Any]] = Field(
+        None,
+        description="Filter criteria as a Python dictionary to narrow down AWS costs. Supports filtering by Dimensions (SERVICE, REGION, etc.), Tags, or CostCategories. You can use logical operators (And, Or, Not) for complex filters. Examples: 1) Simple service filter: {'Dimensions': {'Key': 'SERVICE', 'Values': ['Amazon Elastic Compute Cloud - Compute', 'Amazon Simple Storage Service'], 'MatchOptions': ['EQUALS']}}. 2) Region filter: {'Dimensions': {'Key': 'REGION', 'Values': ['us-east-1'], 'MatchOptions': ['EQUALS']}}. 3) Combined filter: {'And': [{'Dimensions': {'Key': 'SERVICE', 'Values': ['Amazon Elastic Compute Cloud - Compute'], 'MatchOptions': ['EQUALS']}}, {'Dimensions': {'Key': 'REGION', 'Values': ['us-east-1'], 'MatchOptions': ['EQUALS']}}]}."
+    ),
+    metric: str = Field(
+        "UnblendedCost",
+        description="The metric to return in the query. Valid values are AmortizedCost, BlendedCost, NetAmortizedCost, NetUnblendedCost, NormalizedUsageAmount, UnblendedCost, and UsageQuantity."
+    )
+) -> Dict[str, Any]:
+    """Retrieve AWS cost and usage data.
+
+    This tool retrieves AWS cost and usage data for AWS services during a specified billing period,
+    with optional filtering and grouping. It dynamically generates cost reports tailored to specific needs
+    by specifying parameters such as granularity, billing period dates, and filter criteria.
+
+    Note: The end_date is treated as inclusive in this tool, meaning if you specify an end_date of 
+    "2025-01-31", the results will include data for January 31st. This differs from the AWS Cost Explorer 
+    API which treats end_date as exclusive.
+
+
+    Args:
+        date_range: The billing period start and end dates in YYYY-MM-DD format (end date is inclusive)
+        granularity: The granularity at which cost data is aggregated (DAILY, MONTHLY, HOURLY)
+        group_by: Either a dictionary with Type and Key, or simply a string key to group by
+        filter_expression: Filter criteria as a Python dictionary
+        metric: Cost metric to use (UnblendedCost, BlendedCost, etc.)
+
+    Returns:
+        Dictionary containing cost report data grouped according to the specified parameters
+    """
+
+    try:
+        # Process inputs
+        granularity = granularity.upper()
+        if granularity not in ["DAILY", "MONTHLY", "HOURLY"]:
+            return {'error': f"Invalid granularity: {granularity}. Valid values are DAILY, MONTHLY, and HOURLY."}
+            
+        billing_period_start = date_range.start_date
+        billing_period_end = date_range.end_date
+        
+        # Define valid metrics and their expected data structure
+        valid_metrics = {
+            "AmortizedCost": {"has_unit": True, "is_cost": True},
+            "BlendedCost": {"has_unit": True, "is_cost": True},
+            "NetAmortizedCost": {"has_unit": True, "is_cost": True},
+            "NetUnblendedCost": {"has_unit": True, "is_cost": True},
+            "NormalizedUsageAmount": {"has_unit": True, "is_cost": False},
+            "UnblendedCost": {"has_unit": True, "is_cost": True},
+            "UsageQuantity": {"has_unit": True, "is_cost": False}
+        }
+        
+        if metric not in valid_metrics:
+            return {'error': f"Invalid metric: {metric}. Valid values are {', '.join(valid_metrics.keys())}."}
+            
+        metric_config = valid_metrics[metric]
+
+        # Adjust end date for Cost Explorer API (exclusive)
+        # Add one day to make the end date inclusive for the user
+        billing_period_end_adj = (datetime.strptime(
+            billing_period_end, '%Y-%m-%d') + timedelta(days=1)).strftime('%Y-%m-%d')
+
+        # Process filter
+        filter_criteria = filter_expression
+
+        # Validate filter expression if provided
+        if filter_criteria:
+            # This validates both structure and values against AWS Cost Explorer
+            validation_result = validate_expression(
+                filter_criteria, billing_period_start, billing_period_end_adj)
+            if 'error' in validation_result:
+                return validation_result
+
+        # Process group_by
+        if not group_by:
+            group_by = {"Type": "DIMENSION", "Key": "SERVICE"}
+        elif isinstance(group_by, str):
+            group_by = {"Type": "DIMENSION", "Key": group_by}
+
+        # Validate group_by using the existing validate_group_by function
+        validation_result = validate_group_by(group_by)
+        if 'error' in validation_result:
+            return validation_result
+
+        # Prepare API call parameters
+        common_params = {
+            'TimePeriod': {
+                'Start': billing_period_start,
+                'End': billing_period_end_adj
+            },
+            'Granularity': granularity,
+            'GroupBy': [{'Type': group_by['Type'].upper(), 'Key': group_by['Key']}],
+            'Metrics': [metric]
+        }
+
+        if filter_criteria:
+            common_params['Filter'] = filter_criteria
+
+        # Get cost data
+        grouped_costs = {}
+        next_token = None
+        while True:
+            if next_token:
+                common_params['NextPageToken'] = next_token
+
+            try:
+                response = ce.get_cost_and_usage(**common_params)
+            except Exception as e:
+                logger.error(f"Error calling Cost Explorer API: {e}")
+                return {'error': f'AWS Cost Explorer API error: {str(e)}'}
+
+            for result_by_time in response['ResultsByTime']:
+                date = result_by_time['TimePeriod']['Start']
+                for group in result_by_time.get('Groups', []):
+                    if not group.get('Keys') or len(group['Keys']) == 0:
+                        logger.warning(f"Skipping group with no keys: {group}")
+                        continue
+                        
+                    group_key = group['Keys'][0]
+                    
+                    # Validate that the metric exists in the response
+                    if metric not in group.get('Metrics', {}):
+                        logger.error(f"Metric '{metric}' not found in response for group {group_key}")
+                        return {'error': f"Metric '{metric}' not found in response for group {group_key}"}
+                    
+                    metric_data = group['Metrics'][metric]
+                    
+                    # Validate metric data structure
+                    if 'Amount' not in metric_data:
+                        logger.error(f"Amount not found in metric data for {group_key}: {metric_data}")
+                        return {'error': f"Invalid response format: 'Amount' not found in metric data"}
+                    
+                    try:
+                        metric_data = group['Metrics'][metric]
+                        
+                        # Validate metric data structure
+                        if 'Amount' not in metric_data:
+                            logger.error(f"Amount not found in metric data for {group_key}: {metric_data}")
+                            return {'error': f"Invalid response format: 'Amount' not found in metric data"}
+                        
+                        # Process based on metric type
+                        if metric_config["is_cost"]:
+                            # Handle cost metrics
+                            cost = float(metric_data['Amount'])
+                            grouped_costs.setdefault(date, {}).update({group_key: cost})
+                        else:
+                            # Handle usage metrics (UsageQuantity, NormalizedUsageAmount)
+                            if 'Unit' not in metric_data and metric_config["has_unit"]:
+                                logger.warning(f"Unit not found in {metric} data for {group_key}, using 'Unknown' as unit")
+                                unit = 'Unknown'
+                            else:
+                                unit = metric_data.get('Unit', 'Count')
+                            amount = float(metric_data['Amount'])
+                            grouped_costs.setdefault(date, {}).update({group_key: (amount, unit)})
+                    except (ValueError, TypeError) as e:
+                        logger.error(f"Error processing metric data: {e}, data: {metric_data}")
+                        return {'error': f"Error processing metric data: {str(e)}"}
+
+            next_token = response.get('NextPageToken')
+            if not next_token:
+                break
+
+        # Process results
+        if not grouped_costs:
+            logger.info("No cost data found for the specified parameters")
+            return {'message': 'No cost data found for the specified parameters', 'GroupedCosts': {}}
+            
+        try:
+            if metric_config["is_cost"]:
+                # Process cost metrics
+                df = pd.DataFrame.from_dict(grouped_costs).round(2)
+                df['Service total'] = df.sum(axis=1).round(2)
+                df.loc['Total Costs'] = df.sum().round(2)
+                df = df.sort_values(by='Service total', ascending=False)
+            else:
+                # Process usage metrics (UsageQuantity, NormalizedUsageAmount)
+                usage_df = pd.DataFrame({(k, 'Amount'): {
+                                        k1: v1[0] for k1, v1 in v.items()} for k, v in grouped_costs.items()})
+                units_df = pd.DataFrame(
+                    {(k, 'Unit'): {k1: v1[1] for k1, v1 in v.items()} for k, v in grouped_costs.items()})
+                df = pd.concat([usage_df, units_df], axis=1)
+
+            result = {'GroupedCosts': df.to_dict()}
+        except Exception as e:
+            logger.error(f"Error processing cost data into DataFrame: {e}")
+            return {'error': f"Error processing cost data: {str(e)}", 'raw_data': grouped_costs}
+
+        result = {'GroupedCosts': df.to_dict()}
+
+        # Convert all keys to strings for JSON serialization
+        def stringify_keys(d):
+            if isinstance(d, dict):
+                return {str(k): stringify_keys(v) for k, v in d.items()}
+            elif isinstance(d, list):
+                return [stringify_keys(i) for i in d]
+            else:
+                return d
+
+        try:
+            result = stringify_keys(result)
+            return result
+        except Exception as e:
+            logger.error(f"Error serializing result: {e}")
+            return {'error': f"Error serializing result: {str(e)}"}
+
+    except Exception as e:
+        logger.error(f"Error generating cost report: {e}")
+        import traceback
+        logger.error(f"Traceback: {traceback.format_exc()}")
+        return {'error': f'Error generating cost report: {str(e)}'}
+
+def main():
+    """Run the MCP server with CLI argument support."""
+    app.run()
+
+if __name__ == "__main__":
+    main()

awslabs_cost_explorer_mcp_server-0.0.1.dist-info/METADATA

@@ -0,0 +1,187 @@
+Metadata-Version: 2.4
+Name: awslabs.cost-explorer-mcp-server
+Version: 0.0.1
+Summary: MCP server for analyzing AWS costs and usage data through the AWS Cost Explorer API
+Project-URL: Homepage, https://awslabs.github.io/mcp/
+Project-URL: Documentation, https://awslabs.github.io/mcp/servers/cost-explorer-mcp-server/
+Project-URL: Source, https://github.com/awslabs/mcp.git
+Project-URL: Bug Tracker, https://github.com/awslabs/mcp/issues
+Project-URL: Changelog, https://github.com/awslabs/mcp/blob/main/src/cost-explorer-mcp-server/CHANGELOG.md
+Author: Amazon Web Services
+Author-email: AWSLabs MCP <203918161+awslabs-mcp@users.noreply.github.com>
+License: Apache-2.0
+License-File: LICENSE
+License-File: NOTICE
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Python: >=3.10
+Requires-Dist: boto3>=1.36.20
+Requires-Dist: fastmcp>=0.1.0
+Requires-Dist: pandas>=2.2.3
+Requires-Dist: pydantic>=2.10.6
+Description-Content-Type: text/markdown
+
+# Cost Explorer MCP Server
+
+MCP server for analyzing AWS costs and usage data through the AWS Cost Explorer API.
+
+## Features
+
+### Analyze AWS costs and usage data
+
+- Get detailed breakdown of your AWS costs by service, region, and other dimensions
+- Understand how costs are distributed across various services
+- Query historical cost data for specific time periods
+- Filter costs by various dimensions, tags, and cost categories
+
+### Query cost data with natural language
+
+- Ask questions about your AWS costs in plain English
+- Get instant answers about your AWS spending patterns
+- Retrieve historical cost data with simple queries
+
+### Generate cost reports and insights
+
+- Generate comprehensive cost reports based on your AWS Cost Explorer data
+- Get cost breakdowns by various dimensions (service, region, account, etc.)
+- Analyze usage patterns and spending trends
+
+## Prerequisites
+
+1. Install `uv` from [Astral](https://docs.astral.sh/uv/getting-started/installation/) or the [GitHub README](https://github.com/astral-sh/uv#installation)
+2. Install Python using `uv python install 3.10`
+3. Set up AWS credentials with access to AWS Cost Explorer
+   - You need an AWS account with appropriate permissions
+   - Configure AWS credentials with `aws configure` or environment variables
+   - Ensure your IAM role/user has permissions to access AWS Cost Explorer API
+
+## Installation
+
+Here are some ways you can work with MCP across AWS, and we'll be adding support to more products including Amazon Q Developer CLI soon: (e.g. for Amazon Q Developer CLI MCP, `~/.aws/amazonq/mcp.json`):
+
+```json
+{
+  "mcpServers": {
+    "awslabs.cost-explorer-mcp-server": {
+      "command": "uvx",
+      "args": ["awslabs.cost-explorer-mcp-server@latest"],
+      "env": {
+        "FASTMCP_LOG_LEVEL": "ERROR",
+        "AWS_PROFILE": "your-aws-profile"
+      },
+      "disabled": false,
+      "autoApprove": []
+    }
+  }
+}
+```
+
+or docker after a successful `docker build -t awslabs/cost-explorer-mcp-server .`:
+
+```file
+# fictitious `.env` file with AWS temporary credentials
+AWS_ACCESS_KEY_ID=
+AWS_SECRET_ACCESS_KEY=
+AWS_SESSION_TOKEN=
+```
+
+```json
+{
+  "mcpServers": {
+    "awslabs.cost-explorer-mcp-server": {
+      "command": "docker",
+      "args": [
+        "run",
+        "--rm",
+        "--interactive",
+        "--env",
+        "FASTMCP_LOG_LEVEL=ERROR",
+        "--env-file",
+        "/full/path/to/file/above/.env",
+        "awslabs/cost-explorer-mcp-server:latest"
+      ],
+      "env": {},
+      "disabled": false,
+      "autoApprove": []
+    }
+  }
+}
+```
+
+NOTE: Your credentials will need to be kept refreshed from your host
+
+### AWS Authentication
+
+The MCP server uses the AWS profile specified in the `AWS_PROFILE` environment variable. If not provided, it defaults to the "default" profile in your AWS configuration file.
+
+```json
+"env": {
+  "AWS_PROFILE": "your-aws-profile"
+}
+```
+
+Make sure the AWS profile has permissions to access the AWS Cost Explorer API. The MCP server creates a boto3 session using the specified profile to authenticate with AWS services. Your AWS IAM credentials remain on your local machine and are strictly used for accessing AWS services.
+## Security Considerations
+
+### Required IAM Permissions
+The following IAM permissions are required for this MCP server:
+- ce:GetCostAndUsage
+- ce:GetDimensionValues
+- ce:GetTags
+
+Example IAM policy:
+json
+{
+   "Version": "2012-10-17",
+   "Statement": [
+       {
+           "Effect": "Allow",
+           "Action": [
+               "ce:GetCostAndUsage",
+               "ce:GetDimensionValues",
+               "ce:GetTags"
+           ],
+           "Resource": "*"
+       }
+   ]
+}
+
+## Available Tools
+
+The Cost Explorer MCP Server provides the following tools:
+
+1. `get_today_date` - Get the current date and month to determine relevent data when answering last month.
+2. `get_dimension_values` - Get available values for a specific dimension (e.g., SERVICE, REGION)
+3. `get_tag_values` - Get available values for a specific tag key
+4. `get_cost_and_usage` - Retrieve AWS cost and usage data with filtering and grouping options
+
+## Example Usage
+
+Here are some examples of how to use the Cost Explorer MCP Server:
+
+
+### Get dimension values
+
+```
+What AWS services did I use last month?
+```
+
+### Generate a cost report
+
+```
+Show me my AWS costs for the last 3 months grouped by service in us-east-1 region
+```
+
+```
+What were my EC2 costs excluding us-east-2 for January 2025?
+```
+
+## License
+
+This project is licensed under the Apache License 2.0 - see the LICENSE file for details.

awslabs_cost_explorer_mcp_server-0.0.1.dist-info/RECORD

@@ -0,0 +1,10 @@
+awslabs/__init__.py,sha256=4CPNxIm93Eam4qERUKxlLe0tSv0h8nd0ldA40H6lRtI,50
+awslabs/cost_explorer_mcp_server/__init__.py,sha256=hzMRqVmpxqoFbtkSW8Xl6RNRwS_sVKktrHyZM2o9KJ4,168
+awslabs/cost_explorer_mcp_server/helpers.py,sha256=Rt7oVmU3TEVOF8wq4CWb8yJFLjeoSlwKLwSrHb4YA6s,9054
+awslabs/cost_explorer_mcp_server/server.py,sha256=OztURecVW8JOtJZLu6WX97YYOK_GPH-CUgImodI25qE,18388
+awslabs_cost_explorer_mcp_server-0.0.1.dist-info/METADATA,sha256=SUGpDEksg1cJj60shYmdiHtkgVZXAF1NJzPxFPknqGY,5848
+awslabs_cost_explorer_mcp_server-0.0.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+awslabs_cost_explorer_mcp_server-0.0.1.dist-info/entry_points.txt,sha256=nkewGFi8GZCCtHhFofUmYii3OCeK_5qqgLXE4eUSFZg,98
+awslabs_cost_explorer_mcp_server-0.0.1.dist-info/licenses/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+awslabs_cost_explorer_mcp_server-0.0.1.dist-info/licenses/NOTICE,sha256=VL_gWrK0xFaHGFxxYj6BcZI30EkRxUH4Dv1u2Qsh3ao,92
+awslabs_cost_explorer_mcp_server-0.0.1.dist-info/RECORD,,

awslabs_cost_explorer_mcp_server-0.0.1.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.27.0
+Root-Is-Purelib: true
+Tag: py3-none-any

awslabs_cost_explorer_mcp_server-0.0.1.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+awslabs.cost-explorer-mcp-server = awslabs.cost_explorer_mcp_server.server:main

awslabs_cost_explorer_mcp_server-0.0.1.dist-info/licenses/LICENSE

@@ -0,0 +1,175 @@
+
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.

awslabs_cost_explorer_mcp_server-0.0.1.dist-info/licenses/NOTICE

@@ -0,0 +1,2 @@
+Cost Explorer MCP Server
+Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.