o pkj'@s"dZddlmZmZmZmZmZddlZddlZzddl Z Wn e y)dZ Ynwddl m Z erFddl mZddlZddlmZddlZe durNejfneje jfZGdd d eZGd d d eZGd d d eZGdddeZGdddeZGdddeZGdddeZe jZdS)z Error classes for the GenAI SDK.)AnyCallableOptional TYPE_CHECKINGUnionN)_common)ReplayResponse)Responsec seZdZUdZeed<eddejdfed<dZ e e ed<dZ e e ed < d&ded e de eddejdfffd d Zd eedeee e fffddZedee e fd dfddZd e d e fddZd e d e fddZd e d e fddZd ejfddZededejdfd dfddZeded e de edejdfd dfdd Zededejd!dfd dfd"d#Zeded e de edejd!fd dfd$d%ZZ S)'APIErrorz'General errors raised by the GenAI API.codezrequests.Responser 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__Q/home/djax/ivt_ai_plugin/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__Ss 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%Zs  "zAPIError._rebuildcC8z|d|didd}|WStyYdSw)NrerrorgetAttributeError)rrrr r r!rb zAPIError._get_statuscCr,)Nrr-r.)rrrr r r!rmr1zAPIError._get_messagecCs|d|diddS)Nr r-)r/)rrr r r!rxszAPIError._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_record}s zAPIError._to_replay_recordcCs|jdkrdSt|tr+z ||}WnDtjjy*|j}||jd}Yn1wt }durRt||j rRz|}Wn|j jyQ|j|j d}Yn w|jddi}||j||dS)PRaises an error with detailed error message if the response has an error status.Nrrrr-) status_coder_HTTPX_RESPONSE_TYPESreadjsondecoderJSONDecodeErrortext reason_phraserloaded_requestsr exceptionsreason body_segmentsr/ raise_error)clsrrrrequestsr r r!raise_for_responses0      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.ClientResponsecsvt|tr3|jdkr dSz |IdH|}Wntjjy.|j}||jd}Ynw|j}n|t |drQt |drQ|jdkrDdS|j d di}|j}n^zMddl }|}t |dr`|j }t||jr|jdkrnWdSz |IdH}Wn|jjy|IdH}||jd}Ynw|j}n td t|Wntytd t|w||||IdHdS) r4r5Nr6rBr7rr- _responsezUnsupported response type: )rr8r7areadr:r;r<r=r>hasattrrBr/aiohttprLClientResponserclient_exceptionsContentTypeErrorrA ValueErrortype ImportErrorraise_error_async)rDrrrr7rOunwrapped_responser r r!raise_for_async_responsesX          z!APIError.raise_for_async_responsecsVd|kr dkrnnt|||d|krdkr%nnt|||||||rGrHrKr r r!rV s    zAPIError.raise_error_async)N)!__name__ __module__ __qualname____doc__int__annotations__rhttpxr rrstrrrrtuplerdictr' staticmethodr%rrrr StringDictr3 classmethodrFrCrXrV __classcell__r r rr!r .s      #   :r c@eZdZdZdS)rIz%Client error raised by the GenAI API.NrYrZr[r\r r r r!rI&rIc@rg)rJz%Server error raised by the GenAI API.Nrhr r r r!rJ+rirJc@rg) UnknownFunctionCallArgumentErrorzWRaised when the function call argument cannot be converted to the parameter annotation.Nrhr r r r!rj0rirjc@rg)UnsupportedFunctionErrorz*Raised when the function is not supported.Nrhr r r r!rk5rirkc@rg)FunctionInvocationErrorzDRaised when the function cannot be invoked with the given arguments.Nrhr r r r!rl:rirlc@rg)UnknownApiResponseErrorz?Raised when the response from the API cannot be parsed as JSON.Nrhr r r r!rm?rirm)r\typingrrrrrr_r:httpx2rUrreplay_api_clientr rO!google.auth.aio.transport.aiohttpr r rEr8r*r rIrJrSrjrkrlrmExperimentalWarningr r r r!s4     y