import suger_sdk_python

import suger_sdk_python

import suger_sdk_python
from suger_sdk_python.rest import ApiException
from pprint import pprint

# Defining the host is optional and defaults to https://api.suger.cloud
# See configuration.py for a list of all supported configuration parameters.
configuration = suger_sdk_python.Configuration(
    host = "https://api.suger.cloud"
)

# The client must configure the authentication and authorization parameters
# in accordance with the API server security policy.
# Examples for each auth method are provided below, use the example that
# satisfies your auth use case.

# Configure API key authorization: APIKeyAuth
configuration.api_key['APIKeyAuth'] = os.environ["API_KEY"]

# Uncomment below to setup prefix (e.g. Bearer) for API key, if needed
# configuration.api_key_prefix['APIKeyAuth'] = 'Bearer'


# Enter a context with an instance of the API client
with suger_sdk_python.ApiClient(configuration) as api_client:
    # Create an instance of the API class
    api_instance = suger_sdk_python.AIUsageApi(api_client)
    org_id = 'org_id_example' # str | Organization ID
    start_time = 'start_time_example' # str | Start time (RFC3339 format)
    end_time = 'end_time_example' # str | End time (RFC3339 format)
    limit = 56 # int | Maximum number of records to return (default: 1000, max: 10000) (optional)
    offset = 56 # int | Number of records to skip for pagination (default: 0) (optional)

    try:
        # Get AI usage data
        api_response = api_instance.get_ai_usage(org_id, start_time, end_time, limit=limit, offset=offset)
        print("The response of AIUsageApi->get_ai_usage:\n")
        pprint(api_response)
    except ApiException as e:
        print("Exception when calling AIUsageApi->get_ai_usage: %s\n" % e)