rootly-mcp-server 2.1.0__py3-none-any.whl → 2.1.1__py3-none-any.whl

rootly_mcp_server/server.py

@@ -163,131 +163,9 @@ def strip_heavy_nested_data(data: dict[str, Any]) -> dict[str, Any]:
                         # Replace with just count
                         rels[rel_key] = {"count": len(rels[rel_key]["data"])}
 
-    # Process "included" section (common in shifts/alerts with user data)
-    if "included" in data and isinstance(data["included"], list):
-        for item in data["included"]:
-            if item.get("type") == "users":
-                # Keep only essential user fields
-                if "attributes" in item:
-                    attrs = item["attributes"]
-                    keep_fields = {"name", "email", "phone", "time_zone", "full_name"}
-                    item["attributes"] = {k: v for k, v in attrs.items() if k in keep_fields}
-                # Strip heavy relationships
-                if "relationships" in item:
-                    for rel_key in [
-                        "schedules",
-                        "notification_rules",
-                        "teams",
-                        "devices",
-                        "email_addresses",
-                        "phone_numbers",
-                    ]:
-                        if rel_key in item["relationships"]:
-                            rel_data = item["relationships"][rel_key]
-                            if isinstance(rel_data, dict) and "data" in rel_data:
-                                data_list = rel_data.get("data", [])
-                                if isinstance(data_list, list):
-                                    item["relationships"][rel_key] = {"count": len(data_list)}
-
-    # Process alerts in data list
-    if "data" in data and isinstance(data["data"], list):
-        for item in data["data"]:
-            if item.get("type") == "alerts":
-                # Strip heavy attributes from alerts
-                if "attributes" in item:
-                    attrs = item["attributes"]
-                    # Remove heavy fields - raw data, embedded objects, integration fields
-                    heavy_fields = [
-                        "data",  # Raw alert payload from source - very large
-                        "labels",
-                        "external_url",
-                        "pagerduty_incident_id",
-                        "pagerduty_incident_url",
-                        "opsgenie_alert_id",
-                        "opsgenie_alert_url",
-                        "deduplication_key",
-                    ]
-                    for field in heavy_fields:
-                        attrs.pop(field, None)
-
-                    # Simplify embedded objects to just IDs/counts
-                    # groups - keep only group_ids
-                    if "groups" in attrs:
-                        attrs.pop("groups", None)
-                    # environments - keep only environment_ids
-                    if "environments" in attrs:
-                        attrs.pop("environments", None)
-                    # services - keep only service_ids
-                    if "services" in attrs:
-                        attrs.pop("services", None)
-                    # incidents - embedded incident objects
-                    if "incidents" in attrs:
-                        attrs.pop("incidents", None)
-                    # responders - embedded responder objects
-                    if "responders" in attrs:
-                        attrs.pop("responders", None)
-                    # notified_users - embedded user objects
-                    if "notified_users" in attrs:
-                        attrs.pop("notified_users", None)
-                    # alerting_targets - embedded target objects
-                    if "alerting_targets" in attrs:
-                        attrs.pop("alerting_targets", None)
-                    # alert_urgency - keep only alert_urgency_id
-                    if "alert_urgency" in attrs:
-                        attrs.pop("alert_urgency", None)
-                    # alert_field_values - embedded custom field values
-                    if "alert_field_values" in attrs:
-                        attrs.pop("alert_field_values", None)
-
-                # Strip heavy relationships
-                if "relationships" in item:
-                    rels = item["relationships"]
-                    for rel_key in ["events", "subscribers", "alerts"]:
-                        if (
-                            rel_key in rels
-                            and isinstance(rels[rel_key], dict)
-                            and "data" in rels[rel_key]
-                        ):
-                            data_list = rels[rel_key].get("data", [])
-                            if isinstance(data_list, list):
-                                rels[rel_key] = {"count": len(data_list)}
-
     return data
 
 
-class ProcessedResponse:
-    """Wrapper around httpx.Response that processes JSON to reduce payload size."""
-
-    def __init__(self, response: httpx.Response):
-        self._response = response
-        self._processed_json = None
-
-    def json(self, **kwargs):
-        """Parse JSON and strip heavy nested data."""
-        if self._processed_json is None:
-            raw_data = self._response.json(**kwargs)
-            self._processed_json = strip_heavy_nested_data(raw_data)
-        return self._processed_json
-
-    def __getattr__(self, name):
-        """Delegate all other attributes to the wrapped response."""
-        return getattr(self._response, name)
-
-
-class ResponseProcessingClient(httpx.AsyncClient):
-    """AsyncClient subclass that wraps responses to reduce payload size.
-
-    This is necessary because FastMCP.from_openapi() uses the client directly,
-    bypassing any wrapper class. By subclassing httpx.AsyncClient, we ensure
-    all responses go through our processing.
-    """
-
-    async def request(self, method, url, **kwargs):
-        """Override request to wrap response with ProcessedResponse."""
-        response = await super().request(method, url, **kwargs)
-        return ProcessedResponse(response)
-
-
 class MCPError:
     """Enhanced error handling for MCP protocol compliance."""
 
@@ -468,7 +346,7 @@ class AuthenticatedHTTPXClient:
         if self._api_token:
             headers["Authorization"] = f"Bearer {self._api_token}"
 
-        self.client = ResponseProcessingClient(
+        self.client = httpx.AsyncClient(
             base_url=base_url,
             headers=headers,
             timeout=30.0,
@@ -500,16 +378,13 @@ class AuthenticatedHTTPXClient:
         return transformed
 
     async def request(self, method: str, url: str, **kwargs):
-        """Override request to transform parameters and wrap response for payload reduction."""
+        """Override request to transform parameters."""
         # Transform query parameters
         if "params" in kwargs:
             kwargs["params"] = self._transform_params(kwargs["params"])
 
-        # Call the underlying client's request method
-        response = await self.client.request(method, url, **kwargs)
-
-        # Wrap response to process JSON and reduce payload size
-        return ProcessedResponse(response)
+        # Call the underlying client's request method and let it handle everything
+        return await self.client.request(method, url, **kwargs)
 
     async def get(self, url: str, **kwargs):
         """Proxy to request with GET method."""
@@ -612,9 +487,12 @@ def create_rootly_mcp_server(
 
     # Create the MCP server using OpenAPI integration
     # By default, all routes become tools which is what we want
+    # NOTE: We pass http_client (the wrapper) instead of http_client.client (the inner httpx client)
+    # so that parameter transformation (e.g., filter_status -> filter[status]) is applied.
+    # The wrapper implements the same interface as httpx.AsyncClient (duck typing).
     mcp = FastMCP.from_openapi(
         openapi_spec=filtered_spec,
-        client=http_client.client,
+        client=http_client,  # type: ignore[arg-type]
         name=name,
         timeout=30.0,
         tags={"rootly", "incident-management"},

{rootly_mcp_server-2.1.0.dist-info → rootly_mcp_server-2.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rootly-mcp-server
-Version: 2.1.0
+Version: 2.1.1
 Summary: Secure Model Context Protocol server for Rootly APIs with AI SRE capabilities, comprehensive error handling, and input validation
 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.1.0.dist-info → rootly_mcp_server-2.1.1.dist-info}/RECORD

@@ -5,14 +5,14 @@ rootly_mcp_server/exceptions.py,sha256=67J_wlfOICg87eUipbkARzn_6u_Io82L-5cVnk2UP
 rootly_mcp_server/monitoring.py,sha256=k1X7vK65FOTrCrOsLUXrFm6AJxKpXt_a0PzL6xdPuVU,11681
 rootly_mcp_server/pagination.py,sha256=2hZSO4DLUEJZbdF8oDfIt2_7X_XGBG1jIxN8VGmeJBE,2420
 rootly_mcp_server/security.py,sha256=YkMoVALZ3XaKnMu3yF5kVf3SW_jdKHllSMwVLk1OlX0,11556
-rootly_mcp_server/server.py,sha256=tJLRJgurdFlq-7m7jCsRYCeD1LFUbLff0YQ7yUQJWVc,115527
+rootly_mcp_server/server.py,sha256=IQ2lZNZKckLRFr2ALjRL9AAjGWTsfQq4jkhlh0-BUig,109962
 rootly_mcp_server/smart_utils.py,sha256=c7S-8H151GfmDw6dZBDdLH_cCmR1qiXkKEYSKc0WwUY,23481
 rootly_mcp_server/texttest.json,sha256=KV9m13kWugmW1VEpU80Irp50uCcLgJtV1YT-JzMogQg,154182
 rootly_mcp_server/utils.py,sha256=TWG1MaaFKrU1phRhU6FgHuZAEv91JOe_1w0L2OrPJMY,4406
 rootly_mcp_server/validators.py,sha256=z1Lvel2SpOFLo1cPdQGSrX2ySt6zqR42w0R6QV9c2Cc,4092
 rootly_mcp_server/data/__init__.py,sha256=KdWD6hiRssHXt0Ywgj3wjNHY1sx-XSPEqVHqrTArf54,143
-rootly_mcp_server-2.1.0.dist-info/METADATA,sha256=0fXm7kY3z3npobc9BeEm-P0R6Et3yRGAisb-Wcr40vM,13560
-rootly_mcp_server-2.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-rootly_mcp_server-2.1.0.dist-info/entry_points.txt,sha256=NE33b8VgigVPGBkboyo6pvN1Vz35HZtLybxMO4Q03PI,70
-rootly_mcp_server-2.1.0.dist-info/licenses/LICENSE,sha256=c9w9ZZGl14r54tsP40oaq5adTVX_HMNHozPIH2ymzmw,11341
-rootly_mcp_server-2.1.0.dist-info/RECORD,,
+rootly_mcp_server-2.1.1.dist-info/METADATA,sha256=iaXmvsZRLzSyYHpERE-xdn0Rd6rrrCXbO57lWZeIBHs,13560
+rootly_mcp_server-2.1.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+rootly_mcp_server-2.1.1.dist-info/entry_points.txt,sha256=NE33b8VgigVPGBkboyo6pvN1Vz35HZtLybxMO4Q03PI,70
+rootly_mcp_server-2.1.1.dist-info/licenses/LICENSE,sha256=c9w9ZZGl14r54tsP40oaq5adTVX_HMNHozPIH2ymzmw,11341
+rootly_mcp_server-2.1.1.dist-info/RECORD,,