Updates per PR review

Author: adamjmcgrath

packages/auth0_api_python/README.md

@@ -126,7 +126,7 @@ decoded_and_verified_token = await api_client.verify_access_token(
 
 If the token lacks `my_custom_claim` or fails any standard check (issuer mismatch, expired token, invalid signature), the method raises a `VerifyAccessTokenError`.
 
-### 4. DPoP Authentication
+### 5. DPoP Authentication
 
 > [!NOTE]  
 > This feature is currently available in [Early Access](https://auth0.com/docs/troubleshoot/product-lifecycle/product-release-stages#early-access). Please reach out to Auth0 support to get it enabled for your tenant.

packages/auth0_api_python/src/auth0_api_python/api_client.py

@@ -460,7 +460,8 @@ async def get_access_token_for_connection(self, options: dict[str, Any]) -> dict
                 raise ApiError(
                     error_data.get("error", "connection_token_error"),
                     error_data.get(
-                        "error_description", f"Failed to get token for connection: {response.status_code}")
+                        "error_description", f"Failed to get token for connection: {response.status_code}"),
+                    response.status_code
                 )
 
             token_endpoint_response = response.json()

packages/auth0_api_python/src/auth0_api_python/errors.py

@@ -96,30 +96,37 @@ def get_error_code(self) -> str:
         return "invalid_request"
 
 
-class GetAccessTokenForConnectionError(Exception):
+class GetAccessTokenForConnectionError(BaseAuthError):
     """Error raised when getting a token for a connection fails."""
-    code = "get_access_token_for_connection_error"
 
-    def __init__(self, message: str):
-        super().__init__(message)
-        self.name = self.__class__.__name__
+    def get_status_code(self) -> int:
+        return 400
 
+    def get_error_code(self) -> str:
+        return "get_access_token_for_connection_error"
 
-class ApiError(Exception):
+
+class ApiError(BaseAuthError):
     """
     Error raised when an API request to Auth0 fails.
     Contains details about the original error from Auth0.
     """
 
-    def __init__(self, code: str, message: str, cause=None):
+    def __init__(self, code: str, message: str, status_code=500, cause=None):
         super().__init__(message)
         self.code = code
+        self.status_code = status_code
         self.cause = cause
 
-        # Extract additional error details if available
         if cause:
             self.error = getattr(cause, "error", None)
             self.error_description = getattr(cause, "error_description", None)
         else:
             self.error = None
             self.error_description = None
+
+    def get_status_code(self) -> int:
+        return self.status_code
+
+    def get_error_code(self) -> str:
+        return self.code

packages/auth0_api_python/tests/test_api_client.py

@@ -1730,3 +1730,4 @@ async def test_get_access_token_for_connection_token_endpoint_error(httpx_mock:
             "access_token": "user-token"
         })
     assert err.value.code == "invalid_request"
+    assert err.value.status_code == 400