rootly-mcp-server 2.1.3__py3-none-any.whl → 2.2.0__py3-none-any.whl

rootly_mcp_server/och_client.py

@@ -1,10 +1,13 @@
-"""On-Call Health API client for burnout risk analysis."""
+"""On-Call Health API client for workload health risk analysis."""
 
 import os
 from typing import Any
 
 import httpx
 
+# External API field mapping (On-Call Health API response field)
+_OCH_RISK_SCORE_FIELD = "burnout_score"
+
 
 class OnCallHealthClient:
     def __init__(self, api_key: str | None = None, base_url: str | None = None):
@@ -55,7 +58,7 @@ class OnCallHealthClient:
                 "rootly_user_id": member.get("rootly_user_id"),
                 "och_score": member.get("och_score", 0),
                 "risk_level": member.get("risk_level", "unknown"),
-                "burnout_score": member.get("burnout_score", 0),
+                "health_risk_score": member.get(_OCH_RISK_SCORE_FIELD, 0),
                 "incident_count": member.get("incident_count", 0),
             }
 

rootly_mcp_server/server.py

@@ -381,11 +381,21 @@ class AuthenticatedHTTPXClient:
         return transformed
 
     async def request(self, method: str, url: str, **kwargs):
-        """Override request to transform parameters."""
+        """Override request to transform parameters and ensure correct headers."""
         # Transform query parameters
         if "params" in kwargs:
             kwargs["params"] = self._transform_params(kwargs["params"])
 
+        # Ensure Content-Type and Accept headers are always set correctly for Rootly API
+        # This is critical because FastMCP may pass headers from the MCP client request
+        # (e.g., Content-Type: application/json from SSE) which would override our defaults
+        if "headers" in kwargs:
+            headers = dict(kwargs["headers"]) if kwargs["headers"] else {}
+            # Always use JSON-API content type for Rootly API
+            headers["Content-Type"] = "application/vnd.api+json"
+            headers["Accept"] = "application/vnd.api+json"
+            kwargs["headers"] = headers
+
         # Call the underlying client's request method and let it handle everything
         return await self.client.request(method, url, **kwargs)
 
@@ -3158,7 +3168,7 @@ Updated: {attributes.get("updated_at", "N/A")}"""
             }
 
     @mcp.tool()
-    async def check_oncall_burnout_risk(
+    async def check_oncall_health_risk(
         start_date: Annotated[
             str,
             Field(description="Start date for the on-call period (ISO 8601, e.g., '2026-02-09')"),
@@ -3182,11 +3192,11 @@ Updated: {attributes.get("updated_at", "N/A")}"""
             Field(description="Include recommended replacement responders (default: true)"),
         ] = True,
     ) -> dict:
-        """Check if any at-risk responders (based on On-Call Health burnout analysis) are scheduled for on-call.
+        """Check if any at-risk responders (based on On-Call Health analysis) are scheduled for on-call.
 
-        Integrates with On-Call Health (oncallhealth.ai) to identify responders at risk of burnout
-        and checks if they are scheduled during the specified period. Optionally recommends
-        safe replacement responders.
+        Integrates with On-Call Health (oncallhealth.ai) to identify responders with elevated
+        workload health risk and checks if they are scheduled during the specified period.
+        Optionally recommends safe replacement responders.
 
         Requires ONCALLHEALTH_API_KEY environment variable.
         """
@@ -3229,7 +3239,7 @@ Updated: {attributes.get("updated_at", "N/A")}"""
                         "total_at_risk": 0,
                         "at_risk_scheduled": 0,
                         "action_required": False,
-                        "message": "No users above burnout threshold.",
+                        "message": "No users above health risk threshold.",
                     },
                 }
 
@@ -3340,7 +3350,7 @@ Updated: {attributes.get("updated_at", "N/A")}"""
                             "user_id": int(rootly_id),
                             "och_score": user["och_score"],
                             "risk_level": user["risk_level"],
-                            "burnout_score": user["burnout_score"],
+                            "health_risk_score": user["health_risk_score"],
                             "total_hours": round(total_hours, 1),
                             "shifts": user_shifts,
                         }
@@ -3441,7 +3451,7 @@ Updated: {attributes.get("updated_at", "N/A")}"""
 
             error_type, error_message = MCPError.categorize_error(e)
             return MCPError.tool_error(
-                f"Failed to check burnout risk: {error_message}",
+                f"Failed to check health risk: {error_message}",
                 error_type,
                 details={
                     "params": {

{rootly_mcp_server-2.1.3.dist-info → rootly_mcp_server-2.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rootly-mcp-server
-Version: 2.1.3
+Version: 2.2.0
 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
@@ -147,11 +147,11 @@ Alternatively, connect directly to our hosted MCP server:
   - **`suggest_solutions`**: Mines past incident resolutions to recommend actionable solutions
 - **MCP Resources**: Exposes incident and team data as structured resources for easy AI reference
 - **Intelligent Pattern Recognition**: Automatically identifies services, error types, and resolution patterns
-- **On-Call Health Integration**: Detects burnout risk in scheduled responders
+- **On-Call Health Integration**: Detects workload health risk in scheduled responders
 
 ## On-Call Health Integration
 
-Rootly MCP integrates with [On-Call Health](https://oncallhealth.ai) to detect burnout risk in scheduled responders.
+Rootly MCP integrates with [On-Call Health](https://oncallhealth.ai) to detect workload health risk in scheduled responders.
 
 ### Setup
 
@@ -175,7 +175,7 @@ Set the `ONCALLHEALTH_API_KEY` environment variable:
 ### Usage
 
 ```
-check_oncall_burnout_risk(
+check_oncall_health_risk(
     start_date="2026-02-09",
     end_date="2026-02-15"
 )

{rootly_mcp_server-2.1.3.dist-info → rootly_mcp_server-2.2.0.dist-info}/RECORD

@@ -3,17 +3,17 @@ rootly_mcp_server/__main__.py,sha256=mt74vaOpfHnX5rTO0CFAeulatR_9K3NBNCaLAhBLxlc
 rootly_mcp_server/client.py,sha256=Qca2R9cgBxXcyobQj4RHl8gdxLB4Jphq0RIr61DAVKw,6542
 rootly_mcp_server/exceptions.py,sha256=67J_wlfOICg87eUipbkARzn_6u_Io82L-5cVnk2UPr0,4504
 rootly_mcp_server/monitoring.py,sha256=k1X7vK65FOTrCrOsLUXrFm6AJxKpXt_a0PzL6xdPuVU,11681
-rootly_mcp_server/och_client.py,sha256=tz_Qw8NwAmrlSjtl-V6WNVChiIofyur06YJZRfYotKE,2737
+rootly_mcp_server/och_client.py,sha256=6UBP7MLkaDOGcJ78hQgSqTyrtr41XwDTqOOfUnoMJkc,2861
 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=9XcQ5_yOv4ofMRjXoDg3LsPFb8qDiOHEFz9dPNfwl7o,163221
+rootly_mcp_server/server.py,sha256=neVjS-7AmEPIK7U2IfQ5TdAchHPnOaLONH_0nAHDX6g,163869
 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.3.dist-info/METADATA,sha256=321Q8VyO2kJGCHbpEyf1NT73jZpB7Eb-KxbECDZcnVs,9892
-rootly_mcp_server-2.1.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-rootly_mcp_server-2.1.3.dist-info/entry_points.txt,sha256=NE33b8VgigVPGBkboyo6pvN1Vz35HZtLybxMO4Q03PI,70
-rootly_mcp_server-2.1.3.dist-info/licenses/LICENSE,sha256=c9w9ZZGl14r54tsP40oaq5adTVX_HMNHozPIH2ymzmw,11341
-rootly_mcp_server-2.1.3.dist-info/RECORD,,
+rootly_mcp_server-2.2.0.dist-info/METADATA,sha256=0sXGTYmYNvifnLHGoOPC7VJppxKDVWcZio-EbUVy3-0,9907
+rootly_mcp_server-2.2.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+rootly_mcp_server-2.2.0.dist-info/entry_points.txt,sha256=NE33b8VgigVPGBkboyo6pvN1Vz35HZtLybxMO4Q03PI,70
+rootly_mcp_server-2.2.0.dist-info/licenses/LICENSE,sha256=c9w9ZZGl14r54tsP40oaq5adTVX_HMNHozPIH2ymzmw,11341
+rootly_mcp_server-2.2.0.dist-info/RECORD,,