awslabs.prometheus-mcp-server 0.1.1__tar.gz → 0.2.1__tar.gz

{awslabs_prometheus_mcp_server-0.1.1 → awslabs_prometheus_mcp_server-0.2.1}/.gitignore

@@ -563,3 +563,10 @@ samples/mcp-integration-with-nova-canvas/output/*
 
 memory-bank/*
 .clinerules
+
+# OpenAPI MCP Server specific ignores
+src/openapi-mcp-server/generated-diagrams/*
+src/openapi-mcp-server/*.txt
+src/openapi-mcp-server/*-coverage.xml
+src/openapi-mcp-server/junit.xml
+executive-summary.md

awslabs_prometheus_mcp_server-0.2.1/CHANGELOG.md

@@ -0,0 +1,22 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
+and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
+
+## [0.2.0] - 2024-06-01
+
+### Changed
+
+- Refactored to use workspace-based configuration model
+- Added GetAvailableWorkspaces tool to list available Prometheus workspaces
+- Modified all tools to require workspace_id parameter
+- Removed requirement for Prometheus URL at startup
+- Added per-request workspace configuration
+
+## [0.1.1] - 2024-05-15
+
+### Added
+
+- Initial project setup

{awslabs_prometheus_mcp_server-0.1.1 → awslabs_prometheus_mcp_server-0.2.1}/Dockerfile

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 # dependabot should continue to update this to the latest hash.
-FROM public.ecr.aws/sam/build-python3.10@sha256:a40f492a0cd8d76557f8a187fc00e49e8864b3cea683e74718ce317790c1ce61 AS uv
+FROM public.ecr.aws/sam/build-python3.10@sha256:d821662474d65f3cf2fc97dba2fa807a3adb580d02895fc4545527812550ea65 AS uv
 
 # Install the project into `/app`
 WORKDIR /app
@@ -31,11 +31,11 @@ ENV UV_PYTHON_PREFERENCE=only-system
 ENV UV_FROZEN=true
 
 # Copy the required files first
-COPY pyproject.toml uv.lock ./
+COPY pyproject.toml uv.lock uv-requirements.txt ./
 
 # Install the project's dependencies using the lockfile and settings
 RUN --mount=type=cache,target=/root/.cache/uv \
-    pip install uv && \
+    pip install --require-hashes --requirement uv-requirements.txt && \
     uv sync --frozen --no-install-project --no-dev --no-editable
 
 # Then, add the rest of the project source code and install it
@@ -47,7 +47,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/sam/build-python3.10@sha256:a40f492a0cd8d76557f8a187fc00e49e8864b3cea683e74718ce317790c1ce61
+FROM public.ecr.aws/sam/build-python3.10@sha256:d821662474d65f3cf2fc97dba2fa807a3adb580d02895fc4545527812550ea65
 
 # Place executables in the environment at the front of the path and include other binaries
 ENV PATH="/app/.venv/bin:$PATH:/usr/sbin"

{awslabs_prometheus_mcp_server-0.1.1 → awslabs_prometheus_mcp_server-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: awslabs.prometheus-mcp-server
-Version: 0.1.1
+Version: 0.2.1
 Summary: MCP server for interacting with AWS Managed Prometheus
 Project-URL: homepage, https://awslabs.github.io/mcp/
 Project-URL: docs, https://awslabs.github.io/mcp/servers/prometheus-mcp-server/
@@ -47,6 +47,8 @@ This MCP server is designed to be fully compatible with Amazon Q developer CLI,
 
 ## Installation
 
+[![Install MCP Server](https://cursor.com/deeplink/mcp-install-light.svg)](https://cursor.com/install-mcp?name=awslabs.prometheus-mcp-server&config=eyJjb21tYW5kIjoidXZ4IGF3c2xhYnMucHJvbWV0aGV1cy1tY3Atc2VydmVyQGxhdGVzdCAtLXVybCBodHRwczovL2Fwcy13b3Jrc3BhY2VzLnVzLWVhc3QtMS5hbWF6b25hd3MuY29tL3dvcmtzcGFjZXMvd3MtPFdvcmtzcGFjZSBJRD4gLS1yZWdpb24gPFlvdXIgQVdTIFJlZ2lvbj4gLS1wcm9maWxlIDxZb3VyIENMSSBQcm9maWxlIFtkZWZhdWx0XSBpZiBubyBwcm9maWxlIGlzIHVzZWQ%2BIiwiZW52Ijp7IkZBU1RNQ1BfTE9HX0xFVkVMIjoiREVCVUciLCJBV1NfUFJPRklMRSI6IjxZb3VyIENMSSBQcm9maWxlIFtkZWZhdWx0XSBpZiBubyBwcm9maWxlIGlzIHVzZWQ%2BIn19)
+
 ### Prerequisites
 
 - Python 3.10 or higher
@@ -69,23 +71,41 @@ mkdir -p ~/.aws/amazonq/
 ```
 
 2. Add the following to `~/.aws/amazonq/mcp.json`:
+
+### Basic Configuration
+```json
+{
+  "mcpServers": {
+    "prometheus": {
+      "command": "uvx",
+      "args": [
+        "awslabs.prometheus-mcp-server@latest"
+      ],
+      "env": {
+        "FASTMCP_LOG_LEVEL": "DEBUG"
+      }
+    }
+  }
+}
+```
+
+### Configuration with Optional Arguments
 ```json
 {
   "mcpServers": {
-    "awslabs.prometheus-mcp-server": {
+    "prometheus": {
       "command": "uvx",
       "args": [
         "awslabs.prometheus-mcp-server@latest",
         "--url",
-        "https://aps-workspaces.us-east-1.amazonaws.com/workspaces/ws-<Workspace ID>",
+        "https://aps-workspaces.<AWS Region>.amazonaws.com/workspaces/ws-<Workspace ID>",
         "--region",
         "<Your AWS Region>",
         "--profile",
-        "<Your CLI Profile [default] if no profile is used>"
+        "<Your CLI Profile>"
       ],
       "env": {
-        "FASTMCP_LOG_LEVEL": "DEBUG",
-        "AWS_PROFILE": "<Your CLI Profile [default] if no profile is used>"
+        "FASTMCP_LOG_LEVEL": "DEBUG"
       }
     }
   }
@@ -96,30 +116,46 @@ mkdir -p ~/.aws/amazonq/
 
 ## Available Tools
 
-1. **execute_query**
+1. **GetAvailableWorkspaces**
+   - List all available Prometheus workspaces in the specified region
+   - Parameters: region (optional)
+   - Returns: List of workspaces with IDs, aliases, and status
+
+2. **ExecuteQuery**
    - Execute instant PromQL queries against Prometheus
-   - Parameters: query (required), time (optional)
+   - Parameters: workspace_id (required), query (required), time (optional), region (optional)
 
-2. **execute_range_query**
+3. **ExecuteRangeQuery**
    - Execute PromQL queries over a time range
-   - Parameters: query, start time, end time, step interval
+   - Parameters: workspace_id (required), query, start time, end time, step interval, region (optional)
 
-3. **list_metrics**
+4. **ListMetrics**
    - Retrieve all available metric names from Prometheus
+   - Parameters: workspace_id (required), region (optional)
    - Returns: Sorted list of metric names
 
-4. **get_server_info**
+5. **GetServerInfo**
    - Retrieve server configuration details
+   - Parameters: workspace_id (required), region (optional)
    - Returns: URL, region, profile, and service information
 
 ## Example Queries
 
 ```python
+# Get available workspaces
+workspaces = await get_available_workspaces()
+for ws in workspaces['workspaces']:
+    print(f"ID: {ws['workspace_id']}, Alias: {ws['alias']}, Status: {ws['status']}")
+
 # Execute an instant query
-result = await execute_query("up")
+result = await execute_query(
+    workspace_id="ws-12345678-abcd-1234-efgh-123456789012",
+    query="up"
+)
 
 # Execute a range query
 data = await execute_range_query(
+    workspace_id="ws-12345678-abcd-1234-efgh-123456789012",
     query="rate(node_cpu_seconds_total[5m])",
     start="2023-01-01T00:00:00Z",
     end="2023-01-01T01:00:00Z",
@@ -127,10 +163,14 @@ data = await execute_range_query(
 )
 
 # List available metrics
-metrics = await list_metrics()
+metrics = await list_metrics(
+    workspace_id="ws-12345678-abcd-1234-efgh-123456789012"
+)
 
 # Get server information
-info = await get_server_info()
+info = await get_server_info(
+    workspace_id="ws-12345678-abcd-1234-efgh-123456789012"
+)
 ```
 
 ## Troubleshooting

{awslabs_prometheus_mcp_server-0.1.1 → awslabs_prometheus_mcp_server-0.2.1}/README.md

@@ -15,6 +15,8 @@ This MCP server is designed to be fully compatible with Amazon Q developer CLI,
 
 ## Installation
 
+[![Install MCP Server](https://cursor.com/deeplink/mcp-install-light.svg)](https://cursor.com/install-mcp?name=awslabs.prometheus-mcp-server&config=eyJjb21tYW5kIjoidXZ4IGF3c2xhYnMucHJvbWV0aGV1cy1tY3Atc2VydmVyQGxhdGVzdCAtLXVybCBodHRwczovL2Fwcy13b3Jrc3BhY2VzLnVzLWVhc3QtMS5hbWF6b25hd3MuY29tL3dvcmtzcGFjZXMvd3MtPFdvcmtzcGFjZSBJRD4gLS1yZWdpb24gPFlvdXIgQVdTIFJlZ2lvbj4gLS1wcm9maWxlIDxZb3VyIENMSSBQcm9maWxlIFtkZWZhdWx0XSBpZiBubyBwcm9maWxlIGlzIHVzZWQ%2BIiwiZW52Ijp7IkZBU1RNQ1BfTE9HX0xFVkVMIjoiREVCVUciLCJBV1NfUFJPRklMRSI6IjxZb3VyIENMSSBQcm9maWxlIFtkZWZhdWx0XSBpZiBubyBwcm9maWxlIGlzIHVzZWQ%2BIn19)
+
 ### Prerequisites
 
 - Python 3.10 or higher
@@ -37,23 +39,41 @@ mkdir -p ~/.aws/amazonq/
 ```
 
 2. Add the following to `~/.aws/amazonq/mcp.json`:
+
+### Basic Configuration
+```json
+{
+  "mcpServers": {
+    "prometheus": {
+      "command": "uvx",
+      "args": [
+        "awslabs.prometheus-mcp-server@latest"
+      ],
+      "env": {
+        "FASTMCP_LOG_LEVEL": "DEBUG"
+      }
+    }
+  }
+}
+```
+
+### Configuration with Optional Arguments
 ```json
 {
   "mcpServers": {
-    "awslabs.prometheus-mcp-server": {
+    "prometheus": {
       "command": "uvx",
       "args": [
         "awslabs.prometheus-mcp-server@latest",
         "--url",
-        "https://aps-workspaces.us-east-1.amazonaws.com/workspaces/ws-<Workspace ID>",
+        "https://aps-workspaces.<AWS Region>.amazonaws.com/workspaces/ws-<Workspace ID>",
         "--region",
         "<Your AWS Region>",
         "--profile",
-        "<Your CLI Profile [default] if no profile is used>"
+        "<Your CLI Profile>"
       ],
       "env": {
-        "FASTMCP_LOG_LEVEL": "DEBUG",
-        "AWS_PROFILE": "<Your CLI Profile [default] if no profile is used>"
+        "FASTMCP_LOG_LEVEL": "DEBUG"
       }
     }
   }
@@ -64,30 +84,46 @@ mkdir -p ~/.aws/amazonq/
 
 ## Available Tools
 
-1. **execute_query**
+1. **GetAvailableWorkspaces**
+   - List all available Prometheus workspaces in the specified region
+   - Parameters: region (optional)
+   - Returns: List of workspaces with IDs, aliases, and status
+
+2. **ExecuteQuery**
    - Execute instant PromQL queries against Prometheus
-   - Parameters: query (required), time (optional)
+   - Parameters: workspace_id (required), query (required), time (optional), region (optional)
 
-2. **execute_range_query**
+3. **ExecuteRangeQuery**
    - Execute PromQL queries over a time range
-   - Parameters: query, start time, end time, step interval
+   - Parameters: workspace_id (required), query, start time, end time, step interval, region (optional)
 
-3. **list_metrics**
+4. **ListMetrics**
    - Retrieve all available metric names from Prometheus
+   - Parameters: workspace_id (required), region (optional)
    - Returns: Sorted list of metric names
 
-4. **get_server_info**
+5. **GetServerInfo**
    - Retrieve server configuration details
+   - Parameters: workspace_id (required), region (optional)
    - Returns: URL, region, profile, and service information
 
 ## Example Queries
 
 ```python
+# Get available workspaces
+workspaces = await get_available_workspaces()
+for ws in workspaces['workspaces']:
+    print(f"ID: {ws['workspace_id']}, Alias: {ws['alias']}, Status: {ws['status']}")
+
 # Execute an instant query
-result = await execute_query("up")
+result = await execute_query(
+    workspace_id="ws-12345678-abcd-1234-efgh-123456789012",
+    query="up"
+)
 
 # Execute a range query
 data = await execute_range_query(
+    workspace_id="ws-12345678-abcd-1234-efgh-123456789012",
     query="rate(node_cpu_seconds_total[5m])",
     start="2023-01-01T00:00:00Z",
     end="2023-01-01T01:00:00Z",
@@ -95,10 +131,14 @@ data = await execute_range_query(
 )
 
 # List available metrics
-metrics = await list_metrics()
+metrics = await list_metrics(
+    workspace_id="ws-12345678-abcd-1234-efgh-123456789012"
+)
 
 # Get server information
-info = await get_server_info()
+info = await get_server_info(
+    workspace_id="ws-12345678-abcd-1234-efgh-123456789012"
+)
 ```
 
 ## Troubleshooting

{awslabs_prometheus_mcp_server-0.1.1 → awslabs_prometheus_mcp_server-0.2.1}/awslabs/prometheus_mcp_server/__init__.py

@@ -14,4 +14,4 @@
 
 """awslabs Prometheus MCP Server."""
 
-__version__ = '0.1.0'
+__version__ = '0.2.0'

awslabs_prometheus_mcp_server-0.2.1/awslabs/prometheus_mcp_server/models.py

@@ -0,0 +1,50 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Data models for the Prometheus MCP server."""
+
+from pydantic import BaseModel, Field
+from typing import List, Optional
+
+
+class MetricsList(BaseModel):
+    """List of available metrics in Prometheus.
+
+    Attributes:
+        metrics: List of metric names available in the Prometheus server.
+    """
+
+    metrics: List[str] = Field(
+        description='List of metric names available in the Prometheus server'
+    )
+
+
+class ServerInfo(BaseModel):
+    """Information about the Prometheus server configuration.
+
+    Attributes:
+        prometheus_url: URL of the AWS Managed Prometheus endpoint.
+        aws_region: AWS region where the Prometheus service is located.
+        aws_profile: AWS profile name used for authentication.
+        service_name: AWS service name for SigV4 authentication.
+    """
+
+    prometheus_url: Optional[str] = Field(
+        None, description='URL of the AWS Managed Prometheus endpoint'
+    )
+    aws_region: str = Field(description='AWS region where the Prometheus service is located')
+    aws_profile: Optional[str] = Field(
+        None, description='AWS profile name used for authentication'
+    )
+    service_name: str = Field(description='AWS service name for SigV4 authentication')