o
    i                     @   sl   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	 ddl
mZmZ G d	d
 d
ZG dd dZdS )    N   )AsyncClientWrapperSyncClientWrapper)RequestOptions)GetModelV1Response)ListModelsV1Response   )AsyncRawModelsClientRawModelsClientc                   @   x   e Zd ZdefddZedefddZdddd	ej	e
 d
ej	e defddZddded
ej	e defddZdS )ModelsClientclient_wrapperc                C      t |d| _d S N)r   )r
   _raw_clientselfr    r   T/home/ubuntu/.local/lib/python3.10/site-packages/deepgram/manage/v1/models/client.py__init__      zModelsClient.__init__returnc                 C      | j S )z
        Retrieves a raw implementation of this client that returns raw responses.

        Returns
        -------
        RawModelsClient
        r   r   r   r   r   with_raw_response      	zModelsClient.with_raw_responseNinclude_outdatedrequest_optionsr   r   c                C      | j j||d}|jS )a  
        Returns metadata on all the latest public models. To retrieve custom models, use Get Project Models.

        Parameters
        ----------
        include_outdated : typing.Optional[bool]
            returns non-latest versions of models

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

        Returns
        -------
        ListModelsV1Response
            A list of models

        Examples
        --------
        from deepgram import DeepgramClient

        client = DeepgramClient(
            api_key="YOUR_API_KEY",
        )
        client.manage.v1.models.list(
            include_outdated=True,
        )
        r   r   listdatar   r   r   	_responser   r   r   r"      s   zModelsClient.listr   model_idc                C   r    )a  
        Returns metadata for a specific public model

        Parameters
        ----------
        model_id : str
            The specific UUID of the model

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

        Returns
        -------
        GetModelV1Response
            A model object that can be either STT or TTS

        Examples
        --------
        from deepgram import DeepgramClient

        client = DeepgramClient(
            api_key="YOUR_API_KEY",
        )
        client.manage.v1.models.get(
            model_id="af6e9977-99f6-4d8f-b6f5-dfdf6fb6e291",
        )
        r&   r   getr#   r   r'   r   r%   r   r   r   r)   <   s   zModelsClient.get)__name__
__module____qualname__r   r   propertyr
   r   typingOptionalboolr   r   r"   strr   r)   r   r   r   r   r      s    
&!r   c                   @   r   )AsyncModelsClientr   c                C   r   r   )r	   r   r   r   r   r   r   ]   r   zAsyncModelsClient.__init__r   c                 C   r   )z
        Retrieves a raw implementation of this client that returns raw responses.

        Returns
        -------
        AsyncRawModelsClient
        r   r   r   r   r   r   `   r   z#AsyncModelsClient.with_raw_responseNr   r   r   c                      | j j||dI dH }|jS )a*  
        Returns metadata on all the latest public models. To retrieve custom models, use Get Project Models.

        Parameters
        ----------
        include_outdated : typing.Optional[bool]
            returns non-latest versions of models

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

        Returns
        -------
        ListModelsV1Response
            A list of models

        Examples
        --------
        import asyncio

        from deepgram import AsyncDeepgramClient

        client = AsyncDeepgramClient(
            api_key="YOUR_API_KEY",
        )


        async def main() -> None:
            await client.manage.v1.models.list(
                include_outdated=True,
            )


        asyncio.run(main())
        r   Nr!   r$   r   r   r   r"   k      &zAsyncModelsClient.listr&   r'   c                   r4   )a  
        Returns metadata for a specific public model

        Parameters
        ----------
        model_id : str
            The specific UUID of the model

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

        Returns
        -------
        GetModelV1Response
            A model object that can be either STT or TTS

        Examples
        --------
        import asyncio

        from deepgram import AsyncDeepgramClient

        client = AsyncDeepgramClient(
            api_key="YOUR_API_KEY",
        )


        async def main() -> None:
            await client.manage.v1.models.get(
                model_id="af6e9977-99f6-4d8f-b6f5-dfdf6fb6e291",
            )


        asyncio.run(main())
        r&   Nr(   r*   r   r   r   r)      r5   zAsyncModelsClient.get)r+   r,   r-   r   r   r.   r	   r   r/   r0   r1   r   r   r"   r2   r   r)   r   r   r   r   r3   \   s(    
*r3   )r/   core.client_wrapperr   r   core.request_optionsr   types.get_model_v1responser   types.list_models_v1responser   
raw_clientr	   r
   r   r3   r   r   r   r   <module>   s   P