rootly-mcp-server 2.0.2__py3-none-any.whl → 2.0.3__py3-none-any.whl

rootly_mcp_server/rootly_openapi_loader.py

@@ -0,0 +1,96 @@
+#!/usr/bin/env python3
+"""
+Rootly OpenAPI Loader Utility
+
+Shared utility for loading Rootly's OpenAPI specification with smart fallback logic.
+"""
+
+import json
+import logging
+from pathlib import Path
+import httpx
+
+logger = logging.getLogger(__name__)
+
+
+def load_rootly_openapi_spec() -> dict:
+    """
+    Load Rootly OpenAPI spec with smart fallback logic.
+    
+    Loading priority:
+    1. Check current directory for rootly_openapi.json
+    2. Check parent directories for rootly_openapi.json
+    3. Check for swagger.json files  
+    4. Only as last resort, fetch from URL and cache locally
+    
+    Returns:
+        dict: The OpenAPI specification
+        
+    Raises:
+        RuntimeError: If the specification cannot be loaded
+    """
+    current_dir = Path.cwd()
+    
+    # Check for rootly_openapi.json in current directory and parents
+    for check_dir in [current_dir] + list(current_dir.parents):
+        spec_file = check_dir / "rootly_openapi.json"
+        if spec_file.is_file():
+            logger.info(f"Found OpenAPI spec at {spec_file}")
+            try:
+                with open(spec_file, "r") as f:
+                    return json.load(f)
+            except Exception as e:
+                logger.warning(f"Failed to load {spec_file}: {e}")
+                continue
+    
+    # Check for swagger.json in current directory and parents
+    for check_dir in [current_dir] + list(current_dir.parents):
+        spec_file = check_dir / "swagger.json"
+        if spec_file.is_file():
+            logger.info(f"Found Swagger spec at {spec_file}")
+            try:
+                with open(spec_file, "r") as f:
+                    return json.load(f)
+            except Exception as e:
+                logger.warning(f"Failed to load {spec_file}: {e}")
+                continue
+    
+    # Last resort: fetch from URL and cache
+    logger.warning("OpenAPI spec not found locally, fetching from URL (this should only happen once)")
+    return _fetch_and_cache_spec()
+
+
+def _fetch_and_cache_spec() -> dict:
+    """
+    Fetch OpenAPI spec from Rootly's URL and cache it locally.
+    
+    Returns:
+        dict: The OpenAPI specification
+        
+    Raises:
+        RuntimeError: If the specification cannot be fetched
+    """
+    SWAGGER_URL = "https://rootly-heroku.s3.amazonaws.com/swagger/v1/swagger.json"
+    
+    try:
+        logger.info(f"Fetching OpenAPI spec from {SWAGGER_URL}")
+        response = httpx.get(SWAGGER_URL, timeout=30.0)
+        response.raise_for_status()
+        spec_data = response.json()
+        
+        # Cache the spec for next time
+        current_dir = Path.cwd()
+        cache_file = current_dir / "rootly_openapi.json"
+        
+        try:
+            with open(cache_file, "w") as f:
+                json.dump(spec_data, f, indent=2)
+            logger.info(f"Cached OpenAPI spec to {cache_file} for future use")
+        except Exception as e:
+            logger.warning(f"Failed to cache OpenAPI spec: {e}")
+        
+        return spec_data
+        
+    except Exception as e:
+        logger.error(f"Failed to fetch OpenAPI spec: {e}")
+        raise RuntimeError(f"Could not load OpenAPI specification: {e}") 

rootly_mcp_server/routemap_server.py

@@ -11,13 +11,11 @@ from fastmcp import FastMCP
 from fastmcp.server.openapi import RouteMap, MCPType
 import os
 import logging
-import sys
 from pathlib import Path
 from typing import Optional, List
 
 # Import the shared OpenAPI loader
-sys.path.append(str(Path(__file__).parent.parent.parent))
-from rootly_openapi_loader import load_rootly_openapi_spec
+from .rootly_openapi_loader import load_rootly_openapi_spec
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -26,7 +24,7 @@ logger = logging.getLogger(__name__)
 def create_rootly_mcp_server(
     swagger_path: Optional[str] = None,
     name: str = "Rootly API Server (RouteMap Filtered)",
-    allowed_paths: Optional[List[str]] = None,
+    custom_allowed_paths: Optional[List[str]] = None,
     hosted: bool = False,
     base_url: Optional[str] = None,
 ):
@@ -75,105 +73,73 @@ def create_rootly_mcp_server(
     fix_array_types(openapi_spec)
     logger.info("✅ Fixed OpenAPI spec compatibility issues")
     
-    logger.info("Creating FastMCP server with RouteMap filtering...")
+    logger.info("Creating FastMCP server with pre-filtered OpenAPI spec...")
     
-    # Define custom route maps for filtering specific endpoints
-    route_maps = [
-        # Core incident management
-        RouteMap(
-            pattern=r"^/v1/incidents$",
-            mcp_type=MCPType.TOOL
-        ),
-        RouteMap(
-            pattern=r"^/v1/incidents/\{incident_id\}/alerts$",
-            mcp_type=MCPType.TOOL
-        ),
-        RouteMap(
-            pattern=r"^/v1/incidents/\{incident_id\}/action_items$",
-            mcp_type=MCPType.TOOL
-        ),
-        
-        # Alert management
-        RouteMap(
-            pattern=r"^/v1/alerts$",
-            mcp_type=MCPType.TOOL
-        ),
-        RouteMap(
-            pattern=r"^/v1/alerts/\{id\}$",
-            mcp_type=MCPType.TOOL
-        ),
-        
-        # Configuration entities
-        RouteMap(
-            pattern=r"^/v1/severities(\{id\})?$",
-            mcp_type=MCPType.TOOL
-        ),
-        RouteMap(
-            pattern=r"^/v1/incident_types(\{id\})?$",
-            mcp_type=MCPType.TOOL
-        ),
-        RouteMap(
-            pattern=r"^/v1/functionalities(\{id\})?$",
-            mcp_type=MCPType.TOOL
-        ),
-        
-        # Organization
-        RouteMap(
-            pattern=r"^/v1/teams(\{id\})?$",
-            mcp_type=MCPType.TOOL
-        ),
-        RouteMap(
-            pattern=r"^/v1/users(\{id\}|/me)?$",
-            mcp_type=MCPType.TOOL
-        ),
-        
-        # Infrastructure
-        RouteMap(
-            pattern=r"^/v1/services(\{id\})?$",
-            mcp_type=MCPType.TOOL
-        ),
-        RouteMap(
-            pattern=r"^/v1/environments(\{id\})?$",
-            mcp_type=MCPType.TOOL
-        ),
-        
-        # Action items
-        RouteMap(
-            pattern=r"^/v1/incident_action_items(\{id\})?$",
-            mcp_type=MCPType.TOOL
-        ),
-        
-        # Workflows
-        RouteMap(
-            pattern=r"^/v1/workflows(\{id\})?$",
-            mcp_type=MCPType.TOOL
-        ),
-        RouteMap(
-            pattern=r"^/v1/workflow_runs(\{id\})?$",
-            mcp_type=MCPType.TOOL
-        ),
-        
-        # Status pages
-        RouteMap(
-            pattern=r"^/v1/status_pages(\{id\})?$",
-            mcp_type=MCPType.TOOL
-        ),
-        
-        # Exclude everything else
-        RouteMap(
-            pattern=r".*",
-            mcp_type=MCPType.EXCLUDE
-        )
-    ]
+    # Define the specific endpoints we want to include
+    if custom_allowed_paths:
+        allowed_paths = set(custom_allowed_paths)
+    else:
+        allowed_paths = {
+            # Core incident management
+            "/v1/incidents",
+            "/v1/incidents/{incident_id}/alerts", 
+            "/v1/incidents/{incident_id}/action_items",
+            
+            # Alert management  
+            "/v1/alerts",
+            "/v1/alerts/{id}",
+            
+            # Configuration entities
+            "/v1/severities",
+            "/v1/severities/{id}",
+            "/v1/incident_types", 
+            "/v1/incident_types/{id}",
+            "/v1/functionalities",
+            "/v1/functionalities/{id}",
+            
+            # Organization
+            "/v1/teams",
+            "/v1/teams/{id}",
+            "/v1/users",
+            "/v1/users/me", 
+            "/v1/users/{id}",
+            
+            # Infrastructure
+            "/v1/services",
+            "/v1/services/{id}",
+            "/v1/environments",
+            "/v1/environments/{id}",
+            
+            # Action items
+            "/v1/action_items",
+            "/v1/action_items/{id}",
+            
+            # Workflows
+            "/v1/workflows",
+            "/v1/workflows/{id}",
+            
+            # Status pages
+            "/v1/status-pages",
+            "/v1/status-pages/{id}"
+        }
+    
+    # Filter the OpenAPI spec to only include allowed paths
+    original_paths = openapi_spec.get("paths", {})
+    filtered_paths = {path: spec for path, spec in original_paths.items() if path in allowed_paths}
+    
+    logger.info(f"📊 Filtered OpenAPI spec from {len(original_paths)} paths to {len(filtered_paths)} paths")
+    logger.info(f"🔍 Allowed paths: {sorted(allowed_paths)}")
+    logger.info(f"✅ Filtered paths: {sorted(filtered_paths.keys())}")
+    
+    openapi_spec["paths"] = filtered_paths
     
-    # Create MCP server with custom route maps
+    # Create MCP server without route maps for now to test basic functionality  
     mcp = FastMCP.from_openapi(
         openapi_spec=openapi_spec,
         client=client,
         name=name,
         timeout=30.0,
-        tags={"rootly", "incident-management", "evaluation"},
-        route_maps=route_maps
+        tags={"rootly", "incident-management", "evaluation"}
     )
     
     logger.info(f"✅ Created MCP server with RouteMap filtering successfully")

{rootly_mcp_server-2.0.2.dist-info → rootly_mcp_server-2.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rootly-mcp-server
-Version: 2.0.2
+Version: 2.0.3
 Summary: A Model Context Protocol server for Rootly APIs using OpenAPI spec
 Project-URL: Homepage, https://github.com/Rootly-AI-Labs/Rootly-MCP-server
 Project-URL: Issues, https://github.com/Rootly-AI-Labs/Rootly-MCP-server/issues

{rootly_mcp_server-2.0.2.dist-info → rootly_mcp_server-2.0.3.dist-info}/RECORD

@@ -1,12 +1,13 @@
 rootly_mcp_server/__init__.py,sha256=H3Dd4Od-PEff8VdbMgnblS6cCEBCb9XqYdI_-wmtQqk,628
 rootly_mcp_server/__main__.py,sha256=yOBEaQWEMZvsC7zT5no7yo1ISIvfeq7NlIPflBMoAFI,6447
 rootly_mcp_server/client.py,sha256=05TsHVJ3WtLH0k4R19Yzwwx4xcmjCKH6hS3uKcTMwRA,4678
-rootly_mcp_server/routemap_server.py,sha256=SloHu4ZTaFBusoIlhM2h7jfB8kFtMQrT3M_n7JAekug,6101
+rootly_mcp_server/rootly_openapi_loader.py,sha256=GebIDHzEMV6XCuPox3SoBOG42lyaAb3RFlhWoCdIkeE,3159
+rootly_mcp_server/routemap_server.py,sha256=DDcuY68_B4uqg8aOuDmUp0zhrlpIxQtw0h5265ZYULk,5631
 rootly_mcp_server/server.py,sha256=2paE39caYRVuQG4yWEEOqZewbvpFREmf-J01lPQ5elw,19756
 rootly_mcp_server/test_client.py,sha256=8p1aJHrEt_Tj2NuJzTnTHw-ZeW816P99fJi5bhPidyc,5119
 rootly_mcp_server/data/__init__.py,sha256=fO8a0bQnRVEoRMHKvhFzj10bhoaw7VsI51czc2MsUm4,143
-rootly_mcp_server-2.0.2.dist-info/METADATA,sha256=gVQHYX9Z8KlzftzQZluRXsJ45K3kEkzxkxTpyHSdyCM,6141
-rootly_mcp_server-2.0.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-rootly_mcp_server-2.0.2.dist-info/entry_points.txt,sha256=NE33b8VgigVPGBkboyo6pvN1Vz35HZtLybxMO4Q03PI,70
-rootly_mcp_server-2.0.2.dist-info/licenses/LICENSE,sha256=c9w9ZZGl14r54tsP40oaq5adTVX_HMNHozPIH2ymzmw,11341
-rootly_mcp_server-2.0.2.dist-info/RECORD,,
+rootly_mcp_server-2.0.3.dist-info/METADATA,sha256=x79WLJxzQUZ_A9oYtW3qBsRz6V4n7Qb_sXRVaJa0e1g,6141
+rootly_mcp_server-2.0.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+rootly_mcp_server-2.0.3.dist-info/entry_points.txt,sha256=NE33b8VgigVPGBkboyo6pvN1Vz35HZtLybxMO4Q03PI,70
+rootly_mcp_server-2.0.3.dist-info/licenses/LICENSE,sha256=c9w9ZZGl14r54tsP40oaq5adTVX_HMNHozPIH2ymzmw,11341
+rootly_mcp_server-2.0.3.dist-info/RECORD,,