rootly-mcp-server 2.0.8__tar.gz → 2.0.9__tar.gz

{rootly_mcp_server-2.0.8 → rootly_mcp_server-2.0.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rootly-mcp-server
-Version: 2.0.8
+Version: 2.0.9
 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
@@ -14,6 +14,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Software Development :: Build Tools
 Requires-Python: >=3.12
+Requires-Dist: brotli>=1.0.0
 Requires-Dist: fastmcp>=2.9.0
 Requires-Dist: httpx>=0.24.0
 Requires-Dist: pydantic>=2.0.0

{rootly_mcp_server-2.0.8 → rootly_mcp_server-2.0.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "rootly-mcp-server"
-version = "2.0.8"
+version = "2.0.9"
 description = "A Model Context Protocol server for Rootly APIs using OpenAPI spec"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -20,6 +20,7 @@ dependencies = [
     "requests>=2.28.0", # For API calls
     "httpx>=0.24.0",    # For async HTTP client
     "pydantic>=2.0.0",  # For data validation
+    "brotli>=1.0.0",    # For Brotli compression support in httpx
 ]
 
 [project.urls]

{rootly_mcp_server-2.0.8 → rootly_mcp_server-2.0.9}/src/rootly_mcp_server/server.py

@@ -69,7 +69,7 @@ class AuthenticatedHTTPXClient:
     """An HTTPX client wrapper that handles Rootly API authentication and parameter transformation."""
 
     def __init__(self, base_url: str = "https://api.rootly.com", hosted: bool = False, parameter_mapping: Optional[Dict[str, str]] = None):
-        self.base_url = base_url
+        self._base_url = base_url
         self.hosted = hosted
         self._api_token = None
         self.parameter_mapping = parameter_mapping or {}
@@ -77,15 +77,22 @@ class AuthenticatedHTTPXClient:
         if not self.hosted:
             self._api_token = self._get_api_token()
 
-        # Create the HTTPX client
-        headers = {"Content-Type": "application/vnd.api+json", "Accept": "application/vnd.api+json"}
+        # Create the HTTPX client  
+        headers = {
+            "Content-Type": "application/vnd.api+json", 
+            "Accept": "application/vnd.api+json"
+            # Let httpx handle Accept-Encoding automatically with all supported formats
+        }
         if self._api_token:
             headers["Authorization"] = f"Bearer {self._api_token}"
 
         self.client = httpx.AsyncClient(
             base_url=base_url,
             headers=headers,
-            timeout=30.0
+            timeout=30.0,
+            follow_redirects=True,
+            # Ensure proper handling of compressed responses
+            limits=httpx.Limits(max_keepalive_connections=5, max_connections=10)
         )
 
     def _get_api_token(self) -> Optional[str]:
@@ -116,7 +123,7 @@ class AuthenticatedHTTPXClient:
         if 'params' in kwargs:
             kwargs['params'] = self._transform_params(kwargs['params'])
 
-        # Call the underlying client's request method
+        # 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):
@@ -146,8 +153,19 @@ class AuthenticatedHTTPXClient:
         pass
 
     def __getattr__(self, name):
-        # Delegate all other attributes to the underlying client
+        # Delegate all other attributes to the underlying client, except for request methods
+        if name in ['request', 'get', 'post', 'put', 'patch', 'delete']:
+            # Use our overridden methods instead
+            return getattr(self, name)
         return getattr(self.client, name)
+    
+    @property 
+    def base_url(self):
+        return self._base_url
+        
+    @property
+    def headers(self):
+        return self.client.headers
 
 
 def create_rootly_mcp_server(
@@ -225,6 +243,43 @@ def create_rootly_mcp_server(
         return PlainTextResponse("OK")
     
     # Add some custom tools for enhanced functionality
+    @mcp.tool()
+    async def debug_incidents() -> dict:
+        """Debug tool to inspect incidents endpoint response."""
+        try:
+            response = await make_authenticated_request("GET", "/v1/incidents", params={"page[size]": 1})
+            response.raise_for_status()
+            
+            return {
+                "status_code": response.status_code,
+                "headers": dict(response.headers),
+                "content_length": len(response.content) if response.content else 0,
+                "content_preview": response.content[:500].decode('utf-8', errors='ignore') if response.content else "No content",
+                "text_preview": response.text[:500] if hasattr(response, 'text') else "No text",
+                "encoding": response.encoding,
+                "content_type": response.headers.get('content-type', 'unknown')
+            }
+        except Exception as e:
+            return {"error": str(e), "error_type": type(e).__name__}
+
+    @mcp.tool()
+    async def debug_headers() -> dict:
+        """Debug tool to inspect request/response headers for troubleshooting."""
+        try:
+            response = await make_authenticated_request("GET", "/v1/teams", params={"page[size]": 1})
+            response.raise_for_status()
+            
+            return {
+                "request_headers": dict(response.request.headers) if response.request else {},
+                "response_headers": dict(response.headers),
+                "status_code": response.status_code,
+                "content_type": response.headers.get('content-type', 'unknown'),
+                "encoding": response.encoding,
+                "content_preview": str(response.content[:200]) if response.content else "No content"
+            }
+        except Exception as e:
+            return {"error": str(e), "error_type": type(e).__name__}
+
     @mcp.tool()
     def list_endpoints() -> list:
         """List all available Rootly API endpoints with their descriptions."""
@@ -262,8 +317,8 @@ def create_rootly_mcp_server(
             except Exception:
                 pass  # Fallback to default client behavior
         
-        # Make the request using the underlying httpx client
-        return await http_client.client.request(method, url, **kwargs)
+        # Use our custom client with proper error handling instead of bypassing it
+        return await http_client.request(method, url, **kwargs)
 
     @mcp.tool()
     async def search_incidents(
@@ -377,7 +432,7 @@ def _load_swagger_spec(swagger_path: Optional[str] = None) -> Dict[str, Any]:
         logger.info(f"Using provided Swagger path: {swagger_path}")
         if not os.path.isfile(swagger_path):
             raise FileNotFoundError(f"Swagger file not found at {swagger_path}")
-        with open(swagger_path, "r") as f:
+        with open(swagger_path, "r", encoding="utf-8") as f:
             return json.load(f)
     else:
         # First, check in the package data directory
@@ -385,7 +440,7 @@ def _load_swagger_spec(swagger_path: Optional[str] = None) -> Dict[str, Any]:
             package_data_path = Path(__file__).parent / "data" / "swagger.json"
             if package_data_path.is_file():
                 logger.info(f"Found Swagger file in package data: {package_data_path}")
-                with open(package_data_path, "r") as f:
+                with open(package_data_path, "r", encoding="utf-8") as f:
                     return json.load(f)
         except Exception as e:
             logger.debug(f"Could not load Swagger file from package data: {e}")
@@ -398,7 +453,7 @@ def _load_swagger_spec(swagger_path: Optional[str] = None) -> Dict[str, Any]:
         swagger_path = current_dir / "swagger.json"
         if swagger_path.is_file():
             logger.info(f"Found Swagger file at {swagger_path}")
-            with open(swagger_path, "r") as f:
+            with open(swagger_path, "r", encoding="utf-8") as f:
                 return json.load(f)
 
         # Check parent directories
@@ -406,7 +461,7 @@ def _load_swagger_spec(swagger_path: Optional[str] = None) -> Dict[str, Any]:
             swagger_path = parent / "swagger.json"
             if swagger_path.is_file():
                 logger.info(f"Found Swagger file at {swagger_path}")
-                with open(swagger_path, "r") as f:
+                with open(swagger_path, "r", encoding="utf-8") as f:
                     return json.load(f)
 
         # If the file wasn't found, fetch it from the URL and save it
@@ -417,7 +472,7 @@ def _load_swagger_spec(swagger_path: Optional[str] = None) -> Dict[str, Any]:
         swagger_path = current_dir / "swagger.json"
         logger.info(f"Saving Swagger file to {swagger_path}")
         try:
-            with open(swagger_path, "w") as f:
+            with open(swagger_path, "w", encoding="utf-8") as f:
                 json.dump(swagger_spec, f)
             logger.info(f"Saved Swagger file to {swagger_path}")
         except Exception as e: