o êi[@sdZddlZddlmZddlmZmZmZddlZddl m Z ddl m Z m Z mZmZmZmZdd lmZd ejd dfd d ZGdddZGdddeZGdddeZdS)z=HTTP client wrapper for managing requests and authentication.N)JSONDecodeError)AnyDictOptional) __version__)APIErrorAuthenticationError NotFoundErrorPermissionErrorRateLimitError ServerError)RequestOptionsresponsereturncCs|j}z|}|dp|dp|j}Wnty(|jp%tj|}Ynw|dkr4t|||j|dkr?t |||j|dkrJt |||j|dkrUt |||j|dkr`t |||jt |||j)z1Raise appropriate exception based on status code.messagedetailiiiii) status_codejsongettextrhttpxcodesget_reason_phraser r r r r r)rstatus error_datarrQ/home/ubuntu/.local/lib/python3.10/site-packages/fishaudio/core/client_wrapper.py_raise_for_statuss$ rc@szeZdZdZddddeedefddZ dd eeeefd eeeffd d Zd ee deee fd dfddZ dS)BaseClientWrapperz6Base wrapper with shared logic for sync/async clients.Nhttps://api.fish.audioapi_keybase_urlr#r$cCs(|ptd|_|jstd||_dS)N FISH_API_KEYzTAPI key must be provided either as argument or via FISH_API_KEY environment variable)osgetenvr# ValueErrorr$)selfr#r$rrr__init__2s  zBaseClientWrapper.__init__additional_headersrcCs*d|jdtd}|r|||S)z6Build headers including authentication and user agent.zBearer zfish-audio/python/) Authorizationz User-Agent)r#rupdate)r)r+headersrrr get_headers?s  zBaseClientWrapper.get_headersrequest_optionskwargscCs|}|r|jr||ji||di|d<|r*|jdur*t|j|d<|rF|jrH|di}t|t rJ||j||d<dSdSdSdS)zEPrepare request kwargs by merging headers, timeout, and query params.r.Ntimeoutparams) r/r+r-rr2rTimeoutadditional_query_params isinstancedict)r)r0r1r.r3rrr_prepare_request_kwargsKs       z)BaseClientWrapper._prepare_request_kwargsN) __name__ __module__ __qualname____doc__rstrr*rr/rrr8rrrrr /s,     r c eZdZdZddddddeededed eejffd d Z dd d ededee de dej f ddZ edejfddZdddZddZddZZS) ClientWrapperzHWrapper for httpx.Client that handles authentication and error handling.Nr!n@r#r$r2 httpx_clientr#r$r2rCcBtj||d|dur||_dStj|t||d|_dSNr")r$r2r.)superr*_clientrClientr4r/r)r#r$r2rC __class__rrr*d  zClientWrapper.__init__r0methodpathr0r1rcKs4||||jj||fi|}|jst||S)a Make an HTTP request with error handling. Args: method: HTTP method (GET, POST, etc.) path: API endpoint path request_options: Optional request-level overrides **kwargs: Additional arguments to pass to httpx.request Returns: httpx.Response object Raises: APIError: On non-2xx responses r8rGrequest is_successrr)rNrOr0r1rrrrrQws zClientWrapper.requestcC|jS)zBGet underlying httpx.Client for advanced usage (e.g., WebSockets).rGr)rrrclientzClientWrapper.clientcCs|jdSzClose the HTTP client.N)rGcloserVrrrrZszClientWrapper.closecCs|Sr9rrVrrr __enter__szClientWrapper.__enter__cCs |dSr9rZr)exc_typeexc_valexc_tbrrr__exit__s zClientWrapper.__exit__rN)r:r;r<r=rr>floatrrHr*rrResponserQpropertyrWrZr[ra __classcell__rrrJrr@a@ " r@c r?)AsyncClientWrapperzMWrapper for httpx.AsyncClient that handles authentication and error handling.Nr!rArBr#r$r2rCcrDrE)rFr*rGr AsyncClientr4r/rIrJrrr*rLzAsyncClientWrapper.__init__rMrNrOr0r1rcs<||||jj||fi|IdH}|jst||S)a Make an async HTTP request with error handling. Args: method: HTTP method (GET, POST, etc.) path: API endpoint path request_options: Optional request-level overrides **kwargs: Additional arguments to pass to httpx.request Returns: httpx.Response object Raises: APIError: On non-2xx responses NrPrSrrrrQs  zAsyncClientWrapper.requestcCrT)zGGet underlying httpx.AsyncClient for advanced usage (e.g., WebSockets).rUrVrrrrWrXzAsyncClientWrapper.clientcs|jIdHdSrY)rGacloserVrrrrZszAsyncClientWrapper.closecs|Sr9rrVrrr __aenter__szAsyncClientWrapper.__aenter__cs|IdHdSr9r\r]rrr __aexit__szAsyncClientWrapper.__aexit__rb)r:r;r<r=rr>rcrrir*rrrdrQrerWrZrkrlrfrrrJrrhrgrh)r=r&rrtypingrrrr_versionr exceptionsrr r r r r r0rrdrr r@rhrrrrs    2H