o iI%@sdZddlmZmZmZmZmZddlZddlZddl Z ddl m Z er4ddl m Z ddlZddlmZGdd d eZGd d d eZGd d d eZGdddeZGdddeZGdddeZGdddeZe jZdS)z Error classes for the GenAI SDK.)AnyCallableOptional TYPE_CHECKINGUnionN)_common)ReplayResponse)Responsec seZdZUdZeed<eejde jdfed<dZ e e ed<dZ e e ed< d%ded ede eejde jdfffd d Zd eed eee efffddZedee efd dfddZd ed efddZd ed efddZd ed efddZd ejfddZedede jejfd dfddZeded ede ede jejfd dfddZedede jd dfd dfd!d"Zeded ede ede jd fd dfd#d$Z Z!S)&APIErrorz'General errors raised by the GenAI API.coder AsyncAuthorizedSessionResponseresponseNstatusmessage response_jsoncs|t|trt|dkr|d}||_||_|||_|||_|r%|n| ||_ t |j d|jd|jdS)Nrr . ) isinstancelistlenrdetails _get_messager _get_statusr _get_coder super__init__)selfr rr __class__K/home/livre-enfant/venv/lib/python3.10/site-packages/google/genai/errors.pyr,s   &zAPIError.__init__return).r cCs|j}|jj|ffS)zGReturns a tuple that can be used to reconstruct the error for pickling.)__dict__copyr_rebuild)rstater r r! __reduce__Ds zAPIError.__reduce__r&cCs<tt}|j|t||jd|jd|j|S)z"Rebuilds the error from the state.rr) r __new__r#update Exceptionrr rr)r&objr r r!r%Ks  "zAPIError._rebuildcC8z|d|didd}|WStyYdSw)NrerrorgetAttributeError)rrrr r r!rS zAPIError._get_statuscCr,)Nrr-r.)rrrr r r!r^r1zAPIError._get_messagecCs|d|diddS)Nr r-)r/)rrr r r!riszAPIError._get_codecCsd|j|j|jdiS)zReturns a dictionary representation of the error for replay recording. details is not included since it may expose internal information in the replay file. r-r rrr2)rr r r!_to_replay_recordns zAPIError._to_replay_recordcCs|jdkrdSt|tjr,z ||}Wn<tjjy+|j}||j d}Yn)wt|t jrKz|}Wnt j jyJ|j|j d}Yn w|j ddi}||j||dS)PRaises an error with detailed error message if the response has an error status.Nrrrr-) status_coderhttpxr readjsondecoderJSONDecodeErrortext reason_phraserequests exceptionsreason body_segmentsr/ raise_error)clsrrrr r r!raise_for_response|s,       zAPIError.raise_for_responser7cCsTd|kr dkrnnt|||d|krdkr$nnt|||||||aRaises an appropriate APIError subclass based on the status code. Args: status_code: The HTTP status code of the response. response_json: The JSON body of the response, or a dict containing error details. response: The original response object. Raises: ClientError: If the status code is in the 4xx range. ServerError: If the status code is in the 5xx range. APIError: For other error status codes. iiiX ClientError ServerErrorrDr7rrr r r!rCs    zAPIError.raise_errorzaiohttp.ClientResponsecsxt|tjr4|jdkrdSz |IdH|}Wntjjy/|j}||j d}Ynw|j}n|t |drRt |drR|jdkrEdS|j d di}|j}n^zMddl }|}t |dra|j}t||jr|jdkroWdSz |IdH}Wn|jjy|IdH}||jd}Ynw|j}n td t|Wntytd t|w||||IdHdS) r4r5Nr6rBr7rr- _responsezUnsupported response type: )rr8r r7areadr:r;r<r=r>hasattrrBr/aiohttprKClientResponserclient_exceptionsContentTypeErrorrA ValueErrortype ImportErrorraise_error_async)rDrrrr7rNunwrapped_responser r r!raise_for_async_responsesX          z!APIError.raise_for_async_responsecsVd|kr dkrnnt|||d|krdkr%nnt|||||||rFrGrJr r r!rUs    zAPIError.raise_error_async)N)"__name__ __module__ __qualname____doc__int__annotations__rr?r r8rrstrrrrtuplerdictr' staticmethodr%rrrr StringDictr3 classmethodrErCrWrU __classcell__r r rr!r s          :r c@eZdZdZdS)rHz%Client error raised by the GenAI API.NrXrYrZr[r r r r!rHrHc@re)rIz%Server error raised by the GenAI API.Nrfr r r r!rIrgrIc@re) UnknownFunctionCallArgumentErrorzWRaised when the function call argument cannot be converted to the parameter annotation.Nrfr r r r!rhrgrhc@re)UnsupportedFunctionErrorz*Raised when the function is not supported.Nrfr r r r!ri#rgric@re)FunctionInvocationErrorzDRaised when the function cannot be invoked with the given arguments.Nrfr r r r!rj(rgrjc@re)UnknownApiResponseErrorz?Raised when the response from the API cannot be parsed as JSON.Nrfr r r r!rk-rgrk)r[typingrrrrrr8r:r?rreplay_api_clientr rN!google.auth.aio.transport.aiohttpr r r*r rHrIrRrhrirjrkExperimentalWarningr r r r!s&   v