rootly-mcp-server 2.0.8__py3-none-any.whl → 2.0.10__py3-none-any.whl

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,7 @@ def create_rootly_mcp_server(
         return PlainTextResponse("OK")
     
     # Add some custom tools for enhanced functionality
+
     @mcp.tool()
     def list_endpoints() -> list:
         """List all available Rootly API endpoints with their descriptions."""
@@ -262,8 +281,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(
@@ -281,7 +300,7 @@ def create_rootly_mcp_server(
         # Single page mode
         if page_number > 0:
             params = {
-                "page[size]": min(page_size, 20),
+                "page[size]": min(page_size, 5),  # Keep responses very small to avoid errors
                 "page[number]": page_number,
                 "include": "",
             }
@@ -298,10 +317,11 @@ def create_rootly_mcp_server(
         # Multi-page mode (page_number = 0)
         all_incidents = []
         current_page = 1
-        effective_page_size = min(page_size, 10)
+        effective_page_size = min(page_size, 5)  # Keep responses very small to avoid errors
+        max_pages = 10  # Safety limit to prevent infinite loops
 
         try:
-            while len(all_incidents) < max_results:
+            while len(all_incidents) < max_results and current_page <= max_pages:
                 params = {
                     "page[size]": effective_page_size,
                     "page[number]": current_page,
@@ -318,16 +338,23 @@ def create_rootly_mcp_server(
                     if "data" in response_data:
                         incidents = response_data["data"]
                         if not incidents:
+                            # No more incidents available
+                            break
+                        
+                        # Check if we got fewer incidents than requested (last page)
+                        if len(incidents) < effective_page_size:
+                            all_incidents.extend(incidents)
                             break
                         
                         all_incidents.extend(incidents)
 
-                        # Check if we have more pages
+                        # Check metadata if available
                         meta = response_data.get("meta", {})
                         current_page_meta = meta.get("current_page", current_page)
-                        total_pages = meta.get("total_pages", 1)
-
-                        if current_page_meta >= total_pages:
+                        total_pages = meta.get("total_pages")
+                        
+                        # If we have reliable metadata, use it
+                        if total_pages and current_page_meta >= total_pages:
                             break
 
                         current_page += 1
@@ -377,7 +404,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 +412,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 +425,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 +433,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 +444,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:

{rootly_mcp_server-2.0.8.dist-info → rootly_mcp_server-2.0.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rootly-mcp-server
-Version: 2.0.8
+Version: 2.0.10
 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.dist-info → rootly_mcp_server-2.0.10.dist-info}/RECORD

@@ -2,12 +2,12 @@ rootly_mcp_server/__init__.py,sha256=6pLh19IFyqE-Cve9zergkD-X_yApEkInREKmRa73T6s
 rootly_mcp_server/__main__.py,sha256=_F4p65_VjnN84RtmEdESVLLH0tO5tL9qBfb2Xdvbj2E,6480
 rootly_mcp_server/client.py,sha256=diIBINJP_z4nnQIAC1b70vQSiHaNojEfUDARC2nrKHU,4681
 rootly_mcp_server/routemap_server.py,sha256=0LfK2EzwkFQF9SpHNvGcca5ZaxkBC80gIdDojE0aUcs,6100
-rootly_mcp_server/server.py,sha256=os6uxXUOGeaPrPlq3r2GNLAM-iJYVIe2W5loASMkaBc,22953
+rootly_mcp_server/server.py,sha256=O31ByNNfKjBmmYY1y_ooPKL12b-PXaM5eNrekXzpURE,24346
 rootly_mcp_server/test_client.py,sha256=Ytd5ZP7vImm12CT97k3p9tlkY_JNcXHSzcGGnHCBqv0,5275
 rootly_mcp_server/utils.py,sha256=NyxdcDiFGlV2a8eBO4lKgZg0D7Gxr6xUIB0YyJGgpPA,4165
 rootly_mcp_server/data/__init__.py,sha256=fO8a0bQnRVEoRMHKvhFzj10bhoaw7VsI51czc2MsUm4,143
-rootly_mcp_server-2.0.8.dist-info/METADATA,sha256=F1BHBKiIdJmlblKhzWDX55KLNFPXn4FfomeisilOlrw,6159
-rootly_mcp_server-2.0.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-rootly_mcp_server-2.0.8.dist-info/entry_points.txt,sha256=NE33b8VgigVPGBkboyo6pvN1Vz35HZtLybxMO4Q03PI,70
-rootly_mcp_server-2.0.8.dist-info/licenses/LICENSE,sha256=c9w9ZZGl14r54tsP40oaq5adTVX_HMNHozPIH2ymzmw,11341
-rootly_mcp_server-2.0.8.dist-info/RECORD,,
+rootly_mcp_server-2.0.10.dist-info/METADATA,sha256=SyvYvw80MbIJVhxXvAG_ai1F9FV9N3qPx9Ls4bj3mbg,6189
+rootly_mcp_server-2.0.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+rootly_mcp_server-2.0.10.dist-info/entry_points.txt,sha256=NE33b8VgigVPGBkboyo6pvN1Vz35HZtLybxMO4Q03PI,70
+rootly_mcp_server-2.0.10.dist-info/licenses/LICENSE,sha256=c9w9ZZGl14r54tsP40oaq5adTVX_HMNHozPIH2ymzmw,11341
+rootly_mcp_server-2.0.10.dist-info/RECORD,,