rootly-mcp-server 2.0.12__tar.gz → 2.0.13__tar.gz

{rootly_mcp_server-2.0.12 → rootly_mcp_server-2.0.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rootly-mcp-server
-Version: 2.0.12
+Version: 2.0.13
 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.12 → rootly_mcp_server-2.0.13}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "rootly-mcp-server"
-version = "2.0.12"
+version = "2.0.13"
 description = "A Model Context Protocol server for Rootly APIs using OpenAPI spec"
 readme = "README.md"
 requires-python = ">=3.12"

{rootly_mcp_server-2.0.12 → rootly_mcp_server-2.0.13}/src/rootly_mcp_server/server.py

@@ -391,7 +391,7 @@ def create_rootly_mcp_server(
         # Single page mode
         if page_number > 0:
             params = {
-                "page[size]": min(page_size, 5),  # Keep responses very small to avoid errors
+                "page[size]": page_size,  # Use requested page size (already limited to max 20)
                 "page[number]": page_number,
                 "include": "",
             }
@@ -409,7 +409,7 @@ def create_rootly_mcp_server(
         # Multi-page mode (page_number = 0)
         all_incidents = []
         current_page = 1
-        effective_page_size = min(page_size, 5)  # Keep responses very small to avoid errors
+        effective_page_size = page_size  # Use requested page size (already limited to max 20)
         max_pages = 10  # Safety limit to prevent infinite loops
 
         try:
@@ -922,6 +922,73 @@ def _filter_openapi_spec(spec: Dict[str, Any], allowed_paths: List[str]) -> Dict
                                 "description": param.get("description", "Parameter value")
                             }
 
+            # Add/modify pagination limits to alerts and incident-related endpoints to prevent infinite loops
+            if method.lower() == "get" and ("alerts" in path.lower() or "incident" in path.lower()):
+                if "parameters" not in operation:
+                    operation["parameters"] = []
+                
+                # Find existing pagination parameters and update them with limits
+                page_size_param = None
+                page_number_param = None
+                
+                for param in operation["parameters"]:
+                    if param.get("name") == "page[size]":
+                        page_size_param = param
+                    elif param.get("name") == "page[number]":
+                        page_number_param = param
+                
+                # Update or add page[size] parameter with limits
+                if page_size_param:
+                    # Update existing parameter with limits
+                    if "schema" not in page_size_param:
+                        page_size_param["schema"] = {}
+                    page_size_param["schema"].update({
+                        "type": "integer",
+                        "default": 10,
+                        "minimum": 1,
+                        "maximum": 20,
+                        "description": "Number of results per page (max: 20)"
+                    })
+                else:
+                    # Add new parameter
+                    operation["parameters"].append({
+                        "name": "page[size]",
+                        "in": "query",
+                        "required": False,
+                        "schema": {
+                            "type": "integer",
+                            "default": 10,
+                            "minimum": 1,
+                            "maximum": 20,
+                            "description": "Number of results per page (max: 20)"
+                        }
+                    })
+                
+                # Update or add page[number] parameter with defaults
+                if page_number_param:
+                    # Update existing parameter 
+                    if "schema" not in page_number_param:
+                        page_number_param["schema"] = {}
+                    page_number_param["schema"].update({
+                        "type": "integer",
+                        "default": 1,
+                        "minimum": 1,
+                        "description": "Page number to retrieve"
+                    })
+                else:
+                    # Add new parameter
+                    operation["parameters"].append({
+                        "name": "page[number]",
+                        "in": "query", 
+                        "required": False,
+                        "schema": {
+                            "type": "integer",
+                            "default": 1,
+                            "minimum": 1,
+                            "description": "Page number to retrieve"
+                        }
+                    })
+
     # Also clean up any remaining broken references in components
     if "components" in filtered_spec and "schemas" in filtered_spec["components"]:
         schemas = filtered_spec["components"]["schemas"]

{rootly_mcp_server-2.0.12 → rootly_mcp_server-2.0.13}/tests/unit/test_server.py

@@ -237,6 +237,19 @@ class TestOpenAPISpecFiltering:
         assert "/teams" in filtered_spec["paths"]
         assert "/forbidden" not in filtered_spec["paths"]
         
+        # Verify pagination parameters were added to /incidents endpoint
+        incidents_get = filtered_spec["paths"]["/incidents"]["get"]
+        assert "parameters" in incidents_get
+        param_names = [p["name"] for p in incidents_get["parameters"]]
+        assert "page[size]" in param_names
+        assert "page[number]" in param_names
+        
+        # Verify /teams endpoint does not get pagination (doesn't contain "incidents" or "alerts")
+        teams_get = filtered_spec["paths"]["/teams"]["get"]
+        if "parameters" in teams_get:
+            param_names = [p["name"] for p in teams_get["parameters"]]
+            assert "page[size]" not in param_names
+        
         # Verify other properties are preserved
         assert filtered_spec["openapi"] == original_spec["openapi"]
         assert filtered_spec["info"] == original_spec["info"]
@@ -276,6 +289,83 @@ class TestOpenAPISpecFiltering:
         assert "servers" in filtered_spec
         assert "components" in filtered_spec
         assert filtered_spec["servers"] == original_spec["servers"]
+        
+        # Verify pagination parameters were added to /incidents endpoint
+        incidents_get = filtered_spec["paths"]["/incidents"]["get"]
+        assert "parameters" in incidents_get
+        param_names = [p["name"] for p in incidents_get["parameters"]]
+        assert "page[size]" in param_names
+        assert "page[number]" in param_names
+
+    def test_filter_spec_adds_pagination_to_alerts(self):
+        """Test that pagination parameters are added to alerts endpoints."""
+        original_spec = {
+            "openapi": "3.0.0",
+            "info": {"title": "Test API", "version": "1.0.0"},
+            "paths": {
+                "/alerts": {"get": {"operationId": "listAlerts"}},
+                "/incidents/123/alerts": {"get": {"operationId": "listIncidentAlerts"}},
+                "/users": {"get": {"operationId": "listUsers"}},
+            },
+            "components": {"schemas": {}}
+        }
+        
+        allowed_paths = ["/alerts", "/incidents/123/alerts", "/users"]
+        filtered_spec = _filter_openapi_spec(original_spec, allowed_paths)
+        
+        # Verify pagination was added to alerts endpoints
+        alerts_get = filtered_spec["paths"]["/alerts"]["get"]
+        assert "parameters" in alerts_get
+        param_names = [p["name"] for p in alerts_get["parameters"]]
+        assert "page[size]" in param_names
+        assert "page[number]" in param_names
+        
+        incident_alerts_get = filtered_spec["paths"]["/incidents/123/alerts"]["get"]
+        assert "parameters" in incident_alerts_get
+        param_names = [p["name"] for p in incident_alerts_get["parameters"]]
+        assert "page[size]" in param_names
+        assert "page[number]" in param_names
+        
+        # Verify pagination was NOT added to /users (no "incident" or "alerts" in path)
+        users_get = filtered_spec["paths"]["/users"]["get"]
+        if "parameters" in users_get:
+            param_names = [p["name"] for p in users_get["parameters"]]
+            assert "page[size]" not in param_names
+
+    def test_filter_spec_adds_pagination_to_incident_types(self):
+        """Test that pagination parameters are added to incident-related endpoints."""
+        original_spec = {
+            "openapi": "3.0.0",
+            "info": {"title": "Test API", "version": "1.0.0"},
+            "paths": {
+                "/incident_types": {"get": {"operationId": "listIncidentTypes"}},
+                "/incident_action_items": {"get": {"operationId": "listIncidentActionItems"}},
+                "/services": {"get": {"operationId": "listServices"}},
+            },
+            "components": {"schemas": {}}
+        }
+        
+        allowed_paths = ["/incident_types", "/incident_action_items", "/services"]
+        filtered_spec = _filter_openapi_spec(original_spec, allowed_paths)
+        
+        # Verify pagination was added to incident-related endpoints
+        incident_types_get = filtered_spec["paths"]["/incident_types"]["get"]
+        assert "parameters" in incident_types_get
+        param_names = [p["name"] for p in incident_types_get["parameters"]]
+        assert "page[size]" in param_names
+        assert "page[number]" in param_names
+        
+        incident_action_items_get = filtered_spec["paths"]["/incident_action_items"]["get"]
+        assert "parameters" in incident_action_items_get
+        param_names = [p["name"] for p in incident_action_items_get["parameters"]]
+        assert "page[size]" in param_names
+        assert "page[number]" in param_names
+        
+        # Verify pagination was NOT added to /services (no "incident" or "alerts" in path)
+        services_get = filtered_spec["paths"]["/services"]["get"]
+        if "parameters" in services_get:
+            param_names = [p["name"] for p in services_get["parameters"]]
+            assert "page[size]" not in param_names
 
 
 @pytest.mark.unit

{rootly_mcp_server-2.0.12 → rootly_mcp_server-2.0.13}/uv.lock

@@ -873,7 +873,7 @@ wheels = [
 
 [[package]]
 name = "rootly-mcp-server"
-version = "2.0.10"
+version = "2.0.12"
 source = { editable = "." }
 dependencies = [
     { name = "brotli" },