o k Di@sdZddlmZmZmZmZmZddlZddlZddl m Z er*ddl m Z ddl Z GdddeZGd d d eZGd d d eZGd ddeZGdddeZGdddeZGdddeZe jZdS)z Error classes for the GenAI SDK.)AnyCallableOptional TYPE_CHECKINGUnionN)_common)ReplayResponsec seZdZUdZeed<edejdfed<dZ e e ed<dZ e e ed< d$ded e de edejdfffd d Zd eed eee e fffddZedee e fd dfddZd e d e fddZd e d e fddZd e d e fddZd ejfddZededejfd dfddZeded e de edejdfd dfddZededejdfd dfd d!Zeded e de edejdfd dfd"d#ZZ S)%APIErrorz'General errors raised by the GenAI API.coder zaiohttp.ClientResponseresponseNstatusmessage response_jsoncs|t|trt|dkr|d}||_||_|||_|||_|r%|n| ||_ t |j d|jd|jdS)Nrr . ) isinstancelistlenr details _get_messager _get_statusr _get_coder super__init__)selfr rr  __class__S/home/air/goalskill_t/back/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)rstaterrr __reduce__8s zAPIError.__reduce__r$cCs<tt}|j|t||jd|jd|j|S)z"Rebuilds the error from the state.rr) r __new__r!update Exceptionrr r r)r$objrrrr#?s  "zAPIError._rebuildcC|d|diddS)Nr errorgetrrrrrrGzAPIError._get_statuscCr*)Nrr+r,r.rrrrLr/zAPIError._get_messagecCr*)Nr r+r,r.rrrrQr/zAPIError._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 rr r0)rrrr_to_replay_recordVs zAPIError._to_replay_recordcCs~|jdkrdSt|tjr,z ||}Wntjjy+|j}||j d}Yn w|j d di}| |j||dS)PRaises an error with detailed error message if the response has an error status.Nrr rr+) status_coderhttpxResponsereadjsondecoderJSONDecodeErrortext reason_phrase body_segmentsr- raise_error)clsr rrrrrraise_for_responseds    zAPIError.raise_for_responser5cCsTd|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 ServerErrorr@r5rr rrrr?{s    zAPIError.raise_errorcsld}d}t|tjr8|jdkrdSz |IdH|}Wntjjy3|j}||j d}Ynw|j}nrt |drVt |drV|jdkrIdS|j d di}|j}nTzCddl }t||jr|jdkriWdSz |IdH}Wn|jjy|IdH}||jd}Ynw|j}n tdt|Wntytdt|w||||IdHdS) r2rNr3r4r>r5r+zUnsupported response type: )rr6r7r5areadr9r:r;r<r=hasattrr>r-aiohttpClientResponser client_exceptionsContentTypeErrorreason ValueErrortype ImportErrorraise_error_async)r@r r5rrrIrrrraise_for_async_responsesV          z!APIError.raise_for_async_responsecsVd|kr dkrnnt|||d|krdkr%nnt|||||||rBrCrFrrrrQs    zAPIError.raise_error_async)N)!__name__ __module__ __qualname____doc__int__annotations__rr6r7r rstrrrrtuplerdictr% staticmethodr#rrrr StringDictr1 classmethodrAr?rRrQ __classcell__rrrrr s~       4r c@eZdZdZdS)rDz%Client error raised by the GenAI API.NrSrTrUrVrrrrrDrDc@r`)rEz%Server error raised by the GenAI API.NrarrrrrErbrEc@r`) UnknownFunctionCallArgumentErrorzWRaised when the function call argument cannot be converted to the parameter annotation.Nrarrrrrcrbrcc@r`)UnsupportedFunctionErrorz*Raised when the function is not supported.Nrarrrrrdrbrdc@r`)FunctionInvocationErrorzDRaised when the function cannot be invoked with the given arguments.Nrarrrrrerbrec@r`)UnknownApiResponseErrorz?Raised when the response from the API cannot be parsed as JSON.Nrarrrrrfrbrf)rVtypingrrrrrr6r9rreplay_api_clientr rIr(r rDrErNrcrdrerfExperimentalWarningrrrrs"  L