
    ]j                     l    d dl Z ddlmZmZ ddlmZ ddlmZ ddlm	Z	m
Z
  G d d	      Z G d
 d      Zy)    N   )AsyncClientWrapperSyncClientWrapper)RequestOptions)BillingInfoResponse   )AsyncRawBillingClientRawBillingClientc                   ^    e Zd ZdefdZedefd       Zdddej                  e
   defdZy)	BillingClientclient_wrapperc                &    t        |      | _        y N)r   )r
   _raw_clientselfr   s     I/root/env/lib/python3.12/site-packages/label_studio_sdk/billing/client.py__init__zBillingClient.__init__   s    +>J    returnc                     | j                   S )z
        Retrieves a raw implementation of this client that returns raw responses.

        Returns
        -------
        RawBillingClient
        r   r   s    r   with_raw_responsezBillingClient.with_raw_response        r   Nrequest_optionsr   c                R    | j                   j                  |      }|j                  S )a  
        <Card href="https://humansignal.com/goenterprise">
                <img style="pointer-events: none; margin-left: 0px; margin-right: 0px;" src="https://docs.humansignal.com/images/badge.svg" alt="Label Studio Enterprise badge"/>
                <p style="margin-top: 10px; font-size: 14px;">
                    This endpoint is not available in Label Studio Community Edition. [Learn more about Label Studio Enterprise](https://humansignal.com/goenterprise)
                </p>
            </Card>
        Retrieve billing checks and feature flags for the active organization.

        Parameters
        ----------
        request_options : typing.Optional[RequestOptions]
            Request-specific configuration.

        Returns
        -------
        BillingInfoResponse
            Billing information for the active organization

        Examples
        --------
        from label_studio_sdk import LabelStudio

        client = LabelStudio(
            api_key="YOUR_API_KEY",
        )
        client.billing.info()
        r   r   infodatar   r   	_responses      r   r    zBillingClient.info   s'    : $$))/)J	~~r   )__name__
__module____qualname__r   r   propertyr
   r   typingOptionalr   r   r     r   r   r   r      sQ    K*; K  #3     JN v~'F Re r   r   c                   ^    e Zd ZdefdZedefd       Zdddej                  e
   defdZy)	AsyncBillingClientr   c                &    t        |      | _        y r   )r	   r   r   s     r   r   zAsyncBillingClient.__init__<   s    0Or   r   c                     | j                   S )z
        Retrieves a raw implementation of this client that returns raw responses.

        Returns
        -------
        AsyncRawBillingClient
        r   r   s    r   r   z$AsyncBillingClient.with_raw_response?   r   r   Nr   r   c                n   K   | j                   j                  |       d{   }|j                  S 7 w)a  
        <Card href="https://humansignal.com/goenterprise">
                <img style="pointer-events: none; margin-left: 0px; margin-right: 0px;" src="https://docs.humansignal.com/images/badge.svg" alt="Label Studio Enterprise badge"/>
                <p style="margin-top: 10px; font-size: 14px;">
                    This endpoint is not available in Label Studio Community Edition. [Learn more about Label Studio Enterprise](https://humansignal.com/goenterprise)
                </p>
            </Card>
        Retrieve billing checks and feature flags for the active organization.

        Parameters
        ----------
        request_options : typing.Optional[RequestOptions]
            Request-specific configuration.

        Returns
        -------
        BillingInfoResponse
            Billing information for the active organization

        Examples
        --------
        import asyncio

        from label_studio_sdk import AsyncLabelStudio

        client = AsyncLabelStudio(
            api_key="YOUR_API_KEY",
        )


        async def main() -> None:
            await client.billing.info()


        asyncio.run(main())
        r   Nr   r"   s      r   r    zAsyncBillingClient.infoJ   s5     J **///PP	~~ Qs    535)r$   r%   r&   r   r   r'   r	   r   r(   r)   r   r   r    r*   r   r   r,   r,   ;   sQ    P*< P  #8     PT &V__^-L &Xk &r   r,   )r(   core.client_wrapperr   r   core.request_optionsr   types.billing_info_responser   
raw_clientr	   r
   r   r,   r*   r   r   <module>r4      s+     G 1 = ?- -`5 5r   