Implementation:Mage ai Mage ai PowerBI Client

Knowledge Sources	Mage Integrations Power BI API
Domains	Data_Integration, PowerBI, API
Last Updated	2026-02-09 00:00 GMT

Overview

Concrete HTTP client for making authenticated requests to the Power BI REST API, provided by the Mage Power BI source connector.

Description

The PowerbiClient class manages all HTTP communication with the Power BI API at https://api.powerbi.com/v1.0/myorg. It authenticates using Bearer token via access_token. The client is implemented as a context manager, verifying the access token on entry by hitting the /dashboards endpoint and closing the session on exit. Rate limiting is enforced via the Singer @utils.ratelimit(1000, 60) decorator. It uses Singer metrics.http_request_timer for request timing instrumentation. A comprehensive exception hierarchy maps HTTP status codes (400-500) to specific error classes. Configurable request timeout defaults to 300 seconds. Validates token by checking for value key in response.

Usage

Used internally by Power BI stream classes. Requires access_token and optionally request_timeout and user_agent.

Code Reference

Source Location

Repository: mage-ai
File: mage_integrations/mage_integrations/sources/powerbi/client.py
Lines: 1-250

Signature

class PowerbiClient(object):
    def __init__(
        self,
        logger,
        access_token,
        config_request_timeout,
        user_agent=None,
    ):

Import

from mage_integrations.sources.powerbi.client import PowerbiClient

I/O Contract

Inputs

Name	Type	Required	Description
logger	Logger	Yes	Logger instance for diagnostic output
access_token	str	Yes	Bearer token for Power BI API authentication
config_request_timeout	float/str	Yes	Request timeout in seconds (defaults to 300 if falsy)
user_agent	str	No	Optional User-Agent header value

Outputs

Name	Type	Description
response	dict	Parsed JSON response from the Power BI API

Key Methods

Method	Description
`check_access_token()`	Validates the access token by querying `/dashboards`. Checks for `value` key in response.
`request(method, path, url, **kwargs)`	Core request method with rate limiting, metric timing, and error handling.
`get(path, **kwargs)`	Convenience wrapper for GET requests with logging.
`post(path, **kwargs)`	Convenience wrapper for POST requests.
`perform(method, path, **kwargs)`	Dispatches to `get` or `post` based on method string.

Error Handling

Defines a hierarchy of custom exceptions mapped to HTTP status codes (400, 401, 402, 403, 404, 405, 406, 408, 409, 415, 422, 423, 500). Falls back to PowerbiError for unmapped codes. Raises Server5xxError for status codes >= 500.

Usage Examples

from mage_integrations.sources.powerbi.client import PowerbiClient

with PowerbiClient(
    logger=logger,
    access_token='your_powerbi_access_token',
    config_request_timeout=300,
) as client:
    dashboards = client.get('dashboards')
    datasets = client.get('datasets')

Related Pages

Implements Principle

Principle:Mage_ai_Mage_ai_HTTP_Client_Request

Requires Environment

Environment:Mage_ai_Mage_ai_Python_3_9_Runtime

Page Connections

Double-click a node to navigate. Hold to expand connections.

Principle

Implementation

Heuristic

Environment